[Programmers] 프로그래머스 lv.2 튜플

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

• 조건
  1. s의 길이는 5 이상 1,000,000 이하입니다.
  2. s는 숫자와 '{', '}', ',' 로만 이루어져 있습니다.
  3. 숫자가 0으로 시작하는 경우는 없습니다.
  4. s는 항상 중복되는 원소가 없는 튜플을 올바르게 표현하고 있습니다.
  5. s가 표현하는 튜플의 원소는 1 이상 100,000 이하인 자연수입니다.
  6. 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):
    answer = []
    import re
    s = re.sub('^{{','',s)
    s = re.sub('}}$','',s)
    j = sorted(s.split('},{'), key = lambda x : len(x))
    for i in j:
        k = i.split(',')
        for l in k:
            if int(l) not in answer:
                answer.append(int(l))
    return answer

저는 정규표현식을 이용해 앞뒤에 {{, }}를 s에서 각각 제거해주었지만 간단하게 s[2:-2]를 사용해서 날려도 됬네요.. 역시 문자열의 특성을 이용하면 간단했는데 너무 어렵게 접근하지 않았나 싶네요. split와 sorted를 주로 이용해서 우선 길이가 짧은 순대로 나열해서 하나씩 answer 리스트에 포함시켜줍니다. 그래서 길이가 짧은 숫자부터 차례로 들어가 순서가 보전되면서도 중복되지 않는 수를 리스트에 담을 수 있습니다.