[프로그래머스] 튜플

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

풀이

문자열 처리하는 부분에서 상당히 까다로운 문제이다. 문자열을 vector<vector<int>>로 변환하여 vector<int>의 크기를 오름차순으로 정렬한 뒤, answer 배열에 없는 수를 넣어줌으로써 튜플을 찾았다. 이 때 answer 배열에 있는지 확인하는 것은 set을 사용하였다.

코드

#include <string>
#include <vector>
#include <set>
#include <algorithm>
using namespace std;

// 튜플의 크기가 오름차순이 되도록 정렬
bool cmp(vector<int>& a, vector<int>& b) {
    return a.size() < b.size();
}

// 문자열을 벡터로 변환
vector<vector<int>> parse_string(string s) {
    vector<vector<int>> ret;
    int num = 0;
    vector<int> tmp;
    for (int i = 0; i < s.length(); i++) {
        if (s[i] == '{') continue;
        else if ('0' <= s[i] && s[i] <= '9') {
            num = 10 * num + (s[i] - '0');
        }
        else if (s[i] == ',') {
            if (num != 0) {
                tmp.push_back(num);
                num = 0;
            }
        }
        else if (s[i] == '}' && num != 0) {
            tmp.push_back(num);
            ret.push_back(tmp);
            num = 0;
            tmp.clear();
        }
    }
    return ret;
}

vector<int> solution(string s) {
    vector<int> answer;
    set<int> tmp_set;
    vector<vector<int>> tmp = parse_string(s);
    sort(tmp.begin(), tmp.end(), cmp);


    for (int i = 0; i < tmp.size(); i++) {
        for (int j = 0; j < tmp[i].size(); j++) {
            // 처음 등장한 숫자의 경우 answer에 넣어준다
            if (tmp_set.find(tmp[i][j]) == tmp_set.end()) {
                tmp_set.insert(tmp[i][j]);
                answer.push_back(tmp[i][j]);
            }
        }
    }
    return answer;
}