알고리즘/프로그래머스

[프로그래머스] 튜플 (Python)

AquaplaneMode 2022. 12. 23. 14:32

1. 문제

I. 문제 설명

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

II. 제한 사항

  • s의 길이는 5 이상 1,000,000 이하입니다.
  • s는 숫자와 '{', '}', ',' 로만 이루어져 있습니다.
  • 숫자가 0으로 시작하는 경우는 없습니다.
  • s는 항상 중복되는 원소가 없는 튜플을 올바르게 표현하고 있습니다.
  • s가 표현하는 튜플의 원소는 1 이상 100,000 이하인 자연수입니다.
  • return 하는 배열의 길이가 1 이상 500 이하인 경우만 입력으로 주어집니다.

    III. 입출력 예

    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]

2. 풀이

I. 풀이 방법

  • i번째 set은 a0 부터 ai-1까지의 숫자를 담고있기 때문에, i+1번째 set에서 i번째 set을 빼어, 새로이 추가된 숫자를 파악.

II. 소스 코드

def str2list(s):

    # 괄호 제거
    s = s[2:-2].split("},{") 

    # set 묶음
    s = [set(map(int,x.split(","))) for x in s]

    # 길이 순 정렬
    s = sorted(s, key = lambda x : len(x))

    return s
  • str2list(s)
    input이 string으로 주어지기 때문에 string을 list로 우선 변환해줘야 한다.
    • 최초 input : {{2},{2,1},{2,1,3},{2,1,3,4}}
    • 괄호 제거 후 : [["2"], ["2","1"], ["2","1","3"], ["2","1","3","4"]]
    • set로 묶은 후 : [{2}, {1,2}, {1,2,3}, {1,2,3,4}]
    • 길이 순으로 정렬한 후 : [{2}, {1,2}, {1,2,3}, {1,2,3,4}]
def solution(s):
    answer = []

    s = str2list(s)
    init = set()

    for set_ in s:
        ele = (set_-init)
        init = set_
        answer.append(ele.pop())

    return answer
  • solution
    list가 된 input을 순서대로 탐색하여, i번째 set에서 i-1번째 set을 빼 새로이 추가된 숫자를 찾아 정답에 추가한다.