[프로그래머스] 숫자 짝꿍(Python)

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/131128

프로그래머스

소스 코드

from collections import Counter

def solution(X, Y):
    # 숫자 개수 세기
    nums = Counter(X) & Counter(Y)
    if not nums: return '-1' # 공통 없는 경우
    elif list(nums) == ['0']: return '0' # 0만 공통인 경우
    
    nums_order = sorted(list(nums),reverse=True) # 내림차순 정렬
    answer = ''
    for num in nums_order:
        answer += num * nums[num]
    return answer

해설

• Counter 함수를 이용한 풀이
• 내림차순 정렬

 

Counter 함수 활용

Counter에 리스트나 문자열을 입력으로 주게 되면 그 안에 포함된 원소의 개수를 세어 dictionary와 유사한 Counter 객체를 반환한다.

예를 들어 [1,2,2,3,3,3] 라는 리스트가 있을 때 이를 Counter의 입력으로 주면 Counter({3: 3, 2: 2, 1: 1})가 반환된다.

이때 dictionary와 마찬가지로 key로 접근할 수 있다.

 

따라서 문자열 X, Y에 포함된 글자의 개수를 Counter로 세고 교집합 기능(&)을 이용하여 중복을 찾아낸다.

1) 중복이 없는 경우엔 -1을 반환한다.

2) 중복이 0으로 유일한 경우엔 0을 반환한다.

3) 그 외의 경우 큰 숫자부터 Counter에 key로 접근하여 개수만큼 정답 문자열에 추가하면 된다.