LeetCode 658. 找到 K 个最接近的元素

大白羊_Aries

已于 2022-08-25 16:01:48 修改

阅读量1.4k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：题解文章标签： leetcode 算法职场和发展

于 2022-08-25 16:00:41 首次发布

本文链接：https://blog.csdn.net/qq_38204302/article/details/126526605

题解专栏收录该内容

252 篇文章

订阅专栏

本文介绍了一种使用双指针法解决寻找数组中K个最接近目标值x的元素问题的方法。通过在已排序数组上设置两个指针并逐步调整位置来高效找到所需元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

658. 找到 K 个最接近的元素

解法

双指针，既然数组已经排序好了，那么按照 x 就可以分两半边，两边各记录一个指针，分别比较着远离 x

class Solution {
public:
    vector<int> findClosestElements(vector<int>& arr, int k, int x) {
        int right = lower_bound(arr.begin(), arr.end(), x) - arr.begin();
        int 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);
    }
};

有个问题要注意，实例 2 出现 x=-1，起始时 left=-1，这就决定了我们最终取结果时只能是从 arr.begin() + left + 1 开始

为什么要说这个问题，可能你会想按照下面的写法，让 right 先走，但是 arr.begin() + left 越界了

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