본문 바로가기
코딩테스트 연습

특이한 정렬

by 신재은👩🏼‍💻 2024. 5. 7.

https://school.programmers.co.kr/learn/courses/30/lessons/120880

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

이 건은 코드를 머리에 새기고 싶어서 기록한다.

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