[Python] 카카오 2019 INTERN 튜플

Coding Test

목록 보기
12/14

0️⃣ 문제 설명

문제 설명은 좀 길지만 그냥 입출력 보면 어느 정도 이해가 될 것이다.

셀수있는 수량의 순서있는 열거 또는 어떤 순서를 따르는 요소들의 모음을 튜플(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 이하인 경우만 입력으로 주어집니다.

[입출력 예]

sresult
"{{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️⃣ 빈도 순 정렬

  • 내 풀이
from collections import defaultdict

def solution(s):
    nums = defaultdict(int)
    cur = ''
    
    for item in s:
        if item in ['{', '}', ',']:
            if cur: 
                nums[int(cur)] +=1
                cur = ''
            continue
            
        cur += item
    
    nums = sorted(nums.keys(), key = lambda x:nums[x], reverse = True)
    
    return tuple(nums)

⚠️ dict는 sort() 메서드가 없다. 정렬하려면 sorted(dict명, key=..., reverse=...) 형태로 써야 한다.
😯 파이썬의 튜플을 반환해야하는 줄 알았는데 실제 요구되는 반환값은 리스트였다.


2️⃣ 그 외 풀이

1. 길이 순 정렬

내 풀이랑은 다른 방식‼️

def solution(s):
    # 겉의 {{ }} 제거 후 "},{ " 기준으로 split
    s = s[2:-2].split("},{")
    
    # 각 집합을 길이 기준으로 정렬
    s = sorted(s, key=len)

    answer = []
    seen = set()
    
    for group in s:
        nums = group.split(',')
        for num in nums:
            n = int(num)
            if n not in seen:
                seen.add(n)
                answer.append(n)
    
    return answer

문자열에서 각 집합을 추출 → 집합의 길이(원소 개수) 순으로 정렬 → 짧은 집합부터 순서대로 보면서 처음 등장하는 숫자만 결과에 추가


2. Counter

내 풀이와 비슷하지만 다름

from collections import Counter
import re

def solution(s):
    # 정규식으로 숫자만 추출
    nums = re.findall(r'\d+', s)
    nums = list(map(int, nums))
    
    counter = Counter(nums)
    
    # 출현 횟수 기준 내림차순 정렬
    answer = [num for num, _ in counter.most_common()]
    
    return answer

문자열에서 정규식으로 숫자만 전부 추출 → 숫자만 추출한 것으로 카운터 만듦 → 등장 횟수가 많은 순 : 튜플 순서

정규식 쓰기 싫으면 아래처럼.

from collections import Counter

def solution(s):
    cur = ""
    nums = []

    for ch in s:
        if ch.isdigit():
            cur += ch
        else:
            if cur:
                nums.append(int(cur))
                cur = ""

    if cur:
        nums.append(int(cur))

    counter = Counter(nums)

    return [num for num, _ in counter.most_common()]

튜플을 반환하고 싶다면 return tuple(num for num, _ in counter.most_common())


Counter의 메서드

🔥 most_common([n])

등장 횟수가 많은 순서대로 반환. (요소, 개수) 튜플 리스트를 반환.

from collections import Counter
c = Counter('ababc')

print(c.most_common())
# [('a', 2), ('b', 2), ('c', 1)]

print(c.most_common(1))
# [('a', 2)]

그 외에 elements(), update(), subtract() 등이 있다.
또, Counter는 +, -, &, | 과 같은 연산자를 지원한다. (교집합: min값, 합집합: max값을 한꺼번에 구할 수 있음.)

& 연산 (교집합: min 값)

Counter(a=3, b=1) & Counter(a=1, b=2)
# Counter({'a': 1, 'b': 1})

| 연산 (합집합: max 값)

Counter(a=3, b=1) | Counter(a=1, b=2)
# Counter({'a': 3, 'b': 2})
profile
학습 메모장 : 코테 및 알고리즘, 언어 문법, Java 기본 강의...

0개의 댓글