ProblemSolving/String

프로그래머스 Level 2 튜플 (파이썬)

OSNIM 2022. 5. 29. 00:46
반응형

문자열, Level 2,  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]

 

첫번째 제출 코드 (정답)

from collections import defaultdict
def solution(s):
    answer = []
    temp = "".join(s.split("}"))
    temp = "".join(temp.split("{"))
    temp = temp.split(",")
    dic = defaultdict(int)
    for i in temp: #숫자만 세기
        dic[i] += 1

    #가장 많은 개수를 포함한 원소부터 찾아  answer에 넣기
    maxKey = max(dic, key=dic.get)
    n = dic[maxKey]
    for _ in range(n):
        maxKey = max(dic, key=dic.get)
        answer.append(int(maxKey))
        del dic[maxKey]

    return answer

문제를 이해하는데 약간 헤맸고 위 결과가 나오기까지 너무 다소 많은 시행착오가 있었습니다.

순서가 있는 튜플과 순서가 없는 집합의 차이를 처음에는 이해 못했는데 천천히 읽어보니 숫자의 개수가 많으면 먼저오는 것을 알아챘습니다.

그래서 각 집합별 숫자의 출현 개수를 세주어 많이 나오는 수 먼저 answer에 추가했습니다.

하지만 괄호와 콤마를 없애는 과정을 못찾아서 헤매다가 split()을 여러번 만져보니 숫자만 뺄 수 있었습니다.

 

그 후 저번 메뉴 리뉴얼 문제와 비슷하게 딕셔너리에서 값이 최대인 원소를 빼내는 방법인

maxKey = max(dic, key = dic.get)

maxKey = max(dic, key = dic.get)

이용해서 풀 수 있었습니다. 

 

좋아요를 가장 많이 받은 풀이 

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

 

정규 표현식 사용

def solution(s):
    answer = []

    s1 = s.lstrip('{').rstrip('}').split('},{')

    new_s = []
    for i in s1:
        new_s.append(i.split(','))

    new_s.sort(key = len)

    for i in new_s:
        for j in range(len(i)):
            if int(i[j]) not in answer:
                answer.append(int(i[j]))

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

 

 

반응형