[프로그래머스 Level 2] 튜플 (2019 카카오 인턴)

🖊️ 문제

튜플  

 

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

 

🖥️ 코드

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

using namespace std;

bool cmp(const pair<int, int>& a, const pair<int, int>& b)
{
    return a.second > b.second;
}

vector<int> solution(string s)
{
    vector<int> answer;
    unordered_map<int, int> m;
    int num;
    s = regex_replace(s, regex("[{}]"), "");
    s = regex_replace(s, regex(","), " ");
    stringstream ss(s);
    while (ss >> num)
        m[num]++;

    vector<pair<int, int>> v(m.begin(), m.end());

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

    for (int i = 0; i < v.size(); i++)
        answer.push_back(v[i].first);

    return answer;
}

 

🤔 나의 생각

{{a1}, {a1, a2}, {a1, a2, a3}, {a1, a2, a3, a4}, ... {a1, a2, a3, a4, ..., an}}에서 힌트를 얻었다.

순서가 어찌 됐던 1번 원소는 n회 등장하고 2번 원소는 n-1회 등장,  n번 원소는 1회 등장한다. 

 

문자열에서 원소를 바로 구할 수 있으면 좋겠지만 괄호와 쉼표, 원소가 하나의 문자열에 들어가 있다.

for문을 돌며 원소를 구할까 하다가 regex_replace와 정규 표현식으로 문자를 지우고 stringstream으로 숫자만 추출했다.

마지막으로 map으로 원소에 따른 카운트를 세고 정렬하고 값을 추출하면 된다!

 

여담으로 regex_replace를 사용하니 실행 시간이 꽤나 오래 걸렸다. 입력이 많은 경우는 사용하면 안 될듯하다.