[Python] 프로그래머스 - 튜플

문제

셀수있는 수량의 순서있는 열거 또는 어떤 순서를 따르는 요소들의 모음을 튜플(tuple)이라고 합니다. n개의 요소를 가진 튜플을 n-튜플(n-tuple)이라고 하며, 다음과 같이 표현할 수 있습니다.

(a1, a2, a3, ..., an)

튜플은 다음과 같은 성질을 가지고 있습니다.

중복된 원소가 있을 수 있습니다. ex : (2, 3, 1, 2) 원소에 정해진 순서가 있으며, 원소의 순서가 다르면 서로 다른 튜플입니다. ex : (1, 2, 3) ≠ (1, 3, 2) 튜플의 원소 개수는 유한합니다.

원소의 개수가 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 이하인 경우만 입력으로 주어집니다.

문제가 길어서 복잡해 보이지만 간단한 문제입니다.

문자열안에 주어진 집합들을 토대로 하나의 튜플을 반환하는 문제입니다.

집합에는 순서가 없고, 튜플에는 순서가 있다는 점을 유념하고 풀어야합니다.

주어진 집합내에서 튜플의 순서를 찾으려면 가장 원소 수가 적은 집합부터 조회하면 됩니다.

문제에서 주어진 조건 상 하나의 원소를 가진 집합은 튜플의 가장 첫번째 원소가 됩니다.

이후 두개의 원소를 가진 집합을 찾는 다면 첫번째 원소의 위치를 이미 알고 있으므로, 자연스럽게 두번째 원소의 위치도 정해집니다. 이런 방식으로 풀어나가면 됩니다.

풀이

1. 원소 수를 계산한 리스트를 저장합니다.
2. 원소 수 리스트의 가장 최소값부터 조회합니다.
3. 최솟값 즉, 원소 수가 가장 적은 집합의 원소들을 그대로 정답 배열에 저장합니다.
4. 이후 그 다음으로 최솟값을 조회합니다. 이때 이미 정답 배열에 있는 요소들을 제거하고, 남은 원소들을 정답 배열에 저장합니다. 이 과정에서 원소들이 중복으로 저장될 위험이 없어집니다.

import re
import sys


def solution(s):
    # 주어진 문자열의 집합들을 리스트로 변환
    s_tuples = [list(map(int, x.split(','))) for x in re.findall(r'\{([\d,]+)\}', s)]

    # 각 리스트의 원소 수
    count = []
    for i in range(len(s_tuples)):
        count.append(len(s_tuples[i]))

    answer = []

    for i in range(len(count)):
        # 가장 원소 수가 작은 리스트 부터 정답 리스트에 저장
        min_idx = count.index(min(count))
        # 원소 수가 가장 작은 리스트를 조회 후 값을 가장 큰값으로 변경 -> 추후 min 연산 시 배제됨
        count[min_idx] = sys.maxsize
        # 원소 수가 가장 작은 리스트
        s_tuple = s_tuples[min_idx]
        # 리스트에서 정답 리스트에 있는 요소들은 삭제
        for ans in answer:
            if ans in s_tuple:
                s_tuple.remove(ans)
        for s in s_tuple:
            answer.append(s)


    return answer

 

파이썬에서 map과 lamda식을 사용하여 간단하게 해결 할 수 있습니다.

• Counter를 사용해 이 숫자들이 몇 번 등장하는지를 세고, 각 숫자와 빈도수가 저장된 딕셔너리 형태의 객체를 만듭니다.

• s.items()로 숫자와 그 빈도수를 튜플로 가져오고, 이를 빈도수(x[1]) 기준으로 내림차순 정렬합니다.

• 빈도수 기준으로 정렬된 숫자들을 꺼내 int로 변환해 리스트로 반환합니다.

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

import re
from collections import Counter