[프로그래머스] 가장 큰 수 (Java)

programmers.co.kr/learn/courses/30/lessons/42746

코딩테스트 연습 - 가장 큰 수

문제 설명

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"

풀이

1. 숫자 → 문자열

String.valueOf(숫자) 를 사용하여 String 타입으로 변환할 수 있다.

 

2. 내림차순 정렬

람다식, compareTo 사용

Collections.sort(strNumbers, (a, b) -> (b + a).compareTo(a + b));

 

A.compareTo(B)

A와 B가 같으면 0 리턴

A가 B보다 크면 양수 리턴

A가 B보다 작으면 음수 리턴

맨 첫자리부터 차례대로 비교

 

3. numbers가 0으로만 주어지는 경우 예외 처리

4. 정렬된 문자열을 answer에 담아 리턴

 

코드

import java.util.*;

class Solution {
    public String solution(int[] numbers) {
        String answer = "";
        // int -> String 배열로
        ArrayList<String> strNumbers = new ArrayList();
        for(Integer num : numbers){
            strNumbers.add(String.valueOf(num));
        }
        // 내림차순 정렬
        Collections.sort(strNumbers, (a, b) -> (b + a).compareTo(a + b));
        // numbers가 [0,0,0] 등으로 주어지는 경우는 "000"이 아닌 "0" 리턴
        if(strNumbers.get(0).startsWith("0")) return "0";
        // answer에 정렬된 것을 담기
        for(String s : strNumbers){
            answer += s;
        }
        return answer;
    }
}