[Level2] [Java] 가장 큰 수

문제

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"

 

문제 풀이

정렬을 어떻게 할 것인지 생각해야하는 문제입니다. 저 같은 경우 Comparator Interface를 이용해 정렬을 시도했습니다.

다른 사람의 풀이도 확인해봤는데 대부분이 저와 같은 방식으로 문제를 해결했습니다.

저는 개인적으로 Comparator보다 Comparable을 더 많이 사용해서 Comparator의 동작 과정을 찾아보면서 문제를 해결해 보았습니다.

 

이 문제에서 핵심 코드는 아래와 같습니다.

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

strArray = {"6", "10", "2"}; 이렇게 저장되어 있다고 가정했을때, (o2 + o1)을 println으로 콘솔에 찍히는 값을 확인해보면

610

102

102

62

이렇게 찍히는 것을 확인해 보실수 있을것입니다. 즉, 내부적으로 문자열을 합쳐서 가장 큰 수가되는 순으로 정렬이 되고 있습니다.

 

이렇게 코드를 작성하고 제출하게 되면 마지막 테스트케이스에서 실패가 뜰 것입니다.

그래서 한가지 더 추가해줘야하는 경우가 있습ㄴ디ㅏ.

바로 입력으로 주어진 배열값이 {0, 0, 0...) 모두 0으로 이루어진 경우입니다.

이 배열값을 compare() 메소드를 거치게 되면 "0000..."이런식의 값이 될것인데 모두 0인 경우, 우리가 출력해야 되는 답은 "0"입니다.

그렇기 때문에 맨 앞의 숫자가 0이 나오는 경우 무조건 "0"이 출력되고 return 시키면 모든 테스트케이스를 통과시킬수 있습니다.

 

Java Code

package com.programmers.java;

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

public class Solution {
    
    public static void main(String[] args) {
        Solution s = new Solution();
        
        int[] numbers = {6, 10, 2};
        
        s.solution(numbers);
    
    }
    
    public String solution(int[] numbers) {
        String answer = "";
        String[] strArray = new String[numbers.length];
        for (int i = 0; i < strArray.length; i++) {
            strArray[i] = Integer.toString(numbers[i]);
        }
        
        Arrays.sort(strArray, new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                
                return (o2 + o1).compareTo(o1 + o2);
            }
        });
        
        if (strArray[0].equals("0")) {
            return "0";
        }
    
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < strArray.length; i++) {
            sb.append(strArray[i]);
        }
        answer = sb.toString();
        
        return answer;
    }

}