【力扣】[数组] 658.找到 K 个最接近的元素

1、题目

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

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

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

链接：https://leetcode-cn.com/problems/find-k-closest-elements

 

2、思路分析

1）二分
已排序的结果自然想到用二分法

1. 找左边差比右边差相等或者略小的临界位置
2. 结果就是临界位置靠右k个点即可

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