[코테📑] 프로그래머스 - 특이한 정렬

📄 문제

정수 n을 기준으로 n과 가까운 수부터 정렬하려고 합니다. 이때 n으로부터의 거리가 같다면 더 큰 수를 앞에 오도록 배치합니다. 정수가 담긴 배열 numlist와 정수 n이 주어질 때 numlist의 원소를 n으로부터 가까운 순서대로 정렬한 배열을 return하도록 solution 함수를 완성해주세요.

🚫 제한사항

• 1 ≤ n ≤ 10,000
• 1 ≤ numlist의 원소 ≤ 10,000
• 1 ≤ numlist의 길이 ≤ 100
• numlist는 중복된 원소를 갖지 않습니다.

🟢 테스트 통과 코드

import java.util.*;

class Solution {
    public int[] solution(int[] numlist, int n) {
        return Arrays.stream(numlist)
                    .boxed()
                    .sorted((x, y) 
                        -> Math.abs(x-n)==Math.abs(y-n)? //n과의 거리가 같은 경우
                            x>y?-1:1 //큰쪽이 먼저 오게
                            :Math.abs(x-n)>Math.abs(y-n)?1:-1)
                            //n과의 거리가 다른 경우 n과의 거리순으로 오름차순 정렬
                    .mapToInt(Integer::valueOf)
                    .toArray();
    }
}

🧩 덧붙임

이 문제는 푼 지 꽤 오래됐는데 sorted의 비교식 부분을 흔히 쓰는 Comparator을 사용하지 않고 사용했던 예제였다. 처음 제출한 코드는 다음과 같다.

class Solution {
    public int[] solution(int[] numlist, int n) {
        List<Integer> numberlist = Arrays.stream(numlist)
                                        .boxed()
                                        .collect(Collectors.toList());

        return numberlist.stream()
                    .sorted((x, y) -> {
                        if(Math.abs(x-n)==Math.abs(y-n)) {
                            return x>y?-1:1;
                        } else {
                            return Math.abs(x-n)>Math.abs(y-n)?1:-1;
                        }
                    })
                    .mapToInt(Integer::valueOf)
                    .toArray();
    }
}

처음에는 비교식 부분을 if/else로 처리했었는데 삼항 연산자로 바꾸면 좀 더 간결해질 것 같아 수정했다. 또, 리스트도 어차피 또 스트림을 써서 정렬한 다음 배열로 변환해서 리턴 할 건데 리스트를 선언할 필요가 없을 것 같았다. 그래서 리스트를 선언하는 부분을 없애고 선언 부에서 스트림을 쓰는 부분과 리턴문에서 스트림을 쓰는 부분이랑 합친 다음 리턴하는 방향으로 수정했다. 스트림을 자유자재로 쓰고싶다..

[참고]

https://www.baeldung.com/java-8-sort-lambda

[출처]

프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges