[프로그래머스] 튜플 Java 풀이

이 글은 혼자 학습한 내용을 바탕으로 작성되었습니다.

틀리거나 잘못된 정보가 있을 수 있습니다.

댓글로 알려주시면 수정하도록 하겠습니다.

---

1. 문제

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

2. 입력

• s의 길이는 5 이상 1,000,000 이하입니다.
• s는 숫자와 '{', '}', ',' 로만 이루어져 있습니다.
• 숫자가 0으로 시작하는 경우는 없습니다.
• s는 항상 중복되는 원소가 없는 튜플을 올바르게 표현하고 있습니다.
• s가 표현하는 튜플의 원소는 1 이상 100,000 이하인 자연수입니다.
• return 하는 배열의 길이가 1 이상 500 이하인 경우만 입력으로 주어집니다.

3. 예제

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]

4. 풀이

입력으로 주어지는 문자열에서 필요한 부분은 문자열 그룹은 중괄호를 통해 묶여있는 문자열이 필요합니다.

예제 1번으로 보았을 때 "{2}, {2,1}, {2,1,3}, {2,1,3,4}"가 필요한 부분입니다.

 

이제 필요한 부분의 문자열을 괄호 단위로 분리하는 작업이 필요합니다. 분리를 위해 "," 문자로 Split을 할 경우 숫자 사이에 있는 "," 문자로 인해 우리가 원하는 분리가 될 수 없습니다.

 

String[] groups = Pattern.compile("\\{(.*?)}")
		.matcher(s.substring(1, s.length() - 1))
		.results()
		.map(MatchResult::group)
		.toArray(String[]::new);

그래서 위 코드처럼 정규식을 통해 괄호 별로 문자열 분리작업을 수행합니다.

 

분리가 끝난 뒤 만들어진 배열에서 길이 별로 정렬이 필요합니다.

 

정렬이 필요한 이유는 튜플을 구성하는 숫자들 중에서 가장 앞에 있는 숫자는 무조건 자신 혼자 그룹을 이루는 문자열이 존재합니다.

 

 [2, 1, 3, 4] 튜플에서는 무조건 {2}가 존재하며 [3, 2, 4, 1] 튜플에서는 {3}이 무조건 존재합니다.

 

가장 짧은 문자열의 숫자가 바로 튜플의 시작하는 숫자가 되는 것입니다.

 

그다음 긴 문자열은 숫자 2개로 구성된 그룹일 것입니다.

 

예) [2, 1, 3, 4] 튜플에서 숫자 2개로 만들 수 있는 그룹은 {1,2}, {2,1}입니다. 그룹에 숫자의 순서는 다양하게 있을 수 있지만 이미 튜플의 시작 숫자는 2라는 것을 알고 있으므로 튜플의 2 다음 숫자가 1이라는 것을 알 수 있습니다.

 

그다음 긴 문자열은 숫자 3개로 구성된 그룹일 것입니다.

 

예) [2, 1, 3, 4] 튜플에서 숫자 3개로 만들 수 있는 그룹은 {1,2,3}, {3,2,1}, {2,3,1}, {2,1,3} 등입니다. 이번 그룹 또한 숫자의 순서는 다양하게 있을 수 있지만 이미 튜플의 시작 숫자는 2이고 그다음 숫자가 1이라는 것을 알고 있으므로 다음 숫자는 3이라는 것을 알 수 있습니다.

 

Arrays.sort(groups, Comparator.comparingInt(String::length));

이러한 과정을 진행하기 위해서는 문자열의 길이(숫자의 개수가) 짧은 것을 이용하여 튜플의 앞의 숫자를 찾고 그 다음 숫자를 찾아 나가기 위해 위 코드처럼 정렬이 필요합니다.

 

List<Integer> answer = new ArrayList<>();
for(String sub : groups){
	sub = sub.substring(1, sub.length() - 1);
	String[] numbers = sub.split(",");

	for(String number : numbers){
		int num = Integer.parseInt(number);
		if(!answer.contains(num))
			answer.add(num);
	}
}

이제 위에 설명한 방법대로 짧은 문자열에서 찾은 숫자를 제외한 숫자가 튜플의 다음 숫자가 되는 점을 이용하여 List의 Element를 구성합니다.

 

이후 만들어진 List를 int 배열로 변환하여 반환하면 문제를 해결할 수 있습니다.

 

5. 코드

전체 코드는 Git에 있습니다.

String[] groups = Pattern.compile("\\{(.*?)}")
		.matcher(s.substring(1, s.length() - 1))
		.results()
		.map(MatchResult::group)
		.toArray(String[]::new);

Arrays.sort(groups, Comparator.comparingInt(String::length));

List<Integer> answer = new ArrayList<>();
for(String sub : groups){
	sub = sub.substring(1, sub.length() - 1);
	String[] numbers = sub.split(",");

	for(String number : numbers){
		int num = Integer.parseInt(number);
		if(!answer.contains(num))
			answer.add(num);
	}
}

return answer.stream().mapToInt(Integer::intValue).toArray();