[코딩테스트] 프로그래머스 - 튜플 (Lv.2) in Python

원소의 개수가 n개이고, 중복되는 원소가 없는 튜플 (a1, a2, a3, ..., an)이 주어질 때(단, a1, a2, ..., an은 자연수), 이는 다음과 같이 집합 기호 '{', '}'를 이용해 표현 가능.

• {{a1}, {a1, a2}, {a1, a2, a3}, {a1, a2, a3, a4}, ... {a1, a2, a3, a4, ..., an}

특정 튜플을 표현하는 집합이 담긴 문자열 s가 매개변수로 주어질 때, s가 표현하는 튜플을 배열에 담아 return.

 

예시: 입력 - "{{2},{2,1},{2,1,3},{2,1,3,4}}",  리턴 - [2,1,3,4]

 

programmers.co.kr/learn/courses/30/lessons/64065

 

파이썬 코드 구현

def solution(s):
    updatedEl = []
    s = s.split('{')
    
    for i in s:
        new = []
        for j in i.split(','):
            if j != '':
                num =  ''.join(k for k in j if k in '0123456789')    
                new.append(int(num))     
        if len(new) > 0:
            updatedEl.append(new)

    updatedEl.sort(key = lambda x:len(x))
    
    answer = []
    for i in updatedEl:
        for j in i:
            if j not in answer:
                answer.append(j)
    return answer

 

풀이 과정

: 1. 주어지는 string 입력을, a list of (list of numbers) 로 변환

: 2. 원소의 순서는, number list 안의 순서를 따라갈 수밖에 없다.

- numList w/ 1 element ~> 그 안의 원소가 가장 앞에 오는 숫자. answers list에 append하자.

- numList w/ 2 elements ~> 안에 있는 원소 중 answer list에 이미 없는 원소를 찾아 append.

- ...

 

따라서, a list of number list를 list length로 sort하고,

(ex. 입력 - "{{2}, {2,1}}" -> [[2], [2,1]] -> [[2], [2,1]])

앞에서부터 원소를 순회하면 된다.

 

다른 분들의 풀이

- .split()을 잘 쓰는 것도 중요한 것 같다!

s.lstrip('{').rstrip('}').split('},{')

이 풀이는, 맨 왼쪽과 맨 오른쪽의 불필요한 {와 } 들을 없애고, 가운데는 },{로 split하면서 숫자만 골라낸다.

 

ex. s = "{{2},{2,1},{2,1,3},{2,1,3,4}}"일 때, 결과는 ['2','2,1','2,1,3','2,1,3,4' ]

 

- 정규표현식과 Counter를 이용한 멋진 풀이!

from collections import Counter
import re

def solution(s):

    s = Counter(re.findall('\d+', s))
    return list(map(int, [k for k, v in sorted(s.items(), key=lambda x: x[1], reverse=True)]))

가장 많이 등장하는 숫자 순서대로 담으면 된다는 점에 착안,

Counter 모듈을 이용해 숫자 별 등장 횟수를 담고, 내림차순으로 정렬 후,

배열에 담아 리턴한다.