[Java/자바] 프로그래머스 Lv2 - 가장 큰 수 (정렬)

문제 설명

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"

 

---

Solution.java

이 문제는 Comparator의 compareTo() 메서드만 잘 활용한다면 쉽게 해결할 수 있는 문제였습니다.

public int compareTo(String string)
public int compareTo(Object object)

처음에는 복잡하게 생각했지만 (o2 + o1).compareTo(o1 + o2) 이런식으로 두 수의 조합을 기준으로 정렬을 할 수도 있었네요.. 또 하나 배울 수 있었습니다.

 

compareTo()를 이용해 두 문자열을 합친 값(o2 + o1)과 (o1 + o2) 중 값이 큰 값을 기준으로 내림차순 정렬하도록 합니다.

 

import java.util.Arrays;

public class 가장큰수 {
    public static void main(String[] args) {
        System.out.println(solution(new int[] {6, 10, 2}));
    }

    public static String solution(int[] numbers) {
        String[] str = new String[numbers.length];

        for(int i = 0; i < numbers.length; i++){
            str[i] = String.valueOf(numbers[i]);
        }

        // 두 수를 합친 값이 큰 순서대로 (내림차순)
        // o1 : 10, o2 : 2 이면 102과 210 중 뭐가 더 큰지 비교
        Arrays.sort(str, (o1, o2) -> (o2 + o1).compareTo(o1 + o2));

        // 첫 번째 수가 0이면 0으로만 이뤄진 배열이므로 0을 리턴.
        if (str[0].equals("0")) {
            return "0";
        }

        return String.join("", str);
    }
}