프로그래머스 - 튜플(c++, javascript)

문제 설명

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

---

입출력 예

 

입출력 예에 대한 설명

입출력 예 #1

문제 예시와 같습니다.

 

입출력 예 #2

문제 예시와 같습니다.

 

입출력 예 #3

(111, 20)을 집합 기호를 이용해 표현하면 {{111}, {111,20}}이 되며, 이는 {{20,111},{111}}과 같습니다.

 

입출력 예 #4

(123)을 집합 기호를 이용해 표현하면 {{123}} 입니다.

 

입출력 예 #5

(3, 2, 4, 1)을 집합 기호를 이용해 표현하면 {{3},{3,2},{3,2,4},{3,2,4,1}}이 되며, 이는 {{4,2,3},{3},{2,3,4,1},{2,3}}과 같습니다.

 

풀이

음.. 그냥 모두 비교하면서 풀어야겠다는 생각이 가장 먼저 떠올라서 그렇게 풀었다.

먼저 맨 양쪽 끝의 중괄호는 제외하고

각각의 여는 중괄호를 시작으로 숫자를 분리해서 2차원 벡터에 담았다.

 

그리고 벡터에서 가장 적게 들어있는 원소 순서대로 탐색하면서

벡터의 원소와 answer의 원소를 비교하여 없는 원소를 answer에 추가해나간다.

 

c++

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

vector<int> solution(string s) {
    vector<int> answer;
    vector< vector<int> > v;
    
    for(int i=1; i<s.size()-1; i++){
        vector<int> tmp;
        if(s[i] == '{'){
            string tmp_str = "";
            for(int j=i+1; j<s.size()-1; j++){
                if(s[j] == '}') {
                    i = j; 
                    break;
                }
                
                if(s[j] != ','){
                    tmp_str += s[j];
                    if(s[j+1] == '}' || s[j+1] == ','){
                        tmp.push_back( stoi(tmp_str) );
                        tmp_str = "";
                    }
                }
            }
            v.push_back(tmp);
        }
    }
    
    int size = 1;
    while(size <= s.size()){
        for(int i=0; i<v.size(); i++){
            if(v[i].size() == size){
                for(int j=0; j<v[i].size(); j++){
                    if(answer.empty()) answer.push_back(v[i][j]);
                    else{
                        bool exist = false;
                        for(int m=0; m<answer.size(); m++){
                            if(v[i][j] == answer[m]) exist = true;
                        }
                        if(!exist) {
                            answer.push_back(v[i][j]);
                            break;
                        }
                    }
                }
            }
        }
        ++size;
    }
    
    return answer;
}

 

javascript

function solution(s) {
    const answer = [];
    const arr = [];
    
    for(let i=1; i<s.length-1; i++){
        if(s[i] == '{'){
            const tmp = [];
            let tmp_str = "";
            for(let j=i+1; j<s.length-1; j++){
                if(s[j] == '}') {
                    i = j; 
                    break;
                }
                
                if(s[j] != ','){
                    tmp_str += s[j];
                    if(s[j+1] == '}' || s[j+1] == ','){
                        tmp.push( parseInt(tmp_str) );
                        tmp_str = "";
                    }
                }
            }
            arr.push(tmp);
        }
    }
    
    let size = 1;
    while(size <= s.length){
        for(let i=0; i<arr.length; i++){
            if(arr[i].length == size){
                for(let j=0; j<arr[i].length; j++){
                    if(answer.length == 0) answer.push(arr[i][j]);
                    else{
                        let exist = false;
                        for(let m=0; m<answer.length; m++){
                            if(arr[i][j] == answer[m]) exist = true;
                        }
                        if(!exist) {
                            answer.push(arr[i][j]);
                            break;
                        }
                    }
                }
            }
        }
        ++size;
    }
    
    return answer;
}

 

javscript 좋아요 많이 받은 분의 코드

정규식, sort, reduce, concat, includes를 사용하여 훠~~~얼씬 간결하게 짜셨다. 굿!

문자열 관련 문제는 확실히 자바스크립트가 더 편한 것 같다.

function solution(s) {
    return JSON.parse(s.replace(/{/g, '[').replace(/}/g, ']'))
    .sort((a, b) => a.length - b.length)
    .reduce((arr, v, n) => {
        if (n) {
            return arr.concat(v.filter(f => !arr.includes(f)));
        }
        return v;
    }, []);
}