[ 카카오 / Python ] 2019 카카오 개발자 겨울 인턴십 - 튜플

문제 설명

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

풀이

우선 입력이 문자열 형태로 주어지다보니, 각 문자열을 어떻게 집합 즉 배열로 파싱하느냐가 첫번째 문제이다.


나는 스택을 이용해서 숫자를 만나는 경우 스택에 삽입하고, }를 만나면 해당 스택이 집합이라고 판단한다.


그리고 나서 해당 스택이 현재 문자열 형태로 들어가 있으므로 정수형으로 치환해준다.


그런 다음 하나의 튜플에서 여러 개의 집합 표현이 존재하는데, 이 때 어떻게 하면 원래의 튜플 모습을 찾을 수 있을까 고민했다.


집합들을 자세히 보니, 집합의 길이가 1일 때부터 n일 때 까지의 집합들이 있다.


이 말은 길이가 1인 집합은 a1 원소로 이루어져 있고, 길이가 2인 집합은 a1, a2 원소로 이루어져있다. 다만 길이가 2이상 부터는 순서가 뒤 섞일 수 있다.


그러니까, 길이가 1인 집합으로 부터 시작해서 n인 집합까지 한 단계 씩 이동해가면 원소가 하나씩 확정이 된다.


그럼 마지막에 원하는 튜플의 모습을 찾을 수 있다.

코드

from collections import deque
def solution(s):
    answer = []

    combs = []
    s = deque(s)

    comb = []

    while s:
        c = s.popleft()
        if c == "{":
            continue
        elif c == "}":
            if not comb:
                continue
            if comb and comb[0] == ',':
                comb.pop(0)
            comb = list(map(int, ''.join(comb).split(',')))
            combs.append(comb)
            comb = []
        else:
            comb.append(c) 
    combs.sort(key= lambda x: len(x))

    for comb in combs:
        comb = deque(comb)

        if not answer:
            answer.append(comb.popleft())

        else:
            for i in answer:
                if i in comb:
                    comb.remove(i)
            answer.append(comb.popleft())

    return answer