공부/알고리즘
프로그래머스 가장 큰 수 자바 풀이(정렬02)
돌이토스
2021. 5. 12. 16:18
문제 설명
0 또는 양의 정수가 주어졌을 때, 정수를 이어 붙여 만들 수 있는 가장 큰 수를 알아내 주세요.
예를 들어, 주어진 정수가 [6, 10, 2]라면 [6102, 6210, 1062, 1026, 2610, 2106]를 만들 수 있고, 이중 가장 큰 수는 6210입니다.
0 또는 양의 정수가 담긴 배열 numbers가 매개변수로 주어질 때, 순서를 재배치하여 만들 수 있는 가장 큰 수를 문자열로 바꾸어 return 하도록 solution 함수를 작성해주세요.
제한 사항
- numbers의 길이는 1 이상 100,000 이하입니다.
- numbers의 원소는 0 이상 1,000 이하입니다.
- 정답이 너무 클 수 있으니 문자열로 바꾸어 return 합니다.
입출력 예
numbersreturn
[6, 10, 2] | "6210" |
[3, 30, 34, 5, 9] | "9534330" |
#풀이
import java.util.Arrays;
import java.util.Comparator;
class Solution {
public String solution(int[] numbers) {
String answer = "";
StringBuilder sb = new StringBuilder();
String[] num = new String[numbers.length];
for (int i = 0; i < num.length; i++) {
num[i] = String.valueOf(numbers[i]);
}
// 330, 303 배열 위해 o1 + o2 형태로 붙여서 비교
Arrays.sort(num, new Comparator<String>() {
@Override
public int compare(String o1, String o2) {
return (o2 + o1).compareTo(o1 + o2);
}
});
if (num[0].equals("0")) {
return "0";
}
for (int i = 0; i < num.length; i++) {
sb.append(num[i]);
}
answer = sb.toString();
return answer;
}
}
sort를 이용하면 되서 발상자체는 어렵지 않았다. 먼저 각 수를 String으로 바꾼 후 모두 sort하면 맨 앞자리 순으로 내림차순이 된다. 따라서 9가 30 보다 수 자체로 비교하면 더 작지만, String처리한 후 비교하면 더 큰것으로 여겨진다. 이때 문제점은 3과 30의 경우이다. 이 경우 String "30"이 String "3"보다 더 앞에 정렬된다. 그러나 만약 이 두 수를 붙이면 303 또는 330 의 경우가 있는데 배열이 내림차순 된 상태에서 차례대로 수들을 붙일경우 303이 채택되어 제일 큰 수를 만들지 못한다. 따라서 sort할 때 Comparator을 도입할 때 두 수를 붙여서 비교하면, 330 이 303보다 더 크게 되어, 3이 30보다 더 앞으로 정렬되게된다. 이 부분만 이해하고 넘어가면 그 후에는 간단했다.