[JAVA/프로그래머스] 가장 큰 수 / 정렬, 스트림

프로그래머스

문제 설명

문제 설명

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

 

제한 사항 

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

 

 

풀이 과정

처음엔 카운팅 정렬을 이용해보려고 했는데, 정렬을 이중으로 해야하고 조건이 까다로워져서 Stream과 Comparator를 이용하기로 했다.

import java.util.*;
import java.util.stream.*;

public String solution(int[] numbers) {
    String answer = Arrays.stream(numbers) // 1. IntStream으로 변환
            .boxed() // 2. Stream<Integer>로 변환
            .map(i -> i.toString()) // 3. Integer를 String으로 변환
            .sorted(((o1, o2) -> (o2 + o1).compareTo(o1 + o2))) // 4. 내림차순으로 정렬
            .collect(Collectors.joining()); // 5. 문자열로 결합

    return answer.charAt(0) == '0' ? "0" : answer; // 6. 결과 반환
}

1. Arrays.stream()을 이용해 int[] 배열을 intStream으로 변환한다
2. 기본 자료형을 참조 자료형으로 변환하기 위해 boxed()를 이용하며 intStream을 Stream<Integer>로 변환한다
3. Integer를 String으로 변환한다
4. 정렬을 한다
   • 문제 조건에 맞도록 6, 10, 2가 있을 때 62(6+2)가 610(6+10)보다 우선순위가 크다
   • 따라서 두 문자열을 합쳐서 새 문자열로 만들었을 때 우선순위가 더 큰 숫자를 앞에 오도록 내림차순 정렬한다 
5. 정렬된 데이터들로 문자열을 만든다
6. 결과를 반환한다
   • 입력으로 0만 여러개 주어진 경우 출력값이 "0000..."이 되어버리기 때문에 이를 삼항연산자를 이용해 처리한다

 

 

실행 결과