[프로그래머스] - 가장 큰 수

 

문제 설명

0 또는 양의 정수가 주어졌을 때, 정수를 이어 붙여 만들 수 있는 가장 큰 수를 알아내 주세요.

예를 들어, 주어진 정수가 [6, 10, 2]라면 [6102, 6210, 1062, 1026, 2610, 2106]를 만들 수 있고, 이중 가장 큰 수는 6210입니다.

0 또는 양의 정수가 담긴 배열 numbers가 매개변수로 주어질 때, 순서를 재배치하여 만들 수 있는 가장 큰 수를 문자열로 바꾸어 return 하도록 solution 함수를 작성해주세요.

 

 

 

6, 10, 2 을 넣게 되면 6210 이 출력되어야 한다.

 

 

 

 

 

숫자에서 정렬을 시도하면 엉뚱한 값이 나올 수 있으므로 문자로 변환후, 정렬해봐야 알수가 있다.

 

1. 이중 for문 사용

 

// 숫자 -> 문자 -> 내림차순정렬 -> 조합
        String[] strNums = new String[numbers.length];

        for(int i = 0; i < numbers.length ; i++) {
            strNums[i] = "" + numbers[i];
        }

        for(int i = 0 ; i < strNums.length-1 ; i++) {
            for(int j = i+1 ; j < strNums.length ; j++) {
                String s1 = strNums[i];
                String s2 = strNums[j];

                if((s1+s2).compareTo(s2+s1) < 0) {
                    strNums[i] = strNums[j];
                    strNums[j] = s1;
                }
            }
        }

        String answer = "";
        for (String s: strNums) {
            answer += s;
        }

        if(answer.charAt(0) == '0') return "0";
        return answer;

 

결론 적으로는 시간 제한 에러가 떠서 조금 더 빠른 정렬을 해야 할것같다.

 

두번쨰 방법으로는 자바에서 제공해주는 Arrays.sort를 사용하는 것이다.

 

import java.util.Arrays;

public class MaxNumber {

    public String solution(int[] numbers) {
        // 숫자 -> 문자 -> 내림차순정렬 -> 조합
        String[] strNums = new String[numbers.length];

        for(int i = 0; i < numbers.length ; i++) {
            strNums[i] = "" + numbers[i];
        }

        Arrays.sort(strNums, (s1,s2) -> (s2+s1).compareTo(s1+s2));

        String answer = "";
        for (String s: strNums) {
            answer += s;
        }

        if(answer.startsWith("0")) return "0";
        return answer;
    }

}

 

조금 더 코드를 간결하기 위해 stream을 사용하면 밑에와 같이 간결하게 해결할 수 있다.

 

import java.util.stream.Collectors;
import java.util.stream.IntStream;

public class MaxNumber {

    public String solution(int[] numbers) {
        // 숫자 -> 문자 -> 내림차순정렬 -> 조합
        String answer = IntStream.of(numbers)
            .mapToObj(String::valueOf)
            .sorted((s1,s2) -> (s2+s1).compareTo(s1+s2))
            .collect(Collectors.joining());

        if(answer.startsWith("0")) return "0";
        return answer;
    }

}