문제
설명
셀수있는 수량의 순서있는 열거 또는 어떤 순서를 따르는 요소들의 모음을 튜플(tuple)이라고 합니다. n개의 요소를 가진 튜플을 n-튜플(n-tuple)이라고 하며, 다음과 같이 표현할 수 있습니다.
- (a1, a2, a3, ..., an)
튜플은 다음과 같은 성질을 가지고 있습니다.
- 중복된 원소가 있을 수 있습니다. ex : (2, 3, 1, 2)
- 원소에 정해진 순서가 있으며, 원소의 순서가 다르면 서로 다른 튜플입니다. ex : (1, 2, 3) ≠ (1, 3, 2)
- 튜플의 원소 개수는 유한합니다.
원소의 개수가 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] |
풀이
설명
- 문자열을 집합 단위로 파싱하고, 집합의 원소의 개수에 따라 순서대로 정렬한다.
- 집합을 작은 것부터 탐색하면서 추가된 숫자를 찾아 결과 배열에 추가한다.
새로 배운 것
문자열을 파싱하는 것을 막연하게만 알고 있었는데, 이번 기회에 split(), trim() 등 메소드 기능을 정확하게 찾아보고 깃헙에 정리해놓았다.
그리고, 집합을 차례대로 탐색하며 추가할 숫자를 찾을 때, 원래는 contains()로 체크하고 add()를 따로 하는 방식으로 다음과 같이 작성했었다.
if (!numbers.contains(element)) {
result[index++] = Integer.parseInt(element);
numbers.add(element);
}하지만 numbers.add(element) 자체가 결괏값으로 성공했을 때는 true를 반환하고 이미 그 element가 set에 있는 경우에는 false를 반환하기 때문에, 이를 그냥 조건문에 넣어 체크하는 방법도 있다는 것을 알게 되었다. 그래서 더 간결하게 바뀐 코드는 아래와 같다.
if (numbers.add(element)) result[index++] = Integer.parseInt(element);set에 한 번만 접근하면 되어서 효율적이고 딱 보기에도 의미가 명확해졌다.
코드
import java.util.*;
class Solution {
public int[] solution(String s) {
String[] parsedChunks = parseChunks(s);
int[] result = new int[parsedChunks.length];
int index = 0;
Set<String> numbers = new HashSet<>();
for (String chunk : parsedChunks)
for (String element : chunk.split(","))
if (numbers.add(element)) result[index++] = Integer.parseInt(element);
return result;
}
private String[] parseChunks(String s) {
String[] chunks = s.substring(2, s.length() - 2).split("},\\{");
return orderChunks(chunks);
}
private String[] orderChunks(String[] chunks) {
String[] result = new String[chunks.length];
for (String chunk : chunks) {
String[] elements = chunk.split(",");
result[elements.length - 1] = chunk;
}
return result;
}
}