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

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

프로그래머스

📌 문제 설명

두 정수 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

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

---

🗒 풀이

def solution(X, Y):
    answer = []
    for i in (set(X) & set(Y)): # 교집합
        for j in range(min(X.count(i), Y.count(i))):
            answer.append(i)
    answer.sort(reverse=True)
    if len(answer) == 0:
        return '-1'
    if answer[0] == '0':
        return '0'
    return ''.join(answer)

일단 입력된 X와 Y의 교집합 원소를 구하고, 그 원소들을 토대로 for문을 순회한다.

다음 for문에서는 X와 Y에서 i의 값들을 count해서, 더 작은 값을 기준으로 answer 배열에 append한다.

더 작은 값을 기준으로 answer 배열에 append하는 것은 X-100, Y-123450인 경우, 둘의 교집합은 1과 0이고, X에게는 0이 두개지만, Y에서는 0이 하나이므로, 더 적은 Y의 0 개수를 기준으로 삼는다. 문제가 잘 이해되었으면 쉽게 이해할 수 있다.

다음엔 answer을 내림차순으로 정렬한다.

두번째 if문에서 0말고 다른 원소들이 있으면 첫번째 인덱스(0)의 원소가 0이 아니여야하는데(0이 가장 작은 값이므로), 첫 번째 원소가 0이라면 answer에는 0밖에 없으므로 0을 반환한다.

두 개의 조건을 다 통과했다면, answer 배열을 문자열로 반환한다.

---

🤔 후기

처음에 집합을 생각하지 못하고, 다음과 같이 작성했다.

def solution2(X, Y):
    answer = []
    Y = list(Y)
    for x in X:
        if x in Y:
            answer.append(x)
            Y.remove(x)

    if len(answer) == 1:
        return '-1'
    elif answer[0] == 0:
        return '0'
    else:
        return ''.join(sorted(answer, reverse=True))

 

문자열 Y를 리스트로 반환하고, 만약 Y에 x가 있으면, answer에 추가하고, 중복 제거를 위해 Y에서 삭제한다.

일단 Y가 x에 있는지 확인하는 부분과 remove 하는 쪽에서 연산 속도가 떨어질 수 밖에 없다.

때문에 처참한 시간초과를 얻었다.

다시 코드를 읽어보니, x와 y의 원소가 각각 들어있는지 확인을 하고 해당 원소를 제거할 이유가 있을까? 라는 생각이 들어서 교집합을 이용했고 덕분에 시간 초과를 면할 수 있었다.