[Java/자바] 프로그래머스 Lv2 - 튜플

문제 설명

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

 

[입출력 예]

s	result
"{{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]
"{{20,111},{111}}"	[111, 20]
"{{123}}"	[123]
"{{4,2,3},{3},{2,3,4,1},{2,3}}"	[3, 2, 4, 1]

 

---

Solution.java

result에 들어가는 수의 순서를 이해할 필요가 있습니다.

s가 "{{1,2,3},{2,1},{1,2,4,3},{2}}"일 때 어떤 순서로 result가 [2, 1, 3, 4]를 갖는지 봅시다.

result의 첫 번째 수는 "2"인데 왜 2가 먼저 들어갔는지 s가 "{{2},{2,1},{2,1,3},{2,1,3,4}}"일 때와 비교해보면 쉽게 알 수 있습니다. 

배열의 길이가 1로 가장 작은 {2}가 먼저 result에 담기고, 그 다음으로 큰 {2, 1}의 1이 result 담깁니다. (튜플에는 중복된 원소가 없기 때문)

또 그 다음으로 큰 {2, 1, 3}의 3이 result에 담기고, 마지막으로 {2, 1, 3, 4}의 4가 result에 담기게 됩니다.

 

import java.util.ArrayList;
import java.util.Comparator;
import java.util.Arrays;
import java.util.HashSet;
import java.util.stream.Stream;

class 튜플 {
    public static void main(String[] args) {
        System.out.println(Arrays.toString(solution("{{2},{2,1},{2,1,3},{2,1,3,4}}")));
        System.out.println(Arrays.toString(solution("{{1,2,3},{2,1},{1,2,4,3},{2}}")));
        System.out.println(Arrays.toString(solution("{{20,111},{111}}")));
    }

    public static int[] solution(String s) {
        s = s.replaceAll("\\{", "");
        String[] sArr = s.substring(0, s.length() - 2).split("\\}\\,");
        Arrays.sort(sArr, Comparator.comparingInt(String::length));

        HashSet<Integer> duplicate = new HashSet<>();
        ArrayList<Integer> result = new ArrayList<>();

        for (int i = 0; i < sArr.length; i++) {
            int[] tuple = Stream.of(sArr[i].split(",")).mapToInt(Integer::parseInt).toArray();
            for (int num : tuple) {
                if (!duplicate.contains(num)) {
                    result.add(num);
                }
                duplicate.add(num);
            }
        }

        int[] answer = new int[result.size()];

        for (int i = 0; i < result.size(); i++) {
            answer[i] = result.get(i);
        }

        return answer;
    }
}

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

문제 풀이

1. 정규식을 이용하여 "중괄호 {"를 제거합니다.
2. 그러면 "중괄호 }"만 남는데, s의 맨끝의 "}}"를 제거하고, "},"를 기준으로 배열(sArr)을 만듭니다.
3. result에 들어가는 수의 순서를 맞추기 위해 만들어진 배열(sArr)를 배열의 길이 순으로 오름차순 정렬합니다.
4. 중복을 제거하기 위해 Set 클래스를 생성하고 튜플의 원소를 담을 ArrayList 클래스를 생성합니다.(이미 컬렉션 클래스를 사용해 배열 대신 ArrayList 사용)
5. 만약 중복된 값이 없다면 result에 해당 원소를 추가합니다.
6. 완성된 result를 함수 반환 타입 int[]에 맞춰 변경합니다.

반복문에서 어떤 식으로 result에 원소를 추가하는지 이해를 돕고자 아래 그림을 첨부합니다.

ㄴ