알고리즘
[프로그래머스] 가장 큰 수
DevBee
2021. 4. 10. 16:22
programmers.co.kr/learn/courses/30/lessons/42746
코딩테스트 연습 - 가장 큰 수
0 또는 양의 정수가 주어졌을 때, 정수를 이어 붙여 만들 수 있는 가장 큰 수를 알아내 주세요. 예를 들어, 주어진 정수가 [6, 10, 2]라면 [6102, 6210, 1062, 1026, 2610, 2106]를 만들 수 있고, 이중 가장 큰
programmers.co.kr
이 문제는 주어진 정수를 먼저 문자열로 바꾼 뒤, 문자열을 정렬하는 방법을 새로 정의하여 문제를 해결하였습니다. 문자열의 경우 자릿수로 크기를 비교하기 때문에 ["6", "10", "2"] 가 주어지는 경우 내림차순으로 정렬했을 때 ["6", "2", "10"] 이 됩니다. 따라서 문자열로 정수를 바꿔 정렬하면 문제를 해결할 수 있는데 이때 확인이 필요한 경우가 있습니다.
"1"과 "10" 이 주어지는 경우 내림차순으로 정렬할 때 ["10", "1"] 이렇게 정렬이 됩니다. 이 두 문자를 합치는 경우 "101"이 되는데 두 문자의 순서를 바꿔 "110"으로 합치는 경우가 더 큰 수가 됩니다. 따라서 문자열을 정렬할 떄 두문자의 합인 "101"과 "110"을 비교하여 정렬하도록 조건을 주어 문제를 해결하였습니다.
자바 코드는 다음과 같습니다.
import java.util.Arrays;
import java.util.Comparator;
class Solution {
public String solution(int[] numbers) {
// 숫자를 문자로 변경하여 배열을 만듭니다.
String[] numStr = new String[numbers.length];
for (int i = 0; i < numbers.length; i++) {
numStr[i] = String.valueOf(numbers[i]);
}
// 정렬할 때 (뒷문자 o2 + 앞문자 o1)와 (앞문자 o1 + 뒷문자 o2)를 비교하여 정렬
// 예) "10",, "1" 이 주어지는 경우 "110"과 "101" 중 "110"이 더 크기 때문에 "1", "10" 순으로 정렬
Comparator<String> comp = (o1, o2) -> (o2 + o1).compareTo(o1 + o2);
Arrays.sort(numStr, comp);
// 모든 수가 0인 경우는 만들 수 있는 수가 0이므로 0 반환
if ("0".equals(numStr[0])) {
return "0";
}
// 정렬된 문자열을 합쳐서 반환
StringBuffer answer = new StringBuffer();
for (String str : numStr) {
answer.append(str);
}
return answer.toString();
}
}