LeetCode 658 找到K个最接近的元素[二分法] HERODING的LeetCode之路

Posted 2021-06-27 HERODING23

解题思路：
一道使用经典二分做的题目，但是思路需要经过一定的推导，即临界点怎么找，怎么说明该点是差值和最小的位置？观察发现，临界点位于左边差比右边差相等或者略小的临界位置，这时候从最左边连续的k个数与x的差最小，代码如下：

class Solution {
public:
    vector<int> findClosestElements(vector<int>& arr, int k, int x) {
        int low = 0, high = arr.size() - k;
        while(low < high) {
            int mid = (high + low) / 2;
            // 找左边差比右边差相等或者略小的临界位置
            if(x - arr[mid] > arr[mid + k] - x) {
                low = mid + 1;
            } else {
                high = mid;
            }
        }
        // 返回从low开始的k长度的数组
        return vector<int>(arr.begin() + low, arr.begin() + k + low);
    }
};


/*作者：heroding
链接：https://leetcode-cn.com/problems/find-k-closest-elements/solution/cer-fen-fa-by-heroding-a89x/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。*/