LeetCode刷题笔记：658.找到K个最接近的元素

1. 问题描述

给定一个 排序好 的数组 arr ，两个整数 k 和 x ，从数组中找到最靠近 x（两数之差最小）的 k 个数。返回的结果必须要是按升序排好的。

整数 a 比整数 b 更接近 x 需要满足：

• |a - x| < |b - x| 或者
• |a - x| == |b - x| 且 a < b

2. 解题思路

• 维护一个左指针 left 和一个右指针 right

• 从数组 arr 的两端分别与 x 做差并进行比较n - k次比较：

  • 若左指针指向的数与 x 的差值大于右指针指向的数与 x 的差值，则左指针右移
  • 若左指针指向的数与 x 的差值小于右指针指向的数与 x 的差值，则右指针左移

3. 代码实现

class Solution {
    public List<Integer> findClosestElements(int[] arr, int k, int x) {
        int n = arr.length;
        int left = 0;
        int right = n - 1;
        List<Integer> res = new ArrayList<>(k);
        for (int i = 0; i < n - k; i++) {
            if (Math.abs(arr[left] - x) > Math.abs(arr[right] - x)) {
                left++;
            } else {
                right--;
            }
        }
        for (int i = left; i < right + 1; i++) {
            res.add(arr[i]);
        }
        return res;
    }
}