LeetCode探索（132）：658-找到 K 个最接近的元素

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

题目

给定一个 排序好 的数组 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 <= 10^4
• arr 按 升序 排列
• -10^4 <= arr[i], x <= 10^4

思考

本题难度中等。

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

我们可以通过排序的方法来解决问题。首先，将数组 arr 按照「更接近」的定义进行排序。我们可以借助数组的sort方法进行排序，也就是使用自定义的排序函数，函数的内容也就是按照题目中「更接近」的定义。排序完成之后，k 个最接近的元素就是数组 arr 的前 k 个元素，我们只需要将这 k 个元素从小到大进行排序并返回。

解答

方法一：排序

/**
 * @param {number[]} arr
 * @param {number} k
 * @param {number} x
 * @return {number[]}
 */
var findClosestElements = function(arr, k, x) {
  let list = [...arr]
  // 按照整数 a 比整数 b 更接近 x 的定义进行排序
  list.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 ans = list.slice(0, k)
  ans.sort((a, b) => a - b)
  return ans
}

复杂度分析：

• 时间复杂度：O(nlogn)，其中 n 是数组 arr 的长度。排序需要 O(nlogn)。
• 空间复杂度：O(logn)。排序需要 O(logn) 的栈空间。

参考

• 找到 K 个最接近的元素