프로그래머스[Lv.2] - 튜플

 

 

문제 설명

셀수있는 수량의 순서있는 열거 또는 어떤 순서를 따르는 요소들의 모음을 튜플(tuple)이라고 합니다. n개의 요소를 가진 튜플을 n-튜플(n-tuple)이라고 하며, 다음과 같이 표현할 수 있습니다.

• (a1, a2, a3, ..., an)

튜플은 다음과 같은 성질을 가지고 있습니다.

1. 중복된 원소가 있을 수 있습니다. ex : (2, 3, 1, 2)
2. 원소에 정해진 순서가 있으며, 원소의 순서가 다르면 서로 다른 튜플입니다. ex : (1, 2, 3) ≠ (1, 3, 2)
3. 튜플의 원소 개수는 유한합니다.

원소의 개수가 n개이고, 중복되는 원소가 없는 튜플 (a1, a2, a3, ..., an)이 주어질 때(단, a1, a2, ..., an은 자연수), 이는 다음과 같이 집합 기호 '{', '}'를 이용해 표현할 수 있습니다.

• {{a1}, {a1, a2}, {a1, a2, a3}, {a1, a2, a3, a4}, ... {a1, a2, a3, a4, ..., an}}

예를 들어 튜플이 (2, 1, 3, 4)인 경우 이는

• {{2}, {2, 1}, {2, 1, 3}, {2, 1, 3, 4}}

와 같이 표현할 수 있습니다. 이때, 집합은 원소의 순서가 바뀌어도 상관없으므로

• {{2}, {2, 1}, {2, 1, 3}, {2, 1, 3, 4}}
• {{2, 1, 3, 4}, {2}, {2, 1, 3}, {2, 1}}
• {{1, 2, 3}, {2, 1}, {1, 2, 4, 3}, {2}}

는 모두 같은 튜플 (2, 1, 3, 4)를 나타냅니다.

특정 튜플을 표현하는 집합이 담긴 문자열 s가 매개변수로 주어질 때, s가 표현하는 튜플을 배열에 담아 return 하도록 solution 함수를 완성해주세요.

---

 

 

 

 

 

 

제한 사항

• s의 길이는 5 이상 1,000,000 이하입니다.
• s는 숫자와 '{', '}', ',' 로만 이루어져 있습니다.
• 숫자가 0으로 시작하는 경우는 없습니다.
• s는 항상 중복되는 원소가 없는 튜플을 올바르게 표현하고 있습니다.
• s가 표현하는 튜플의 원소는 1 이상 100,000 이하인 자연수입니다.
• return 하는 배열의 길이가 1 이상 500 이하인 경우만 입력으로 주어집니다.

---

 

 

 

 

 

 

풀이

문제를 풀기 위해선 우선적으로 문자열에서 숫자들만 뽑아낼 수 있어야 한다.

Stream을 이용하여 문자열을 파싱할 수 있지만, 이번 문제에서는 ',' , '{' , '}' 같은 문자들이 여러 개 있어 직접 파싱하였다.

'{'가 나온다면 뒤에 나오는 문자들을 포함해야 하며 '}'가 나온다면 포함을 종료하는 것을 뜻한다.

이때, 중간에 ','가 나오면 현재 숫자가 끝난다는 의미이다.

즉, ','가 나오면 임시로 저장을 하고 임시로 저장된 숫자들을 return할 vector에 넣어주면 된다.

파싱이 끝나면 간단하게 문제를 해결할 수 있다.

파싱이 끝난 후, vector의 상태는 문자열에서 특수문자들을 모두 제거하고 숫자만 남은 상태이다.

이때, 이 숫자의 vector들을 길이순으로 정렬하면 된다.

정렬이 완료되면 가장 앞에는 1개의 수만 있을 것이다.

그렇다면 튜플에 가장 먼저 오는 수는 그 수이다.

이후, 앞에서 포함되지 않은 수가 1개씩 등장한다.

이를 확인해 하나씩 넣으면 된다.

---

 

 

 

 

 

전체 코드

#include <string>
#include <vector>
#include <iostream>
#include <algorithm>
#include <set>

using namespace std;

void Parse(vector<vector<int>>& nums, string s)
{
    vector<int> temp;
    bool bInsert = false;
    int num = 0;
    
    for(int i = 1; i < s.size()-1; i++)
    {
        char c = s[i];
        if(c == '{')
        {
            bInsert = true;
            continue;
        }
        
        if(c == '}')
        {
            bInsert = false;
            continue;
        }
        
        if(bInsert) 
        {
            if(c == ',')
            {
                temp.push_back(num);
                num = 0;
                continue;
            }
            else
            {
                num *= 10;
                num += c - '0';
            }
        }
        else
        {
            temp.push_back(num);
            nums.push_back(temp);
            temp.clear();
            num = 0;
        }
    }
    
    temp.push_back(num);
    nums.push_back(temp);
}

vector<int> solution(string s) {
    vector<int> answer;
    //{} 단위로 잘라서 vector에 넣음
    vector<vector<int>> tuples;
    set<int> checkInclude;
    
    
    Parse(tuples, s);
    
    //개수순서대로 정렬
    sort(tuples.begin(), tuples.end(), [](vector<int>& a, vector<int>& b){ return a.size() < b.size(); });
    for(auto tuple : tuples)
    {
        for(auto num : tuple)
        {
            // cout << num << " ";
            if(checkInclude.count(num) == 0)
            {
                checkInclude.insert(num);
                answer.push_back(num);
            }
        }
        // cout << "\n";
    }
    
    
    
    return answer;
}

---