[프로그래머스] 숫자 짝궁 -Java

▤ 목차

1. 문제설명

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

프로그래머스

문제 설명
두 정수 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
지문에 설명된 예시와 같습니다.

2. 문제풀이

나의 풀이(시간초과)

import java.util.*;
import java.util.stream.*;
class Solution {
    public String solution(String X, String Y) {
        String answer ="";
        HashMap<String, Integer> array1 = new HashMap<>();
        HashMap<String, Integer> array2 = new HashMap<>();
        List<String> list = new ArrayList<>();
        
        for(String key: X.split("")) {
            array1.put(key, array1.getOrDefault(key, 0)+1);
        }
        
        for(String key: Y.split("")) {
            array2.put(key, array2.getOrDefault(key, 0)+1);
        }
        
        for(String key: array1.keySet()) {
            if(!array2.containsKey(key)){
                continue;
            } 
            
        int length = Math.min(array1.get(key), array2.get(key));
            for(int i = 0; i < length; i++) {
                list.add(key);
            }
        }
        Collections.sort(list, Collections.reverseOrder());
        if (list.size() == 0) {
            return "-1";
        }else if (list.get(0).equals("0")) {
            return "0";
        }else{
           for(int i=0; i<list.size(); i++){
             answer += list.get(i);   
           } 
             
        }
      
        
        return answer;
    }
}

위에 코드는 정답은 맞지만 시간초과가 되어 시간을 단축할 방법을 찾아야 했다.

핵심은 x , y 가 3,000,000까지 인 것을 단축할 방법을 찾아보는 것이다.

 

나의 풀이 2

import java.util.*;
class Solution {
    public String solution(String X, String Y) {
        HashMap<String, Integer> array1 = new HashMap<>();
        HashMap<String, Integer> array2 = new HashMap<>();
        List<String> list = new ArrayList<>();
        
        for(String key: X.split("")) {
            array1.put(key, array1.getOrDefault(key, 0)+1);
        }
        
        for(String key: Y.split("")) {
            array2.put(key, array2.getOrDefault(key, 0)+1);
        }
        
        for(String key: array1.keySet()) {
            if(!array2.containsKey(key)){
                continue;
            } 
            
        int length = Math.min(array1.get(key), array2.get(key));
            for(int i = 0; i < length; i++) {
                list.add(key);
            }
        }
        if (list.size() == 0) {
            return "-1";
        }
        Collections.sort(list, Collections.reverseOrder());
        if (list.get(0).equals("0")) {
            return "0";
        }
        
        StringBuilder result = new StringBuilder(list.size());
        for (String s : list) {
            result.append(s);
        }
        
        return result.toString();
    }
}

해설

1. X, Y를 쪼개서 HashMap으로 넣어줄 것이다. 중복되는 숫자는 숫자만큼 value 값에 추가된다. (ex 0=2, 1=1...)
2. array1과 array2의 키값을 비교하여 arrayList에 넣어준다. (length는 x, y 중 길이가 작은 것만큼 for문을 돈다.)
3. list의 size가 0일 때는 -1, 내림차순 정렬 후 첫번째 값이 0일때는 0을 return 한다.
4. 3,000,000까지 빠르게 돌기 위해 StringBuilder를 통해 return값을 추출한다.

다른 풀이

class Solution {
    public String solution(String X, String Y) {
        StringBuilder answer = new StringBuilder();
        int[] x = {0,0,0,0,0,0,0,0,0,0};
        int[] y = {0,0,0,0,0,0,0,0,0,0};
        for(int i=0; i<X.length();i++){
           x[X.charAt(i)-48] += 1;
        }
        for(int i=0; i<Y.length();i++){
           y[Y.charAt(i)-48] += 1;
        }

        for(int i=9; i >= 0; i--){
            for(int j=0; j<Math.min(x[i],y[i]); j++){
                answer.append(i);
            }
        }
        if("".equals(answer.toString())){
           return "-1";
        }else if(answer.toString().charAt(0)==48){
           return "0";
        }else {
            return answer.toString();
        }
    }
}

잘못된 내용이 있다면 지적부탁드립니다. 방문해주셔서 감사합니다.