공부/알고리즘

프로그래머스 가장 큰 수 자바 풀이(정렬02)

돌이토스 2021. 5. 12. 16:18

문제 설명

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

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

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

제한 사항

  • numbers의 길이는 1 이상 100,000 이하입니다.
  • numbers의 원소는 0 이상 1,000 이하입니다.
  • 정답이 너무 클 수 있으니 문자열로 바꾸어 return 합니다.

입출력 예

numbersreturn

[6, 10, 2] "6210"
[3, 30, 34, 5, 9] "9534330"

 

 

#풀이


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

class Solution {

		public String solution(int[] numbers) {
			String answer = "";
			StringBuilder sb = new StringBuilder();

			String[] num = new String[numbers.length];
			for (int i = 0; i < num.length; i++) {
				num[i] = String.valueOf(numbers[i]);

			}

			// 330, 303 배열 위해 o1 + o2 형태로 붙여서 비교
			Arrays.sort(num, new Comparator<String>() {

				@Override
				public int compare(String o1, String o2) {
					return (o2 + o1).compareTo(o1 + o2);
				}

			});
            
			if (num[0].equals("0")) {
				return "0";
			}
            
			for (int i = 0; i < num.length; i++) {
				sb.append(num[i]);
			}
			answer = sb.toString();

			return answer;
		}

	}

sort를 이용하면 되서 발상자체는 어렵지 않았다. 먼저 각 수를 String으로 바꾼 후 모두 sort하면 맨 앞자리 순으로 내림차순이 된다. 따라서 9가 30 보다 수 자체로 비교하면 더 작지만, String처리한 후 비교하면 더 큰것으로 여겨진다. 이때 문제점은 3과 30의 경우이다. 이 경우 String "30"이 String "3"보다 더 앞에 정렬된다. 그러나 만약 이 두 수를 붙이면 303 또는 330 의 경우가 있는데 배열이 내림차순 된 상태에서 차례대로 수들을 붙일경우 303이 채택되어 제일 큰 수를 만들지 못한다. 따라서 sort할 때 Comparator을 도입할 때 두 수를 붙여서 비교하면, 330 이 303보다 더 크게 되어, 3이 30보다 더 앞으로 정렬되게된다. 이 부분만 이해하고 넘어가면 그 후에는 간단했다.