[프로그래머스/Java] 정렬 - 가장 큰 수

문제

문제 출처

programmers.co.kr/learn/courses/30/lessons/42746

문제 설명

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"

 

풀이

1. 주어진 숫자 배열을 문자열 배열로 변경한다.
2. 배열의 두 문자열을 붙였을 때 더 큰 수가 되도록 정렬한다.
3. 배열이 오름차순으로 정렬되어 있으므로 마지막 원소부터 첫번째 원소까지 차례대로 덧붙여 가장 큰 수를 만든다.
4. 첫 번째 문자가 0인 경우 배열의 모든 수가 0인 경우이므로 가장 큰 수는 0이다.

3번에서 문자열을 연결할 때 StringBuilder를 사용해 append하면 실행시간이 더 짧아질 것이다.

import java.util.Arrays;
import java.util.Comparator;

class Solution {
    public String solution(int[] numbers) {
        String[] nums = new String[numbers.length];
        
        //문자열 배열로 변경
        for (int i=0;i<numbers.length;i++) {
            nums[i] = String.valueOf(numbers[i]);
        }
        
        //붙였을 때 더 큰 수가 되도록 정렬
        Arrays.sort(nums, new Comparator<String>() {
            @Override
            public int compare(String s1, String s2) {
                return (s1+s2).compareTo(s2+s1);
            }
        });
        
        //큰 수부터 차례대로 연결
        String answer = "";
        for (int i=numbers.length-1;i>-1;i--) {
            answer+=nums[i];
        }
        
        //첫 번째 문자가 0인 경우 "0" return
        return answer.charAt(0) == '0' ? "0" : answer;
    }
}