[알고리즘] 튜플 (프로그래머스 2019 Kakao Winter Internship)

튜플 (2019 카카오 개발자 겨울 인턴십)

코딩테스트 연습 - 튜플

문제 설명

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

 

 

💡나의 문제 풀이

문제 해결 아이디어

• 튜플을 집합으로 표현할 때, 항상 맨 앞의 원소 a1 는 항상 포함된다. a2 는 a3 부터 마지막 튜플까지 모두 포함되어 있다. 이를 바탕으로, 먼저 입력되는 집합 문자열을 리스트 배열로 변경하고, 길이순으로 정렬하여, 마지막 튜플에서 바로 앞의 튜플에 있는 원소들을 모두 제거하는 방식으로 마지막 an 부터 역순으로 찾아가는 아이디어를 떠올렸다.
• 문자열 s를 어떻게 자를 것이냐가 관건이다.
  • 먼저 strip() 함수를 이용해 양 옆의 { 와 } 를 제거하고 , 를 공백으로 replace 하고 이를 바탕으로 split 하여 문자열 토큰들을 얻고, 길이 순으로 배열에 집어넣는다.
  • 각 배열 내의 문자열은, , 를 기준으로 split 하여 정수형으로 변환해 다시 저장하도록 하자.

작성 코드

def solution(s):
    # 집합 문자열을 리스트로 변경 후 길이 짧은 순으로 정렬
    arr = list(sorted(s.replace('{', '').replace('},', ' ').replace('}', '').split(), key=lambda x:len(x)))
    # 리스트의 각 요소를 추출
    for i in range(len(arr)):
        arr[i] = list(int(a) for a in arr[i].split(','))
    # 다음 요소에 있는 값을 모두 제거하기
    for i in range(len(arr)-1, 0, -1): 
        tmp = [j for j in arr[i] if j in arr[i-1]]
        for t in tmp:
            arr[i].remove(t)
    return [a[0] for a in arr]

 

답안 코드

• 튜플을 표현한 집합에 등장하는 정수의 개수가 많은 순이 바로 튜플의 순서라는 아이디어를 이용해 구현한 코드
  • re 모듈의 findall() 함수를 이용해 정수 리스트를 각각 얻고, 각 숫자가 몇 번 사용되었는지를 Counter() 함수를 이용해 얻는다.
  • Counter() 로 얻은 dictionary의 key 값들(튜플의 원소)을 value(문자열에 등장한 횟수)를 기준으로 역순으로 정렬한 리스트를 리턴
  • 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)]))
• 위에서 re 를 사용하지 않은 것 제외하고는 동일

from collections import Counter

def solution(s):
    new_s = [sss.replace('{','').replace('}','') for sss in s.split(',')]
    return [int(c[0]) for c in sorted(Counter(new_s).items(), key = lambda x: x[1],reverse=True )]

 

리뷰

• 문자열에서 정수가 등장한 횟수를 기준으로 각 요소를 정렬함으로써 튜플을 얻을 수 있다는 아이디어를 발견하지 못한 것이 아쉽다.