coding test

[파이썬] 튜플

잔망루피 2021. 5. 5. 21:06

문제 설명

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

 

 

👩 나의 풀이

 

def solution(s):
    dic={}	# key=숫자, 값=갯수
    temp=""
    for i in s :
        if i.isdigit() :    # 숫자면
            temp+=i
        elif temp : 
            if temp in dic :
                dic[temp]+=1
            else :
                dic[temp]=1
            temp=''		# 비우기
        
    return list(map(int, sorted(dic, reverse=True, key=lambda x : dic[x])))

 

temp에 숫자만 추가했다.

숫자가 아닌 문자를 만나면 temp가 비어있지 않을 때 딕셔너리에 추가했다.

각 숫자의 갯수를 세서 많은 순으로 반환했다.

 

 

👩‍🦳 다른 사람 풀이

 

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

 

우와 정규표현식을 쓰면 이렇게 간단하고 아름다운 풀이가 나온다👏

나도 최근에 공부했는데 생각을 못했네😥😥

\d 숫자를 의미

+는 1번 이상 반복

정리하면 1자리수 이상인 숫자를 findall로 찾고 Counter로 갯수를 센다.

items로 key와 값 쌍을 뽑아내고 값을 기준으로 내림차순 정렬한다.

정렬된 값을 반복문으로 뽑아내고 키 k를 int로 변환하고 리스트에 담아 반환한다. 

마지막 부분이 복잡해서 아래에 내가 고쳐본 코드다.

 

# 다른 사람 풀이를 내가 좀 고친 코드
import re
from collections import Counter

def solution(s) :
    s=Counter(re.findall('\d+', s))
    return list(map(int, sorted(s, key=lambda x:s[x], reverse=True)))

 

딕셔너리 값을 정렬하고 key를 반환해야할 때 key=lambda x:s[x] 이 부분은 앞으로도 자주 써먹을 것 같다🙂

 

 

 

문제 출처 💁‍♀️ 프로그래머스

반응형

'coding test' 카테고리의 다른 글

[파이썬] 순위 검색  (0) 2021.05.07
[파이썬] [1차] 뉴스 클러스터링  (0) 2021.05.06
[파이썬] 메뉴 리뉴얼  (0) 2021.05.05
[파이썬] 실패율  (0) 2021.05.04
[파이썬] 오픈채팅방  (0) 2021.05.04