https://school.programmers.co.kr/learn/courses/30/lessons/120880
이 건은 코드를 머리에 새기고 싶어서 기록한다.
import java.util.Arrays;
import java.util.Comparator;
class Solution {
public int[] solution(int[] numlist, int n) {
// Custom Comparator를 정의하여 정렬
Integer[] numlistInteger = Arrays.stream(numlist).boxed().toArray(Integer[]::new);
Arrays.sort(numlistInteger, new Comparator<Integer>() {
@Override
public int compare(Integer a, Integer b) {
int distA = Math.abs(a - n);
int distB = Math.abs(b - n);
if (distA == distB) {
return b - a; // 거리가 같으면 더 큰 수를 앞으로
}
return distA - distB; // 그 외에는 n과의 거리가 가까운 순으로 정렬
}
});
// Integer 배열을 int 배열로 변환하여 반환
return Arrays.stream(numlistInteger).mapToInt(Integer::intValue).toArray();
}
}
문제에 접근하는 방법이 파악되어도 코드 구현이 잘 안 될 때가 있다.
이번이 그랬다...
'코딩테스트 연습' 카테고리의 다른 글
크기가 작은 부분 문자열 (0) | 2024.05.08 |
---|---|
안전지대 (0) | 2024.05.07 |
외계어 사전 (0) | 2024.05.06 |
최대공약수와 최소공배수 (0) | 2024.05.06 |
같은 숫자는 싫어 (0) | 2024.05.06 |