LeetCode-658-找到K个最接近的元素

LeetCode-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

class Solution {
    /**
     * 二分查找 + 双指针;
     * 先通过二分查找找到最接近 x 的最小下标, 然后通过双指针移动, 找到最靠近 x 的区间范围
     */
    public List<Integer> findClosestElements(int[] arr, int k, int x) {
        int index = binarySearch(arr, x);
        int left = index, right = index + 1;
        while (k-- > 0) {
            if (right >= arr.length || (left >= 0 && x - arr[left] <= arr[right] - x)) {
                left--;
            } else {
                right++;
            }
        }
        List<Integer> res = new ArrayList<>(k);
        for (int i = left + 1; i < right; i++) {
            res.add(arr[i]);
        }
        return res;
    }

    /**
     * 找到等于 target 的最小下标， 如果没有则找到小于 target 的最小下标
     */
    private int binarySearch(int[] arr, int target) {
        int left = -1, right = arr.length - 1;
        while (left < right) {
            int mid = (right - left + 1) / 2 + left;
            if (target >= arr[mid]) {
                left = mid;
            } else {
                right = mid - 1;
            }
        }
        return left;
    }

    public static void main(String[] args) {
        System.out.println(new Solution().findClosestElements(
                new int[]{1, 2, 3, 4, 5},
                4, -1
        ));
    }
}