[프로그래머스] Lv.1 숫자 짝꿍 (파이썬)

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의 짝꿍은 상당히 큰 정수일 수 있으므로, 문자열로 반환합니다.

---

나의 풀이

def solution(X, Y):
    temp = ''
    y_list = [j for j in Y]
    for i in X:
        if i in y_list:
            temp += i
            y_list.remove(i)
    if temp:
        if int(''.join(temp)) == 0:
            return "0"
        else:
            return ''.join(sorted(temp, reverse = True))
    else:
        return "-1"

 

• 답은 맞았는데, 시간 초과돼서 틀림
• 더 효율적으로 코드를 작성해야 됨 !

for i in X:
        if i in y_list:
            temp += i
            y_list.remove(i)

 

• 요 부분에서 시간복잡도가 O(n*m) 이 된다고 함! (계속 왔다갔다 반복해야돼서?)

 

다른 사람의 풀이

def solution(X, Y):
    x_count = [0]*10
    y_count = [0]*10
    for x in X:
        x_count[int(x)] += 1
    for y in Y:
        y_count[int(y)] += 1

    result = ''
    for i in range(9, -1, -1):
        result += min(x_count[i], y_count[i])*str(i)

    if len(result) == 0:
        return "-1"
    if result[0] == "0":
        return "0" 

    return result

 

• 우선 x_count, y_count를 10개로 제한을 시켜버려서 반복량 ↓
• result 만들 때도 깔끔하게 만들어짐 !

 

전에는 결과만 잘 나오면 정답이었는데 난이도가 높아지니까 효율성까지 고려해야 된다,,,!

비전공자에게는 너무 가혹해 ... :_: