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

프로그래머스의 Lv.2 가장 큰 수 문제 풀이

문제 설명

0 또는 양의 정수가 주어졌을 때, 정수를 이어 붙여 만들 수 있는 가장 큰 수를 알아내 주세요.
예를 들어, 주어진 정수가 [6, 10, 2]라면 [6102, 6210, 1062, 1026, 2610, 2106]를 만들 수 있고, 이중 가장 큰 수는 6210입니다.
0 또는 양의 정수가 담긴 배열 numbers가 매개변수로 주어질 때, 순서를 재배치하여 만들 수 있는 가장 큰 수를 문자열로 바꾸어 return 하도록 solution 함수를 작성해주세요.

제한 사항

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

문제 풀이
1. 현재 주어진 숫자 배열을 문자열 배열로 변환
2. Comparator를 사용하여 문자열 정렬(내림차순)
2-1. 문자열 이어 붙이기
3. 정렬된 문자열을 하나의 문자열로 결합
4. 000이나 0000의 같은 수를 0으로 변환(앞자리가 0이면 0으로 변환)

코드

public String solution(int[] numbers) {
    String[] numStr = Arrays.stream(numbers)
            .mapToObj(String::valueOf)
            .toArray(String[]::new);

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

    String result = String.join("", numStr);
    if (result.charAt(0) == '0') {
      return "0";
    }
    return result;
  }