leetcode 力扣 658 找到 k 个最接近的元素

题目：

给定一个排序好的数组，两个整数 k 和 x，从数组中找到最靠近 x（两数之差最小）的 k 个数。返回的结果必须要是按升序排好的。如果有两个数与 x 的差值一样，优先选择数值较小的那个数。

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

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

说明:

• k 的值为正数，且总是小于给定排序数组的长度。
• 数组不为空，且长度不超过 104
• 数组里的每个元素与 x 的绝对值不超过 104

分析：

排序按照差值其实也是可以做的，但是理论的时间复杂度比二分高，而且重写comparator比较麻烦。

二分的思路大概就是：

1. 先找到 x 的位置，如果没有 x，那么就应该找到距离 x 最小的位置；
2. 以这个位置为基准，双指针向左右进行扩散；
3. 至于哪边扩散的多少，就要用到题目所说的条件，差值尽量小，一样尽量向左，这样思路就很简单了。

那么，二分的过程用到 模板 3 的，为什么是模板3？

1. 如果数组里有 x ，那要最左边的 x ，也就是左边不是 x，右边是 x 或者>x 才是答案；
2. 如果数组没有 x ，那要尽量接近 x 的位置，则需要判断左右。

显然，是一个考虑每一个位置 及 左右位置 的情况。

在找到第一个目标位置后，要根据需要的 k ，向左右扩散，需要考虑的有三点：

• 边界，找到的目标位置向左右扩散不能越界；
• 不越界的情况下，比较两边位置和 x 的差，挑差值小的那个。

class Solution {
    public List<Integer> findClosestElements(int[] arr, int k, int x) {
        List<Integer> list = new ArrayList();
        int i = 0;
        int j = arr.length;
        //二分，采用模板3
        while( i+1<j ){
            int mid = i+(j-i)/2;
            if( arr[mid]>=x ){
                j = mid;
            }else{
                i = mid;
            }
        }
        //扩散位置，避免ij越界
        while( k>0 ){
            if( i<0 ){
                j++;
            }else if( j>=arr.length ){
                i--;
            }else if( x - arr[i] > arr[j] - x ){
                j++;
            }else{
                i--;
            }
            k--;
        }
        //组合答案
        for( int left = i+1; left < j ;left++ ){
            list.add(arr[left]);
        }
        return list;
    }
}

这里采用模板 3 ，j 的初始值又取了 nums.length 而没有-1，也是调试发现的，这个题目的特殊性。