프로그래머스 - 튜플 (Java풀이)

출처 : 프로그래머스 - 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 이하인 경우만 입력으로 주어집니다.

입출력 예에 대한 설명입출력 예 #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}}과 같습니다.

 

 

 

[ Java 풀이 ]

import java.util.Comparator;
import java.util.HashMap;
import java.util.PriorityQueue;

class Solution {
    public int[] solution(String s) {
        s = s.substring(1, s.length()-1).replace("{", ""); // 입력 문자열에서 필요 없는 부호들을 없애줍니다.
        String[] str = s.split("}"); // }을 기준으로 나눠줘요.
        int max=0;
        
        // 길이가 작은 순으로 정렬해주는 PriorityQueue를 만듭니다.
        PriorityQueue<String> pq = new PriorityQueue<>(new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                if (o1.length()<o2.length()) return -1;
                else return 1;
            }
        });
        
        // 중복된 값을 방지하기 위한 HashMap 생성/사용
        HashMap<Integer, Integer> hm = new HashMap<>();
        for (int i=0;i<str.length;i++) {
        	// 위에서 }을 기준으로 split해줬을 때, ','로 시작하는 문자열이 될 수 있어요.
            // 그래서 앞에 ','로 시작하는 문자열은 ','를 없애줘요.
            if (str[i].charAt(0)==',') str[i] = str[i].substring(1);
            
            pq.add(str[i]);
            int l = str[i].split(",").length;
            
            // 나중에 answer의 배열 길이를 선언해주기 위한 max입니다.
            if (max<l) max = l;
        }
        int[] answer = new int[max];
        int idx=0;
        
        while (!pq.isEmpty()) {
            str = pq.poll().split(",");
            for (int i=0;i<str.length;i++) {
                int num = Integer.parseInt(str[i]);
                
                // 길이가 작은 순으로 HashMap에 없는 걸 넣는 과정을 반복해주면, 자연스럽게 튜플의 순서가 결정됩니다.
                if (!hm.containsKey(num)) {
                    answer[idx++] = num;
                    hm.put(num, 0);
                }
            }
        }
        return answer;
    }
}