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

https://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"

 

---

풀이

풀이 과정

숫자들이 담긴 배열에서 마주한 두 수 a, b를 비교 했을 때 a+b와 b+a중 큰 값을 기준으로 배열을 정렬한다면 가장 큰 수를 얻어 낼 수 있다.

따라서 Comparator의 compare 메소드를 재정의 하여 a+b, b+a중 큰 값을 기준으로 정렬하도록 한다

String[] arr = new String[numbers.length];

for(int i = 0; i < numbers.length; i++){ // 문자열 배열로 변환
	arr[i] = String.valueOf(numbers[i]);
}

//문자열 내림차순 정렬, (a + b) 와 (b + a) 크기 비교
Arrays.sort(arr, new Comparator<String>(){
	@Override
	public int compare(String a, String b){
		return (b + a).compareTo(a + b);
	}
});

 

정렬된 배열을 문자열로 하나로 합쳐 반환을 하면 되는데, 이 때 numbers의 원소가 0 여러개로만 이루어진 경우 "0000"과 같은 답이 출력될 수 있기 때문에 배열의 가장 큰 값이 0일 경우에는 0을 return 하도록 한다

//가장 큰 값이 0이면 0을 리턴
if(arr[0].equals("0")) return "0";

StringBuilder sb = new StringBuilder();
//아니면 문자열을 더하여 리턴
for(String s : arr) sb.append(s);

return sb.toString();

 

최종 코드

import java.util.*;

class Solution {
    public String solution(int[] numbers) {
        String answer = "";
        String[] arr = new String[numbers.length];
        
        for(int i = 0; i < numbers.length; i++){
            arr[i] = String.valueOf(numbers[i]);
        }
        
        //문자열 내림차순 정렬, (a + b) 와 (b + a) 크기 비교
        Arrays.sort(arr, new Comparator<String>(){
            @Override
            public int compare(String a, String b){
                return (b + a).compareTo(a + b);
            }
        });
        
        //가장 큰 값이 0이면 0을 리턴
        if(arr[0].equals("0")) return "0";
        
        StringBuilder sb = new StringBuilder();
        //아니면 문자열을 더하여 리턴
        for(String s : arr) sb.append(s);
        
        return sb.toString();
    }
}

---