LeetCode题解:找到 K 个最接近的元素

模版III - 找到 K 个最接近的元素(middle)

一、题目

LeetCode658.找到 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]

二、基础模版 III

因为我们的判断区间最少为 2 个元素,所以我们要注意循环的执行条件

  1. 简单的判断边界: nums.length === 0,return -1
  2. 定义初始的左右边界:left = 0, right = nums.length - 1
  3. 确定执行条件:left + 1 < right,这也意味着查找区间要存在 3 个元素;
  4. 向左查找时:right = mid
  5. 向左查找时:left = mid
  6. 判断剩下的两个元素哪个符合目标元素,并返回结果;

三、题解

分析模版

  • 我们的目标是:寻找目标值的起始下标和终止下标,但它是可能重复的
  • 针对这样的情况,我们要将判断拆解成查找目标首次出现的位置,和最后一次出现的位置

JavaScript代码

var findClosestElements = function(arr, k, x) {
    let left = 0;
    let right = arr.length - k; // 注意我们让出来k的值,当作我们的“备用值”
    let res = [];
    while (left < right) {
        let mid = left + ((right - left) >> 1);
        if (x - arr[mid] > arr[mid + k] - x) {
            // x-arr[mid] > arr[mid+k]-x,arr[mid+k]的点一定距离目标值更近
            // 且它距离arr[mid]是k个,证明arr[mid]一定不会存在与结果区间中,它也不可能是起始点
            left = mid + 1; // 我们在移动满足条件的起始点
        } else {
            // x-arr[mid] <= arr[mid+k]-x,arr[mid]下标的点一定距离目标值更近
            // 所以我们需要收缩右边界
            right = mid; // 压缩右区间
        }
    }
    for(let i = left; i < left + k; i++){
        res.push(arr[i])
    }
    return res;
};

四、写在最后

本文是二分查找-模版III 的第最后一题,我会在之后将二分法的常见问题整理成完成的文档,我们一起加油~

如果对你有所帮助不妨给本项目的github 点个 star,这是对我最大的鼓励

关于我

  • 花名:余光
  • WX:j565017805
  • 沉迷 JS,水平有限,虚心学习中

其他沉淀

````
相关推荐
©️2020 CSDN 皮肤主题: 博客之星2020 设计师:CY__ 返回首页