[프로그래머스] 튜플

문제

• n-튜플(n-tuple): n개의 요소를 가진 튜플
  • (a1, a2, a3, ..., an)
• 튜플 성질
  • 중복된 원소가 있을 수 있다.
  • 원소에 정해진 순서가 있으며, 원소의 순서가 다르면 서로 다른 튜플이다.
  • 튜플의 원소 개수는 유한하다.

특정 튜플을 표현하는 집합이 담긴 문자열 s가 매개변수로 주어질 때, s가 표현하는 튜플을 배열에 담아 return 하도록 solution 함수를 완성해주세요.

제한사항

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

코드

def solution(s):
    answer = []
    # 숫자로 이루어진 배열로 변경
    start, tmp = 0, []
    for i in range(len(s)-1):
        if s[i] == '{':
            start = i
        elif s[i] == '}':
            tmp.append(list(map(int,s[start+1:i].split(','))))
    tmp.sort(key=lambda x: len(x))
    for t in tmp:
        answer.append([x for x in t if x not in answer][0])
    return answer