본문 바로가기

코딩 테스트(Coding test)/Lv. 2

[프로그래머스/C++] 튜플

미숙한 블로그 주인이 코딩테스트 문제를 풀어가는 과정을 담은 글입니다. 이 풀이가 효율적인 풀이가 아닐 수 있으며, 부정확한 정보가 많이 있을 수 있습니다. 보완해야할 점이 있다면 댓글로 남겨주세요!

 

튜플

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

 

코딩테스트 연습 - 튜플

"{{2},{2,1},{2,1,3},{2,1,3,4}}" [2, 1, 3, 4] "{{1,2,3},{2,1},{1,2,4,3},{2}}" [2, 1, 3, 4] "{{4,2,3},{3},{2,3,4,1},{2,3}}" [3, 2, 4, 1]

programmers.co.kr

문제

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

 

풀이

입출력 예시

 

풀이

문제를 바꿔서 말하면, 수들을 중복이 없도록 저장하고 많이 나온 순서대로 반환하라는 소리다. 이를 위해, 수를 세어 줄 map 하나와, 우선순위 큐 하나를 선언해주자.

#include <string>
#include <vector>
#include <map>
#include <queue>

using namespace std;

struct comp {
    bool operator()(pair<int, int> a, pair<int, int> b) {
        return a.second < b.second; // 내림차순으로 정렬
    }
};

vector<int> tuplet(string s) {
    vector<int> answer;
    map<int, int> map_tuple;
    priority_queue<pair<int, int>, vector<pair<int, int>>, comp> pq_tuple;
    
    return answer;
}

수가 '{',  ',', '}' 사이사이에 존재하므로 숫자일 경우에만 tmp에 계속 추가시켜 준 후, 숫자가 아닌 것이 나오는 순간 저장을 하면 된다. 이때, map_tuple에 수가 저장되어 있지 않으면 key를 해당 수로, value를 1로 해서 추가시켜주자. 이미 추가되어 있다면 값을 1 추가해주면 된다.

...
    ...
    string tmp = "";
    for (int i = 0; i < s.length(); i++) {
        if (isdigit(s[i])) {
            tmp += s[i];
        }
        else if (tmp != "") {
            int check = stoi(tmp);
            if (map_tuple.count(check)) {
                map_tuple[check] += 1;
            }
            else {
                map_tuple.insert({ check, 1 });
            }
            tmp = "";
        }
    }
    ...
...

나머지 연산은 그저 우선순위 큐에 value를 기준으로 값을 저장 후, answer로 추가시켜 주는 과정만 해주면 된다.

...
    ...
    for (const auto& t : map_tuple) {
        pq_tuple.push(t);
    }

    int qsize = pq_tuple.size();

    for (int i = 0; i < qsize; i++) {
        answer.push_back(pq_tuple.top().first);
        pq_tuple.pop();
    }
    ...
...

전체 코드

#include <string>
#include <vector>
#include <map>
#include <queue>

using namespace std;

struct comp {
    bool operator()(pair<int, int> a, pair<int, int> b) {
        return a.second < b.second; // 내림차순으로 정렬
    }
};

vector<int> solution(string s) {
    vector<int> answer;
    map<int, int> map_tuple;
    priority_queue<pair<int, int>, vector<pair<int, int>>, comp> pq_tuple;
    
    string tmp = "";
    for (int i = 0; i < s.length(); i++) {
        if (isdigit(s[i])) {
            tmp += s[i];
        }
        else if (tmp != "") {
            int check = stoi(tmp);
            if (map_tuple.count(check)) {
                map_tuple[check] += 1;
            }
            else {
                map_tuple.insert({ check, 1 });
            }
            tmp = "";
        }
    }

    for (const auto& t : map_tuple) {
        pq_tuple.push(t);
    }

    int qsize = pq_tuple.size();

    for (int i = 0; i < qsize; i++) {
        answer.push_back(pq_tuple.top().first);
        pq_tuple.pop();
    }

    return answer;
}

 

결과