프로그래머스 - 숫자 짝꿍(JAVA)

성혜영·2023년 6월 7일

프로그래머스

목록 보기

3/67

문제 설명

두 정수 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"

입출력 예 설명

입출력 예 #1

X, Y의 짝꿍은 존재하지 않습니다. 따라서 "-1"을 return합니다.

입출력 예 #2

X, Y의 공통된 숫자는 0으로만 구성되어 있기 때문에, 두 수의 짝꿍은 정수 0입니다. 따라서 "0"을 return합니다.

입출력 예 #3

X, Y의 짝꿍은 10이므로, "10"을 return합니다.

입출력 예 #4

X, Y의 짝꿍은 321입니다. 따라서 "321"을 return합니다.

입출력 예 #5

지문에 설명된 예시와 같습니다.

문제 풀이 과정

시간복잡도에 대해서 제일 많이 고려했었던 문제였다.

시간복잡도를 고려하지 않고 풀이를 하면 반이상 케이스에서 실패가 나온다.

for(String data : X.split("")){
    int cur = Integer.parseInt(data);
    xmap.put(cur, xmap.getOrDefault(cur, 0) + 1);
}
 
for(String data : Y.split("")){
    int cur = Integer.parseInt(data);
    ymap.put(cur, ymap.getOrDefault(cur, 0) + 1);
}

각 숫자가 몇개씩 들어있는지 카운트 하기 위해서 HashMap 을 이용하였다.

StringBuilder sb = new StringBuilder();
 
for(int i = 9; i >= 0 ; i --){
    if(xmap.containsKey(i) && ymap.containsKey(i)){
    int cnt = Math.min(xmap.get(i), ymap.get(i));
    for (int j = 0; j < cnt; j++) {
        sb.append(i);
        }
    }
}

여기서 포문이 제일 중요한 것 같다.
처음에는 포문에 keySet() 으로 했었는데 시간 초과가 된다.
숫자는 0 ~ 9 의 범위에 해당 되므로 포문을 0 ~ 9 범위로 진행하였다.
X 와 Y 에서 일치한 수가 있다면 cnt 가 적은 것을 선택하여 StringBuilder 에 넣어 주었다.

if(sb.toString().startsWith("0")){
    return "0";
}
else if(sb.toString().equals("")){
    return "-1";
}

StringBuilder 에 넣어 준 값이 00 이라도 0 이 나와야 하고 000 이라도 0 이 나와야 하기 때문에 startsWith("0") 조건을 넣어 주어 0 이 return 되도록 하였다.

또한 일치한 숫자가 없을 때는 -1 일 되어야 하므로 equals("") 을 사용하였다.

전체 코드

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

public class Solution {
    public static String solution(String X, String Y) {
        HashMap<Integer, Integer> xmap = new HashMap<>();
        HashMap<Integer, Integer> ymap = new HashMap<>();
        
        for(String data : X.split("")){
            int cur = Integer.parseInt(data);
            xmap.put(cur, xmap.getOrDefault(cur, 0) + 1);
        }

        for(String data : Y.split("")){
            int cur = Integer.parseInt(data);
            ymap.put(cur, ymap.getOrDefault(cur, 0) + 1);
        }

        StringBuilder sb = new StringBuilder();

        for(int i = 9; i >= 0 ; i --){
            if(xmap.containsKey(i) && ymap.containsKey(i)){
                int cnt = Math.min(xmap.get(i), ymap.get(i));
                for (int j = 0; j < cnt; j++) {
                    sb.append(i);
                }
            }
        }

        if(sb.toString().startsWith("0")){
            return "0";
        }
        else if(sb.toString().equals("")){
            return "-1";
        }

        return sb.toString();
    }
}

성혜영

개발자가 되고싶은 hye_ung

이전 포스트

프로그래머스 - 최소직사각형 (JAVA)

다음 포스트

프로그래머스 - 숫자 짝꿍(JAVA)