숫자 짝꿍

문제 설명

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

입출력 예시

X	Y	result
"100"	"2345"	"-1"
"100"	"203045"	"0"
"100"	"123450"	"10"
"12321"	"42531"	"321"
"5525"	"1255"	"552"

풀이

처음에는, 가장 큰 수를 구해야하므로, X와 Y를 내림차순으로 정렬하고, 정렬된 배열을 기준으로 X와 Y의 교집합을 filter하고, 빈 배열인 경우, “-1” 배열의 원소가 0만 있으면, “0” 을 반환하도록 구현했다.

테스트 케이스에서는 성공했으나, 제출했을 때는 실패 및 런타임 에러를 만나게 되었다. 실패한 코드는 아래와 같다.

JavaScript

function solution(X, Y) {
    var answer = '';
    let temp = []
    let xArray = X.split('').sort((a,b)=> b-a)
    let yArray = Y.split('').sort((a,b)=> b-a)
    temp = yArray.filter(x=> xArray.includes(x))
    if(temp.length < 1) return '-1'
    answer = temp.join('')
    if(Number(answer) === 0) return '0'
    return answer;
}

아래와 같은 방법으로 해결할 수 있었다.

0부터 9까지의 빈도수를 세기 위해, 길이가 10인 배열을 선언한다
반복문을 통해, 각 숫자가 몇 번 나왔는지를 계산한다.
결과를 담을 빈 배열을 선언한다.
0부터 9까지 반복하면서, 0이 아니면서, 최소 빈도를 가지는 수를 구해 배열에 추가한다.
배열이 빈 경우, “-1”을 반환한다.
가장 큰 값을 만들어야 하므로, 배열을 내림차순으로 정렬한다.
배열이 0으로만 구성된 경우 “0”을 반환한다.
문자열로 반환해야 하므로, join()으로 변환한 문자열을 반환한다.

function solution(X, Y) {
    let freqX = new Array(10).fill(0);
    let freqY = new Array(10).fill(0);
    
    for (let i=0; i<X.length; i++) {
        freqX[+X[i]]++;
    }
    
    for (let i=0 ; i<Y.length; i++) {
        freqY[+Y[i]]++;
    }
    
    let resultDigits = [];
    
    for (let digit = 0; digit < 10; digit++) {
        if (freqX[digit] > 0 && freqY[digit] > 0) {
            let minFreq = Math.min(freqX[digit], freqY[digit]);
            for (let i = 0; i < minFreq; i++) {
                resultDigits.push(digit);
            }
        }
    }
    
    if (resultDigits.length === 0) {
        return "-1";
    }
    
    resultDigits.sort((a, b) => b - a);
    
    if (resultDigits[0] === 0 && resultDigits.length === resultDigits.filter(d => d === 0).length) {
        return "0";
    }
    
     return resultDigits.join('');
}

테스트는 통과했지만, 일부 케이스의 시간이 오래 걸리는 것을 확인하고 이를 개선했다.

모든 숫자를 배열로 모아서 정렬하는 대신 결과 문자열을 숫자의 내림차순으로 직접 구성 ⇒ 정렬 과정 제거
가장 높은 숫자(9)부터 0까지 시작하여 결과 문자열에 직접 추가되므로 나중에 추가 문자열 조작 및 조건 방지
문자열 작성 프로세스 중에 검사를 통합하여 불필요한 배열에서 문자열로의 변환이나 추가 루프가 없는지 확인

function solution(X, Y) {
    let freqX = new Array(10).fill(0);
    let freqY = new Array(10).fill(0);
    
    for (let i=0; i<X.length; i++) {
        freqX[+X[i]]++;
    }
    
    for (let i=0 ; i<Y.length; i++) {
        freqY[+Y[i]]++;
    }
    
    let answer = "";
    let hasNonZero = false;
    
    for(let digit = 9; digit >= 0; digit -- ) {
        if (freqX[digit] > 0 && freqY[digit] > 0) {
            const count = Math.min(freqX[digit], freqY[digit]);
            const segment = digit.toString().repeat(count);
            answer += segment;
            if(digit > 0 ) hasNonZero = true
        }
    }
    
    if (answer.length === 0) {
        return "-1";
    }else if(!hasNonZero && answer[0] === '0'){
        return "0"
    }else{
        return answer;
    }
}

수정한 함수를 Python으로 구현하면 아래와 같다.

Python

def solution(X, Y):
    freqX = [0] * 10
    freqY = [0] * 10

    for digit in X:
        freqX[int(digit)] += 1
    
    for digit in Y:
        freqY[int(digit)] += 1

    result = []
    has_non_zero = False

    for digit in range(9, -1, -1):
        min_count = min(freqX[digit], freqY[digit])
        if min_count > 0:
            result.append(str(digit) * min_count)
            if digit > 0:
                has_non_zero = True

    if not result:
        return "-1"
    if not has_non_zero:
        return "0"  

    return ''.join(result)

다른 풀이

def solution(X, Y):
    answer = ''

    for i in range(9,-1,-1) :
        answer += (str(i) * min(X.count(str(i)), Y.count(str(i))))

    if answer == '' :
        return '-1'
    elif len(answer) == answer.count('0'):
        return '0'
    else :
        return answer