[프로그래머스 코딩테스트] 튜플(Java)

  문제 설명

  셀수있는 수량의 순서있는 열거 또는 어떤 순서를 따르는 요소들의 모음을 튜플(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]

import java.util.*;
class Solution {
    public int[] solution(String s) {
        String[] array = s.replaceAll("[{}]", "").split(",");
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < array.length; i++) {
            int num = Integer.parseInt(array[i]);
            map.put(num, map.getOrDefault(num, 0) + 1);
        }
        int[] answer = map.entrySet()
            .stream()
            .sorted(Map.Entry.<Integer, Integer>comparingByValue().reversed())
            .mapToInt(Map.Entry::getKey)
            .toArray();
        return answer;
    }
}

  배열 array를 생성합니다. 문자열 s를 replaceAll 메소드를 이용하여 중괄호를 공백으로 바꿔줍니다. 이를 다시 split 메소드를 이용하여 쉼표로 구분하여 배열로 할당합니다.

  제네릭 타입으로 Integer를 갖는 Map을 생성합니다.

  for문이 array의 길이만큼 순회합니다.

  변수 num을 생성합니다. 현재 array는 문자열 원소로 이루어진 배열이므로 Integer 클래스의 parseInt 메소드를 이용하여 array의 i 번째 원소를 정수로 변환한 후, 할당합니다.

  map에 put 메소드를 이용하여 num과 getOrDefault 메소드를 이용하여 num이 이미 맵에 존재한다면 값이 증가하도록 합니다.

  배열 answer을 생성한 뒤 stream을 이용하여 map의 value를 기준으로 내림차순 정렬하여 map을 배열로 변환합니다. entrySet으로 map을 순회합니다. stream을 생성합니다. sorted로 정렬하되, comparingByValue로 값을 기준으로 정렬합니다. reversed로 내림차순을 기준으로 정렬합니다. mapToInt로 stream을 IntStream으로 변환합니다. map을 순환하며 key를 가져옵니다. toArray로 배열로 변경합니다.

 

  스트림을 이용한 풀이는 처음입니다. 대개 일반적인 구현의 성능이 더 좋기 때문에 스트림을 잘 쓰지 않아왔는데 막상 써보니까 가독성은 물론이거니와 풀이할 때에도 매우 편하네요…. 그래도 일반적인 구현을 더 우선시하여 풀어보겠지만, 스트림을 사용하는 것도 좋은 선택인 것 같습니다.