문제설명
정수 n을 기준으로 n과 가까운 수부터 정렬하려고 합니다. 이때 n으로부터의 거리가 같다면 더 큰 수를 앞에 오도록 배치합니다. 정수가 담긴 배열 numlist와 정수 n이 주어질 때 numlist의 원소를 n으로부터 가까운 순서대로 정렬한 배열을 return하도록 solution 함수를 완성해주세요.
제한사항
입출력 예
numlist | n | result |
---|---|---|
[1, 2, 3, 4, 5, 6] | 4 | [4, 5, 3, 6, 2, 1] |
[10000,20,36,47,40,6,10,7000] | 30 | [36, 40, 20, 47, 10, 6, 7000, 10000] |
입출력 예 설명
입출력 예 #1
입출력 예 #2
풀이
스트림이 간단할 것 같아 스트림을 사용하여 풀이 했다.
n과의 차이가 같은 숫자의 경우 더 큰숫자가 앞으로 와야 하므로,
내림차순으로 배열을 먼저 정렬한 뒤,
다시 절대값이 더 작은 값들 순으로 정렬 하였다.
import java.util.*;
class Solution{
public int[] solution(int[] numlist, int n) {
return Arrays.stream(numlist)
.boxed()
.sorted(Comparator.reverseOrder())
.sorted((o1, o2) -> Math.abs(o1-n) - Math.abs(o2-n))
.mapToInt(Integer::intValue)
.toArray();
}
}
참고