leetcode刷题记录-658. 找到 K 个最接近的元素

121 阅读1分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第25天,点击查看活动详情

前言

今天的题目为中等,本文的做法比较取巧,利用了js提供的排序方法进行解决。

每日一题

今天的题目是 658. 找到 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]

示例 2:

输入:arr = [1,2,3,4,5], k = 4, x = -1
输出:[1,2,3,4]

 

提示:

  • 1 <= k <= arr.length
  • 1 <= arr.length <= 104
  • arr 按 升序 排列
  • -104 <= arr[i], x <= 104

题解

sort 排序

做这道题之前首先来了解一下 sort 这个数组方法的妙用,这个方法要是不传入参数,会把数组的每一个都当作字符串去取第一位然后根据 ASCII 值来进行排序,所以要是传入了 40 和 5 ,5 是会排在 40 前面的

image.png

但是这个方法可以传入一个回调函数,这个回调函数返回 true 则位置互换,返回 false,则不变,回调能够拿到数组按顺序往下的两个两个元素。

image.png

所以根据题目的要求,我们能够得出我们需要的一个排序规则,拿着前后两个数去和 k 做一个比较,分别用两个数的减去 k 取得绝对值,谁的绝对值更小,说明谁离 k 更近。

特殊情况就是当两个值减去 k 的绝对值相等的时候,我们就需要判断两个值谁比较大,要留下来小的那一个。

function findClosestElements(arr: number[], k: number, x: number): number[] {
    arr.sort((a, b) => {
        if (Math.abs(a - x) !== Math.abs(b - x)) {
            return Math.abs(a - x) - Math.abs(b - x);
        } else {
            return a - b;
        }
    });
    const res = arr.slice(0, k);
    res.sort((a, b) => a - b);
    return res;
};

image.png