코딩 테스트

프로그래머스) 숫자 짝꿍 Lv1

우루쾅 2024. 2. 15. 01:29
728x90
반응형
SMALL

문제

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

 

풀이

각각 String X 와 String Y 가 주어젔을 때 List 형식으로 배열을 선언하여 뽑아냈다.

그 뒤에 X와 Y 를 비교하면서 새로만든 List 배열에 같은 값이 있을 경우 추가를 하고,

Y 리스트를 삭제하면서 중복을 없애나갔다!

 

그 뒤로 짝꿍이 없을 경우 -1 을 리턴!

짝꿍이 있다면 큰 숫자를 위로 가게끔 정렬을 한 뒤 

만약 0만 있다면 0을 리턴하는 방식으로 문제를 해결했다~!

... 고 생각했으나 제출을 해보니 타임아웃

import java.util.*;

class Solution {
    public String solution(String X, String Y) {
        List<String> xVal = new ArrayList<>(Arrays.asList(X.split("")));
        List<String> yVal = new ArrayList<>(Arrays.asList(Y.split("")));
        List<String> common = new ArrayList<>();
        
        for(int i=0; i<xVal.size(); i++){
            if(yVal.contains(xVal.get(i))){
                common.add(xVal.get(i));
                yVal.remove(xVal.get(i));
            }
        }
        
        if(common.size() == 0) return "-1";
        
        Collections.sort(common, Collections.reverseOrder());
        String answer = String.join("", common);
        
        if(answer.replaceAll("0", "").length() == 0) return "0";
        
        return answer;
    }
}

 

확인을 해보니 contains와 remove 가 생각보다 자원을 많이 잡아먹었고

이 때문에 시간초과가 발생한 것으로 보였다.

 

위 방식대로 진행하려면 contains와 remove 를 배제시킬 수 없었기 때문에 아예 다시 만들었다...

 

여기저기 다른 사람들의 문제 풀이를 확인해봤는데 배열 10개를 만들어놓고

같은 숫자가 확인될때마다 해당 배열의 자리에 1씩 더해주고 이렇게 생성된 x 배열과 y 배열을 비교하여

문제를 해결하는 방식으로 해결했다!

 

이 방식은 문제를 풀 때 생각도 못했던 방식인데...

문제를 더 많이 풀어보면 나두 이런 방식으로 접근이 가능해질 날이 올까

갈길이 멀다~

 

 

 


정답

import java.util.*;

class Solution {
    public String solution(String X, String Y) {
        int[] xx = new int[10];
        int[] yy = new int[10];
        StringBuilder answer = new StringBuilder();
        
        for(String x : X.split("")){
            xx[Integer.parseInt(x)]++;
        }
        
        for(String y : Y.split("")){
            yy[Integer.parseInt(y)]++;
        }
        
        for(int i=xx.length-1; i>=0; i--){
            while(xx[i] > 0 && yy[i] > 0){
                answer.append(Integer.toString(i));
                xx[i]--;
                yy[i]--;
            }
        }
        
        if(answer.length() == 0) return "-1";
        if("".equals(answer.toString().replaceAll("0",""))) return "0";
        
        return answer.toString();
    }
}

 

 

출처

프로그래머스 - https://school.programmers.co.kr/learn/courses/30/lessons/131128

728x90
반응형
LIST