[프로그래머스] 숫자 짝꿍 파이썬

문제 설명

두 정수 X, Y의 임의의 자리에서 공통으로 나타나는 정수 k(0 ≤ k ≤ 9)들을 이용하여 만들 수 있는 가장 큰 정수를 두 수의 짝꿍이라 합니다(단, 공통으로 나타나는 정수 중 서로 짝지을 수 있는 숫자만 사용합니다). X, Y의 짝꿍이 존재하지 않으면, 짝꿍은 -1입니다. X, Y의 짝꿍이 0으로만 구성되어 있다면, 짝꿍은 0입니다.

예를 들어, X = 3403이고 Y = 13203이라면, X와 Y의 짝꿍은 X와 Y에서 공통으로 나타나는 3, 0, 3으로 만들 수 있는 가장 큰 정수인 330입니다. 다른 예시로 X = 5525이고 Y = 1255이면 X와 Y의 짝꿍은 X와 Y에서 공통으로 나타나는 2, 5, 5로 만들 수 있는 가장 큰 정수인 552입니다(X에는 5가 3개, Y에는 5가 2개 나타나므로 남는 5 한 개는 짝 지을 수 없습니다.)
두 정수 X, Y가 주어졌을 때, X, Y의 짝꿍을 return하는 solution 함수를 완성해주세요.

제한사항

• 3 ≤ X, Y의 길이(자릿수) ≤ 3,000,000입니다.
• X, Y는 0으로 시작하지 않습니다.
• X, Y의 짝꿍은 상당히 큰 정수일 수 있으므로, 문자열로 반환합니다.

---

입출력 예

X	Y	result
"100"	"2345"	"-1"
"100"	"203045"	"0"
"100"	"123450"	"10"
"12321"	"42531"	"321"
"5525"	"1255"	"552"

---

입출력 예 설명

입출력 예 #1

• X, Y의 짝꿍은 존재하지 않습니다. 따라서 "-1"을 return합니다.

입출력 예 #2

• X, Y의 공통된 숫자는 0으로만 구성되어 있기 때문에, 두 수의 짝꿍은 정수 0입니다. 따라서 "0"을 return합니다.

입출력 예 #3

• X, Y의 짝꿍은 10이므로, "10"을 return합니다.

입출력 예 #4

• X, Y의 짝꿍은 321입니다. 따라서 "321"을 return합니다.

입출력 예 #5

• 지문에 설명된 예시와 같습니다.

제출 답안

• 처음에는 같은 자리수에 있는 숫자가 같아야지 맞는 건 줄 알았다.
• 그게 아니라 그냥 같은 숫자가 있는지가 핵심이다.
• Counter를 사용해서 각각 숫자의 개수를 담았다.
• 그 다음에 길이가 작은 것을 기준으로 해야 없는 숫자를 바로 날려버릴 수 있다.
  • 이게 min_len과 max_len이다.
• for loop를 돌면서 만약 숫자가 공통적으로 있다면 그 숫자가 적게 나온 것을 기준으로 num에 담는다.
• num을 다시 내림차순으로 정렬한다.
• 정렬 후 for loop를 돌면서 나온 횟수만큼 answer에 더한다.

from collections import Counter

def solution(X, Y):
    num = {}
    x_c = Counter(X)
    y_c = Counter(Y)
    if len(x_c) < len(y_c):
        min_len = x_c
        max_len = y_c
    else:
        min_len = y_c
        max_len = x_c
    for key, val in min_len.items():
        if key in max_len.keys():
            num[key] = min(val, max_len[key])
    if not num:
        return '-1'
    if len(num) == 1 and '0' in num.keys():
        return '0'
    answer = ''
    num = sorted(num.items(), key=lambda x:x[0], reverse=True)
    for key, val in num:
        answer += key * val
    return answer