[프로그래머스]특이한 정렬

특이한 정렬

문제 설명

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

---

제한사항

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

---

입출력 예

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

• 4에서 가까운 순으로 [4, 5, 3, 6, 2, 1]을 return합니다.
• 3과 5는 거리가 같으므로 더 큰 5가 앞에 와야 합니다.
• 2와 6은 거리가 같으므로 더 큰 6이 앞에 와야 합니다.

입출력 예 #2

• 30에서 가까운 순으로 [36, 40, 20, 47, 10, 6, 7000, 10000]을 return합니다.
• 20과 40은 거리가 같으므로 더 큰 40이 앞에 와야 합니다.

---

나의 풀이 코드

import java.util.*;
class Solution {
    public int[] solution(int[] numlist, int n) {
        //해시맵 (거리, 값)을 지정
        //거리에 따른 정렬
        HashMap<Double,Integer> map = new HashMap<>();
        for(int i=0; i<numlist.length; i++){
            double distance = Math.abs(n - numlist[i]);
            if(map.containsKey(distance)){
                if(map.get(distance) < numlist[i]){
                    map.put((distance)-0.5,numlist[i]);
                    }
                else{
                    map.put(distance+0.5,numlist[i]);
                }
            }
            else{
                map.put(distance,numlist[i]);
            }
        }
        
        List<Double> keylist = new ArrayList<>(map.keySet());
        Collections.sort(keylist);

        List<Integer> valuelist = new ArrayList<>();
        for (double key : keylist) {
            valuelist.add(map.get(key));
        }
        int[] answer = new int[valuelist.size()];
        for (int i = 0; i < valuelist.size(); i++) {
            answer[i] = valuelist.get(i);
        }
        return answer;
    }
}

해시맵 (거리, 값)을 지정후 키를 정렬후 키에 맞는 값 반환한다.

정렬로 풀으려다 시간 많이 소비했다. 정렬 정리하기

배운점

import java.util.Arrays;

class Solution {
    public int[] solution(int[] numList, int n) {
        return Arrays.stream(numList)
                .boxed()
                .sorted((a, b) -> Math.abs(a - n) == Math.abs(b - n)
                        ? b.compareTo(a) 
                        : Integer.compare(Math.abs(a - n), Math.abs(b - n)))
                .mapToInt(Integer::intValue)
                .toArray();
    }
}

다른사람이 스트림으로 푼 코드이다.

numList 배열을 스트림으로 변환하고,

.boxed() 기본 자료형을 해당하는 참조형으로 변환하여 스트림의 각 요소를 box로 감싸주고,

sorted 메서드를 사용하여 배열을 정렬한다.

• 이때 Comparator로 정렬 순서를 지정하는데
  두 수의 n과의 차이의 절댓값이 같은 경우, b.compareTo(a)를 통해 원래 값에 대한 내림차순으로 정렬하도록 한다.
• 아니면 두 요소의 거리를 비교하여 오름차순으로 정렬한다.

스트림의 각 요소를 정수로 변환 후,

정렬된 정수 배열을 반환한다.