[프로그래머스 / Python파이썬] 튜플

문제 링크: https://programmers.co.kr/learn/courses/30/lessons/64065

코딩테스트 연습 - 튜플

 

 

문제 설명

튜플 (a1, a2, a3, a4)를 표현하는 집합을 원소의 개수가 작은 순서대로 정렬하면 {{a1}, {a1, a2}, {a1, a2, a3}, {a1, a2, a3, a4}}가 됩니다. 이때 집합에서 튜플을 구하는 방법은, {ai} = s[i] - s[i - 1] 입니다. (편의상 인덱스를 1부터 카운트했습니다. ) 예를 들어 i는 3일 때, s[3] - s[2] = {a1, a2, a3} - {a1, a2} = {a3}입니다. 

 

따라서 주어진 s의 원소들을 리스트에 담아서 원소(집합)의 개수가 작은 순서대로 정렬합니다. {ai} = s[i] - s[i - 1]을 이용해서 튜플의 원소를 앞에서부터 구합니다. 

문자열 형태를 리스트로 변환하기 위해서 ast.literal_eval()을 사용했습니다. 

 

 

 

 

소스코드

import ast

def solution(s):
    answer = []
    s = "[" + s[1:-1] + "]"
    s = ast.literal_eval(s)         # s의 원소들을 리스트로 변환하여 저장
    s.sort(key=lambda x: len(x))    # 원소의 개수가 작은 순서대로 s 정렬
    
    answer.append(list(s[0])[0])
    for i in range(0, len(s) - 1):
        answer.append(list(s[i + 1] - s[i])[0])

    return answer