숫자 짝꿍 / 프로그래머스, 파이썬 코딩테스트 연습 연습문제

문제

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

 

제출답안

def solution(X, Y):
    def zip_num_count(str):
        """
        스트링 내 글자와 빈도수를 딕셔너리로 리턴하는 함수
        """
        dict_str = {}
        for n in str:
            try:
                dict_str[n] += 1
            except:
                dict_str[n] = 1
        return dict_str
    
    # X, Y를 {글자:카운트}로 변환
    dict_x, dict_y = zip_num_count(X), zip_num_count(Y)
    
    # X, Y에 공통으로 나오는 글자
    commons = dict_x.keys() & dict_y.keys()
    
    # Commons가 비지 않았을 때
    if commons:
        if commons == {'0'}: # Commons가 0만 있을 때
            answer = '0'
        else: # Commons가 0외에 다른 숫자도 있을 때
            answer = ''
            for c in commons:
                cnt = dict_x[c] if dict_x[c] <= dict_y[c] else dict_y[c] #최소 카운트
                answer += c*cnt
            answer = ''.join(sorted(answer, reverse=True))
    
    # Commons가 비었을 때
    else:
        answer = '-1'
            
    return answer

 

 

제출 후 개선답안

a. 딕셔너리 형태 대신 count 활용하여 글자 붙이는 과정 간소화
b. 결과물을 sort하는 대신 큰 수부터 체크하여 answer에 붙여가기

def solution(X, Y):
    answer = ''
    for i in range(9, -1, -1):
        i_str = str(i)
        answer += i_str * min(X.count(i_str), Y.count(i_str))
    
    if answer =='': answer = '-1'
    elif answer.replace('0', '') == '': answer = '0'
    
    return answer

 

카운트의 복잡도가 크다고 생각하여 
딕셔너리 자료형태로 접근하였었음.
하지만 생각보다 복잡도가 크지 않은듯함.

개선 전 / 개선 후

 

문제 바로가기