[프로그래머스/python] - 튜플

문제 설명

셀수있는 수량의 순서있는 열거 또는 어떤 순서를 따르는 요소들의 모음을 튜플(tuple)이라고 합니다. n개의 요소를 가진 튜플을 n-튜플(n-tuple)이라고 하며, 다음과 같이 표현할 수 있습니다.

 

• (a1, a2, a3, ..., an)

 

원소의 개수가 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]

 

---

풀이

 

def solution(s):
	s = s[1:-1]
	s = s.replace('},','}')
	li = []; k = 0; i = 0; r = ''
	for i in range(len(s)):
		if s[i] == '{':
			li.append([])
		elif s[i] == '}':
			li[k].append(int(r))
			r = ''
			k += 1
		else :
			if s[i] == ',':
				li[k].append(int(r))
				r = ''
				continue
			r += s[i]

	li.sort(key = lambda x : len(x), reverse = True)
	answer = []


	for i in range(len(li)-1):
		answer.insert(0,list(set(li[i]) - set(li[i+1]))[0])

	answer.insert(0,li[len(li)-1][0])

	return answer

 

---

 

풀이 순서

 

1) 매개 변수 s를 리스트에 저장합니다.

 

	for i in range(len(s)):
		if s[i] == '{':
			li.append([])
		elif s[i] == '}':
			li[k].append(int(r))
			r = ''
			k += 1
		else :
			if s[i] == ',':
				li[k].append(int(r))
				r = ''
				continue
			r += s[i]

 

2) 리스트안에 원소인 리스트들을 원소 개수 별로 정렬합니다.

 

       li.sort(key = lambda x : len(x), reverse = True)

 

 

3) set() 함수를 사용하여 차집합을 구합니다.

 

	for i in range(len(li)-1):
		answer.insert(0,list(set(li[i]) - set(li[i+1]))[0])

	answer.insert(0,li[len(li)-1][0])

 

 

---

사용된 개념

 

* 집합 자료형 set()

* 사용자 정의로 리스트 정렬하기