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

解题报告

首先二分查找第一个大于 x 的元素，然后向两边扩展查找即可。

实现代码

class Solution {
public:
    vector<int> findClosestElements(vector<int>& arr, int k, int x) {
        //二分法求第一个大于 x 的数字
        vector<int>ans;
        int l=0, r=arr.size()-1, mid;
        while(l<r){
            mid=l+(r-l)/2;
            if(arr[mid]<x) l=mid+1;
            else r=mid;
        }
        cout<<arr[r]<<" ";
        int i=r-1, j=r;
        int counter=0;
        while(i>=0&&j<arr.size()){
            if(abs(arr[i]-x)<=abs(arr[j]-x)) ans.push_back(arr[i--]);
            else if(abs(arr[i]-x)>abs(arr[j]-x)) ans.push_back(arr[j++]);
            counter++;
            if(counter==k) break;
        }
        while(counter!=k&&i>=0){
            ans.push_back(arr[i--]);
            counter++;
        }
        while(counter!=k&&j<arr.size()){
            ans.push_back(arr[j++]);
            counter++;
        }
        sort(ans.begin(), ans.end());
        return ans;
    }
};

参考资料

[1] Leetcode 658. 找到 K 个最接近的元素