[프로그래머스 Level1] 숫자 짝꿍 - 파이썬(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 함수를 완성해주세요.

 

 

---

 이 문제는 많이 어렵진 않았지만 처음에는 시간 초과가 났다가 코드를 수정하여 AC를 받은 문제이다.

문제의 제한 조건에서 X, Y 의 길이(자릿수)가 3,000,000까지 될 수 있다고 말했지 때문이다. 그렇기에 코드를 구현할 때 아래의 두 가지 상황이 발생한다면 시간 초과가 나기 쉽다. 내 경우가 그랬다..

1. X, Y에 대해 이중 for 반복문을 사용할 경우.. O(N^2)
2. int형을 str형으로 or str형을 int형으로 바꿀 경우

 

2번의 주의해야할 사항을 간과하고 구현했다가 시간 초과가 일어난 코드는 다음과 같다.

def solution(X, Y):
    answer = 0
    listX = list(''.join(X))
    listY = list(''.join(Y))
    
    # listY에 있는 '0'~'9'까지의 개수 세기
    numsDic = {}
    for c in listY:
        if c in numsDic:
            numsDic[c] += 1
        else:
            numsDic[c] = 1

    # X, Y의 짝꿍 찾기
    listXY = list()
    for c in listX:
        if c in numsDic and numsDic[c] > 0:
            listXY.append(int(c))
            numsDic[c] -= 1
    if listXY == []:  # X, Y의 짝꿍이 존재하지 않는 경우
        return "-1"

    # X, Y의 짝꿍 숫자들로 만들 수 있는 가장 큰 정수 반환
    listXY.sort(reverse=True)
    for num in listXY:
        answer *= 10
        answer += num
    return str(answer)

짝꿍을 찾는 과정에서도 str형을 int형으로 변환했고, 마지만 return 과정에서도 int형을 str형으로 바꾸는 과정에서 시간 초과가 발생한 것이다.

 

그러한 시간 초과를 줄이기 위해 수정한 코드는 AC를 받을 수 있었다.

짝꿍 숫자들로 만들 수 있는 가장 큰 정수를 만들 때에 int형을 사용하지 않으므로 0으로만 여러 개 이루어진 answer값이 반환되지 않도록 예외는 추가로 처리한다.

 

 

<< 맞은 코드 >>

def solution(X, Y):
    answer = ""
    listX = list(''.join(X))
    listY = list(''.join(Y))
    
    # listY에 있는 '0'~'9'까지의 개수 세기
    numsDic = {}
    for c in listY:
        if c in numsDic:
            numsDic[c] += 1
        else:
            numsDic[c] = 1

    # X, Y의 짝꿍 찾기
    listXY = list()
    for c in listX:
        if c in numsDic and numsDic[c] > 0:
            listXY.append(c)
            numsDic[c] -= 1
    if listXY == []:  # X, Y의 짝꿍이 존재하지 않는 경우
        return "-1"

    # X, Y의 짝꿍 숫자들로 만들 수 있는 가장 큰 정수 반환
    listXY.sort(reverse=True)
    for v in listXY:
        answer += v
    if answer[0] == "0" and "00" in answer:  # 짝꿍 숫자들이 0으로 이루어진 예외 처리
        return "0"
    return answer