본문 바로가기
프로그래머스/코딩테스트 고득점 kit

프로그래머스 정렬 LV2 - 가장 큰 수

by 맴썰 2021. 8. 25.

문제 설명

0 또는 양의 정수가 주어졌을 때, 정수를 이어 붙여 만들 수 있는 가장 큰 수를 알아내 주세요.

예를 들어, 주어진 정수가 [6, 10, 2]라면 [6102, 6210, 1062, 1026, 2610, 2106]를 만들 수 있고, 이중 가장 큰 수는 6210입니다.

0 또는 양의 정수가 담긴 배열 numbers가 매개변수로 주어질 때, 순서를 재배치하여 만들 수 있는 가장 큰 수를 문자열로 바꾸어 return 하도록 solution 함수를 작성해주세요.

제한 사항

  • numbers의 길이는 1 이상 100,000 이하입니다.
  • numbers의 원소는 0 이상 1,000 이하입니다.
  • 정답이 너무 클 수 있으니 문자열로 바꾸어 return 합니다.

입출력 예

numbers                                                                              return

[6, 10, 2] "6210"
[3, 30, 34, 5, 9] "9534330"

 

내 풀이

import java.util.*;
class Solution {
    public String solution(int[] numbers) {
        String answer = "";
        String[] temp = new String[numbers.length];
      
        for(int i=0;i<numbers.length;i++){
            temp[i] = Integer.toString(numbers[i]);
        }
        Arrays.sort(temp,new Comparator<String>(){
            @Override
            public int compare(String a, String b){
                return (b.concat(a)).compareTo(a.concat(b));
            }
        });
       
        for(int i=0;i<numbers.length;i++){
            answer = answer.concat(temp[i]);
        }
        String last = answer.replace("0","");
        if(last.length()==0) answer = "0";
        return answer;
    }
}

맨 처음에 풀 당시에는 Comparator를 쓰지 않고 String 배열을 정렬하였는데, 30과 3을 정렬할 때 303으로 정렬하여서 실패했었다. 그래서 Comparator를 사용하여 두 개의 String을 앞뒤를 바꾸어 연결한 것들 중 compareTo 함수를 통해 사전적으로 더 큰 것을 찾아서 return해 주도록 했다. 이 문제에서 사전적으로 더 큰 값은 아스키코드 상의 더 큰 값을 의미한다. 그렇게 정렬을 하고, answer 문자열을 만들고 제출했는데 1개의 테스트 케이스에서 오류가 발생했다.

만약 테스트 케이스가 [0,0,0,0,0] 으로 주어진다면 내 코드는 "00000"을 제출할 것인데, 정답은 "0"이므로 차이가 발생한다. 그래서 정답배열의 모든 "0"을 공백으로 바꿨을 때 그 길이가 0이 된다면, 해당 문자열은 "0"이라고 선언해준 후 제출하니 통과되었다.