[Java] 숫자 짝꿍

문제

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

 

import java.util.HashMap;
import java.util.ArrayList;
import java.util.Collections;

class Solution {
    public String solution(String X, String Y) {
        StringBuilder answer = new StringBuilder();
        ArrayList<Character> numList = new ArrayList<Character>();
        HashMap<Character, Integer> xMap = new HashMap<Character, Integer>();
        for(int m = 0; m < X.length(); m++) {
            xMap.put(X.charAt(m), xMap.getOrDefault(X.charAt(m), 0) + 1);
        }
        // System.out.println(xMap);

        for(int i = 0; i < Y.length(); i++) {
            char ch = Y.charAt(i);
            if(xMap.get(ch) != null && xMap.get(ch) != 0) {
                xMap.put(ch, xMap.get(ch) - 1);
                numList.add(ch);
            }
        }
        // System.out.println(xMap);
        // System.out.println(numList);
        Collections.sort(numList);

        for(int i = numList.size()-1; i >= 0; i--) {
            answer.append(numList.get(i));
            if(i == numList.size()-1 && numList.get(i) == '0') {
                break;
            }
        }
        if(answer.length() == 0) {
            answer.append("-1");
        }

        return answer.toString();
    }
}

이거 은근히 어려운 문제였다.

처음에 생각한 방식은 이러했다.

 

x와 y중 가장 짧은 문자열이 무엇인지를 찾는다.

그 짧은 문자열을 기준으로 문자를 하나씩 꺼내왔고

그 꺼내온 문자를 가지고 x에도 포함되고 y에도 포함되면 공통된 문자이니

해당 문자는 x와 y에 각각 replaceFirst(문자, "")를 하였다. 왜냐하면 100 과 1230 같은 경우, 0이 두번이나 공통된 번호라고 판단할 수 있기 때문에 문자를 하나씩 지웠다고 보면 된다.

그리고 공통된 문자를 정렬을 하고

뒤에서부터 하나씩 꺼내오는 방식이였다.

 

말로 보니까 많은 연산이 필요하긴 했구나 싶었다.

아니나다를까 시간초과가 났다.

그래서 다시 생각해낸 방법이 위 코드이다.

 

x문자열을 map에 <문자, 그 갯수>로 나눠 담는다.

y문자열을 하나씩 꺼내와서 map에 있고 그 갯수(value)가 0이 아니라면(꺼내올 수 있는 문자가 하나 이상 존재한다면) 꺼내온다.

그리고 정렬 후

0이 맨 뒤에 있다면 0보다 큰 값이 없다는 뜻이므로 결과는 0이 되고

빈 문자열이면 "-1"을 반환하고

뭐 그런식으로 남은 조건을 해결했다.

 

이렇게 해결하니 좀 아슬아슬하게 통과했다.

 

조금 부끄럽지만 아래와 같은 결과가 나왔다.

살짝 좀 괜히 억울한데...

원래 800~900ms 정도 왔다갔다 했는데 지금 노트북이 좀 무리가 있는지 결국 천을 찍었다.

 

 

출처

https://school.programmers.co.kr/learn/courses/30/lessons/131128