[프로그래머스][문자열][C++] 튜플

튜플

https://programmers.co.kr/learn/courses/30/lessons/64065

코딩테스트 연습 - 튜플

문제 설명

셀수있는 수량의 순서있는 열거 또는 어떤 순서를 따르는 요소들의 모음을 튜플(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 이하인 경우만 입력으로 주어집니다.

풀이 방법

처음엔 unordered_set의 중복이 안된다는 성질을 이용해서 unorderd_set에 넣어놓고 다시 answer에 넣는 방식으로 해결하려 했는데 unorderd_set에 insert할 때 순서대로 들어가는 게 아니라 랜덤한 위치에 들어가서 실패했다.

결국 unordered_map을 이용해서 해결했다.

튜플에는 순서가 있고 집합에는 순서가 없으므로 순서를 잘 지켜서 answer에 넣어줘야 하는데, 먼저 2중 벡터를 만들어서 원소의 집합을 그대로 벡터에 넣어주고 벡터의 size가 작은 순으로 정렬했다.

그러면 벡터는 {{원소 1개}, {원소 2개}, {원소 3개}, ..., {원소 n개}}로 정렬되므로 원소 1개부터 원소 n개까지 안 넣은 숫자들만 넣어주면 올바른 순서로 튜플이 완성된다.

unorderd_map을 통해 넣었는지 안넣었는지를 구분했는데, unorderd_map 또한 중복 Key가 들어갈 수 없으므로 Key값을 원소(int), Value값을 boolean으로 설정하여 BFS/DFS의 visited처럼 사용하였다.

소스 코드

#include <string>
#include <vector>
#include <algorithm>
#include <unordered_map>

using namespace std;

vector<int> solution(string s) {
    vector<int> answer;
    vector<vector<int>> num_array;
    unordered_map<int, bool> visited;

    s.erase(0, 1);
    s.erase(s.size() - 1, 1);
    for(int i = 0; i < s.size(); i++) {
        if(s[i] == '{') {
            vector<int> tmp;
            int num_start = i + 1, num_end = 0;
            for(int j = i + 1; j < s.size(); j++) {
                if(s[j] == ',' || s[j] == '}') {
                    num_end = j - 1;
                    int num = stoi(s.substr(num_start, num_end - num_start + 1));
                    tmp.push_back(num);
                    visited[num] = false;
                    if(s[j] == '}')
                        break;
                    else
                        num_start = j + 1;
                }
            }
            num_array.push_back(tmp);
        }
    }

    struct {
        bool operator()(vector<int> a, vector<int> b) {
            return a.size() < b.size();
        }
    } cmp;

    sort(num_array.begin(), num_array.end(), cmp);

    for(auto array : num_array) {
        for(auto num : array) {
            auto iter = visited.find(num);
            if(!iter->second) {
                answer.push_back(iter->first);
                visited.at(num) = true;
            }
        }
    }

    return answer;
}