정렬 - 가장 큰 수

2019-11-10

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

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

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

제한 사항

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

워낙 많이 본 문제라 처음엔 별 생각없이 코드를 마구 써냈더니 시간 초과로 통과가 안됬다.

처음 코드

class Solution {
    public String solution(int[] numbers) {

        for(int i=0; i<numbers.length; i++){
            for(int j=1; j< numbers.length-i; j++){
                String iTemp = Integer.toString(numbers[i]) + Integer.toString(numbers[i+j]);
                String jTemp = Integer.toString(numbers[i+j]) + Integer.toString(numbers[i]);
                if(Integer.parseInt(iTemp) < Integer.parseInt(jTemp)){
                    int temp = numbers[i];
                    numbers[i] = numbers[i+j];
                    numbers[i+j] = temp;
                }
            }
        }
        String answer = "";
        for(int num : numbers){
            answer += Integer.toString(num);
        }
        return answer;
    }
}

int<->string 변환을 너무 많이 써서 시간 초과가 된 것 같다…

두번째로는 처음에 numbers를 string 배열로 바꾸고 시작한 후 comparator을 사용해 sort를 하기로 했다.

import java.util.*;

class Solution {

    public String solution(int[] numbers) {

        for(int i=0; i<numbers.length; i++){
            if(numbers[i] != 0) break;
            if(i == numbers.length-1) return "0";
        }

        String[] num = new String[numbers.length];
        String answer = "";

        for (int i = 0; i < num.length; i++) {
            num[i] = Integer.toString(numbers[i]);
        }

        Arrays.sort(num, new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                return (o2 + o1).compareTo(o1 + o2);
            }
        });


        for (String s : num) {
            answer += s;
        }
        return answer;
    }
}

마지막 케이스에서만 오류가 나서 꽤 오래 고민했는데 모든 수가 0일 경우를 고려하지 않아 발생한 오류였다.

이런 특수한 경우는 미리 조건에 공지를 해줘야 되는게 아닌가 싶다.