[Python][프로그래머스] 64065_튜플

2019 카카오 개발자 겨울 인턴십

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

코딩테스트 연습 - 튜플

문제 설명

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

문제 예시와 같습니다.

입출력 예 #2

문제 예시와 같습니다.

입출력 예 #3

(111, 20)을 집합 기호를 이용해 표현하면 {{111}, {111,20}}이 되며, 이는 {{20,111},{111}}과 같습니다.

입출력 예 #4

(123)을 집합 기호를 이용해 표현하면 {{123}} 입니다.

입출력 예 #5

(3, 2, 4, 1)을 집합 기호를 이용해 표현하면 {{3},{3,2},{3,2,4},{3,2,4,1}}이 되며, 이는 {{4,2,3},{3},{2,3,4,1},{2,3}}과 같습니다.

---

튜플의 정의와 튜플을 통해 집합만드는 과정을 보여준 뒤, 집합을 input으로 받아 튜플을 구해보는 문제이다.

어려운 문제는 아닌데, input type이 string이라서 자료 처리가 까다로웠던 듯 하다.

 

[풀이순서]

1. string input을 배열로 만들고 원소의 개수 순으로 정렬

"{{4,2,3},{3},{2,3,4,1},{2,3}}" --> ['3', '2,3', '4,2,3', '2,3,4,1']

 

2. 1에서 만든 배열의 원소를 int type의 list로 변환

[[3], [2, 3], [4, 2, 3], [2, 3, 4, 1]]

 

3. 만들어진 배열을 순회하며 li[i] - li[i-1] 차집합을 구해 answer에 추가

 

# 튜플으로 집합을 만드는 방법이 있을 때, 집합을 통해 튜플 찾아내기
def solution(s):
    li = s[2:-2].split('},{')
    li = sorted(li, key = lambda x: len(x))
    
    
    for r in range(len(li)):
        li[r] = list(map(int, li[r].split(',')))
    
    answer = [li[0][0]]
    for r in range(1, len(li)):
        sub = set(li[r]) - set(li[r-1])
        print(list(sub)[0])
        answer.append(list(sub)[0])
        
    return answer

 

다른 사람의 풀이 참고 중 정규식과 Counter를 사용한 방법을 보게 되었는데, 입력받은 string input 상태에서 정규직으로 숫자만 꺼내어 이를 Counter로 갯수를 세어 빈도순으로 정렬하면 원소개수가 1개였을 때 부터 계속 존재했던 튜플의 첫번째 원소 -> 마지막 원소 순으로 빈도가 세어지는것을 이용하여 단 2줄로 구현하였다.

 

import re
from collections import Counter

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)]))

 

얼른 스스로 이런 멋진 풀이를 떠올릴 수 있게 되기를..!