프로그래머스 – 숫자 짝꿍 Python

프로그래머스 문제풀이

문제

문제설명

두 정수 XY의 임의의 자리에서 공통으로 나타나는 정수 k(0 ≤ k ≤ 9)들을 이용하여 만들 수 있는 가장 큰 정수를 두 수의 짝꿍이라 합니다(단, 공통으로 나타나는 정수 중 서로 짝지을 수 있는 숫자만 사용합니다). XY의 짝꿍이 존재하지 않으면, 짝꿍은 -1입니다. XY의 짝꿍이 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 한 개는 짝 지을 수 없습니다.)
두 정수 XY가 주어졌을 때, XY의 짝꿍을 return하는 solution 함수를 완성해주세요.

제한사항

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

입출력 예

XYresult
“100”“2345”“-1”
“100”“203045”“0”
“100”“123450”“10”
“12321”“42531”“321”
“5525”“1255”“552”

입출력 예 설명

입출력 예 #1

  • XY의 짝꿍은 존재하지 않습니다. 따라서 “-1″을 return합니다.

입출력 예 #2

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

입출력 예 #3

  • XY의 짝꿍은 10이므로, “10”을 return합니다.

입출력 예 #4

  • XY의 짝꿍은 321입니다. 따라서 “321”을 return합니다.

입출력 예 #5

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

문제해결과정

  1. x,y의 임의의 자리에서 공통으로 나타나는 정수들 구한다.
  2. 구해진 정수들로 만들 수 있는 가장 큰 정수값을 출력한다.
  3. 짝수없으면 -1, 공통수가 0으로만 구성되면 0 반환한다.
  4. 제한사항을 확인한다.

위 같은 문제 및 제한사항을 해석하고 문제풀이를 시작하면 됩니다.

나의 풀이

1차 실패 소스

실패원인 : 일부 테스트에서 시간초과 발생함

실패원인분석 : for문 이용으로 인해 입력값 길이가 길때, 시간소비가 많이 발생하는 것으로 추측 됐어요

def solution(X, Y):
    answer = ''
    X_List = list(X)
    Y_list = list(Y)
    Result_List = []
    
    #1.
    for v in X_List:
        if v in Y_List:
            Result_List.append(v)
            Y_List.remove(v)
    
    #2.
    Result_List.sort(reverse=True)

    #3.
    if len(Result_List) == 0:
        answer = "-1"
    else : 
        if len(Result_List) == Result_List.count("0"):
            answer="0"
        else:
            answer="".join(Result_List)
    return answer

2차 성공 소스

1차 실패로 인해서 소스 변경하였어요.

저는 dict를 사용하여 X,Y의 0~9까지 각각 구해서 둘 중 작은 값을 dict에 추가해줬어요.
그 후 cnt를 통해서 cnt – 0카운트가 0일때는 0카운트 유무에 따라 -1이나 0을 넣어주도록 코딩했습니다.

def solution(X, Y):
    answer = ''
    numdict = {"0":0,"1":0,"2":0,"3":0,"4":0,"5":0,"6":0,"7":0,"8":0,"9":0}

    for i in range(10):
        xcnt = X.count(str(i))
        ycnt = Y.count(str(i))
        numdict[str(i)] = min(xcnt, ycnt)

    cnt = 0
    for i in range(9,-1,-1):
        cnt += numdict[str(i)]
        answer += str(i)*numdict[str(i)]

    if cnt - numdict["0"] == 0:
        if numdict["0"]==0:
            answer = "-1"
        else:
            answer = "0"
    return answer

다른 사람의 풀이

아래와 같이 다른 분 좋은 풀이가 있어서 보면서 반성해보았어요.
dict는 쓰지않고 0~9 카운트를 리스트로 받았어도 됐었고, 저처럼 answer 결과로 -1과 0을 대입할때 cnt를 안쓰고
answer = “”로 검색을 한 뒤 0카운트 유무에 따라 0을 대입해줬어도 됐는데 다른 분 풀이보고 하나 또 배워갑니다.

def solution(X, Y):
    xList = list(X.count(str(x)) for x in range(10))
    yList = list(Y.count(str(y)) for y in range(10))
    answer = ""
    for i in range(9, -1, -1):
        answer += str(i) * min(xList[i], yList[i])

    if answer == "":
        return "-1"
    elif answer[0] == "0" and answer[len(answer) - 1] == "0":
        return "0"
    else:
        return answer