Programmers - 숫자 짝꿍

프로그래머스

 

 

문제

 

두 정수 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 함수를 완성해주세요.

 

 

접근방식

X와 Y가 문자열로 주어지니까 먼저 문자열을 하나씩 잘라서 각각의 리스트에 저장함

그리고 같은 문자열을 가지는 배열을 빈배열로 선언하고 x_arr의 원소들을 돌면서 그 원소가 y_arr에도 있으면 같은 문자열을 담는 배열에 원소를 저장하고 중복해서 담기는 것을 방지하기 위해 remove로 원소를 지워줌

가장 큰 값을 출력해야 하므로 역순으로 정렬하고 빈 배열의 경우 '-1'을, 배열이 0으로만 있을 경우 맨 앞의 원소가 0이므로 비교를 통해 '0'을, 그 외의 경우 join 함수를 써서 문자를 문자열로 합쳐서 리턴함

 

하지만 이렇게 풀게 될 경우 tc에서 시간초과가 뜸

검색해보니 두 배열을 비교하면서 내부에서 remove 하는 과정에서 연산속도가 많이 떨어짐을 알게 되었음

 

두 문자열에서 비교를 하는 것이니, 제거하지 않고 일단 set함수와 & 연산자를 통해 두 문자열이 교집합으로 가지는 문자를 탐색하고 그 문자가 문자열 X와 Y 중 길이가 짧은 곳에서 몇 번 나타나는지 for문을 통해 탐색 후 그만큼 새로운 배열에 그 문자열을 더해줬음

 

그 후 리턴 부분은 그대로 작성했더니 시간초과를 피할 수 있었다.

 

코드

def solution(X, Y):
    answer = ''
    same_arr = []
    x_arr = [x for x in X]
    y_arr = [y for y in Y]
    for z in x_arr:
        if z in y_arr:
            same_arr.append(z)
            y_arr.remove(z)
    same_arr.sort(reverse=True)
    
    if same_arr == []:
        return '-1'
    elif same_arr[0] == '0':
        return '0'
    else:
        return ''.join(same_arr)

 

 

수정 코드

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 answer == []:
        return '-1'
    elif answer[0] == '0':
        return '0'
    else:
        return ''.join(answer)