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]

方法一：排除法

class Solution {
public:
    vector<int> findClosestElements(vector<int>& arr, int k, int x) {
        int left = 0, right = arr.size() - 1;
        int count = arr.size() - k;
        while(count-- > 0){
           if(arr[right] - x >= x - arr[left])
                right--;
            else 
                left++;
        }
        vector<int> result(arr.begin() + left, arr.begin() + left + k);
        return result;
    }
};

 

 方法二：二分法

class Solution {
public:
    vector<int> findClosestElements(vector<int>& arr, int k, int x) {
        int left = 0, right = arr.size() - k;
        while (left < right){
            int mid = left + (right - left) / 2;
            if (x - arr[mid] > arr[mid + k] - x)
                left = mid + 1;
            else 
                right = mid;
        }
        return vector<int>(arr.begin() + left, arr.begin() + left + k);
    }
};

二分查找 返回的结果数组肯定是数组arr中的连续k个数，利用二分法寻找这段数组的左边界。

 

方法三： 二分+双指针

class Solution {
public:
    vector<int> findClosestElements(vector<int>& arr, int k, int x) {
        int left = 0, right = arr.size();
        while (left < right){
            int mid = left + (right - left) / 2;
            if (arr[mid] >= x)
                right = mid;
            else 
                left = mid + 1;
        }//寻找第一个大于等于x的元素的下标
        left = right - 1;
        while (k--){
            if (left < 0)
                right++;
            else if (right >= arr.size())
                left--;
            else if(x - arr[left] <= arr[right] - x)
                left--;
            else
                right++;
        }
        return vector<int>(arr.begin() + left + 1, arr.begin() + right);
    }
};