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

 

 

특정 튜플을 표현하는 집합이 담긴 문자열 s가 매개변수로 주어질 때,

s가 표현하는 튜플을 배열에 담아 반환하는 문제

 

 

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

a1은 n번, a2는 n-1번 ... an은 1번 입력되는걸 확인할 수 있다

: 입력값을 빈도수대로 정렬하면 끝나는 문제!!

 

예를들어 "{{4,2,3},{3},{2,3,4,1},{2,3}}" 이 문자열은

3은 4번, 2는 3번, 4는 2번 1은 1번 입력됐으니

a1 = 3, a2 = 2, a3 = 4, a4 = 1이 된다

∴ [3, 2, 4, 1]

 

코드상으로 네 과정을 거쳐서 튜플을 찾을 수 있다

1. 입력된 문자열에서 괄호와 쉼표를 제외한 숫자만 map 변수에 저장하기

2. 정렬하기 위해 map > vector 값 옮기기

3. 빈도수 기준으로 내림차순 정렬

4. answer vector에 원소만 값 옮기기

 

 

#include <string>
#include <vector>
#include <map>
#include <algorithm>
#include <iostream>
 
using namespace std;
 
//두 번째 값 기준 내림차순 정렬
bool compare(pair<int, int> &a, pair<int, int> &b) {
    return a.second > b.second;
}
 
vector<int> solution(string s) {
    vector<int> answer;
    map<int, int, greater<int>> temp;
    int element = 0;
 
    //입력된 문자열에서 숫자만 temp에 저장하기
    for (int i = 0; i < s.size(); i++) {
        if (s[i] >= '0' && s[i] <= '9')
            element = element * 10 + s[i] - '0';
        else if (element != 0) {
            temp[element]++;
            element = 0;
        }
    }
 
    //정렬하기 위해 벡터에 값 옮기기 (first = 원소, second = 빈도수)
    vector<pair<int, int>> sortedTemp;
    for (auto x : temp) {
        sortedTemp.push_back({ x.first,x.second });
    }
 
    sort(sortedTemp.begin(), sortedTemp.end(), compare);
 
    //정렬된 벡터 출력변수에 옮기기
    for (auto x : sortedTemp) {
        answer.push_back(x.first);
    }
 
    return answer;
}

 

map 쓸 생각을 못해서 좀 헤맸다

코딩 너무 오랜만..^^