双指针中等 LeetCode658. 找到 K 个最接近的元素

658. 找到 K 个最接近的元素

描述

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

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

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

示例 1：

输入：arr = [1,2,3,4,5], k = 4, x = 3
输出：[1,2,3,4]
示例 2：

输入：arr = [1,2,3,4,5], k = 4, x = -1
输出：[1,2,3,4]
 

提示：

1 <= k <= arr.length
1 <= arr.length <= 104
arr 按 升序 排列
-104 <= arr[i], x <= 104

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/find-k-closest-elements
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析

双指针排除不可能的元素。
原数组是升序的，所以一定有一端是需要排除的，不可能两端都符合条件，所以用双指针剔除掉与x距离更大的一端，直到剩余的元素个数是k个。

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