뒤죽박죽 코딩 이야기

2022년 09월 29일 목요일 - 어제 다비치 영접함!!

오늘 올려볼 문제는 658번 Find K Closest Elements 이라는 문제이다.

사진을 클릭하면 해당 문제로 이동합니다.

오늘도 LeetCode 사이트 오늘의 문제를 가지고 왔다.

오랜만에 올리는 미디엄 문제같네...

입력

사진에서도 볼 수 있듯이 정렬된 int 배열 1개와 int 값 2개가 입력으로 들어온다.

풀이 및 코드

x와 가장 가까운 원소들로만 이루어진 길이가 k인 subarray를 찾는 문제이다.

이 때 차이는 절대값으로 판별하고 절대값이 작을 때는 원소가 더 작은쪽을 택한다.

오늘은 처음부터 정답을 생각해냈다.

Queue에 원소들을 넣어두고 넣은 값들과 x 차이의 합을 구한다.

이후 window slide를 사용하여 합이 커지는 순간 직전까지 찾고 리턴하는 방식으로 구현했다.

이제 코드를 봐보자!

풀이코드

class Solution {
    public List<Integer> findClosestElements(int[] arr, int k, int x) {
        Queue<Integer> q = new LinkedList<>();
        int sum = 0;
        for(int i = 0; i < k; i++) {
            q.add(arr[i]);
            sum += Math.abs(arr[i] - x);
        }
        
        for(int i = k; i < arr.length; i++) {
            int num = Math.abs(arr[i] - x) - Math.abs(q.peek() - x);

            if(0 <= num && q.peek() < arr[i]) break;
            
            q.poll();
            q.add(arr[i]);
            sum += num;
            continue;
        }
        
        return (List)q;
    }
}

제출 화면

오늘 자바 서버가 별로 안좋은지 속도가 생각보다 안나온다.

내일도 문제를 풀어서 블로그에 글을 쓸 수 있으면 좋겠다.