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

来源：力扣（LeetCode）

描述：

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

方法一：排序

首先将数组 arr 按照「更接近」的定义进行排序，如果 a 比 b 更接近 x，那么 a 将排在 b 前面。排序完成之后，k 个最接近的元素就是数组 arr 的前 k 个元素，将这 k 个元素从小到大进行排序后，直接返回。

代码：

class Solution {public:vector<int> findClosestElements(vector<int>& arr, int k, int x) {sort(arr.begin(), arr.end(), [x](int a, int b) -> bool {return abs(a - x) < abs(b - x) || abs(a - x) == abs(b - x) && a < b;});sort(arr.begin(), arr.begin() + k);return vector<int>(arr.begin(), arr.begin() + k);}
};

执行用时：52 ms, 在所有 C++ 提交中击败了22.20%的用户
内存消耗：30.2 MB, 在所有 C++ 提交中击败了93.42%的用户
复杂度分析
时间复杂度： O(nlogn)，其中 n 是数组 arr 的长度。排序需要 O(nlogn)。
空间复杂度： O(logn)。返回值不计算时间复杂度。排序需要 O(logn) 的栈空间。

方法二：二分查找 + 双指针

假设数组长度为 n，注意到数组 arr 已经按照升序排序，我们可以将数组 arr 分成两部分，前一部分所有元素 [0, left] 都小于 x，后一部分所有元素 [right, n − 1] 都大于等于 x，left 与 right 都可以通过二分查找获得。

left 和 right 指向的元素都是各自部分最接近 x 的元素，因此我们可以通过比较 left 和 right 指向的元素获取整体最接近 x 的元素。如果 x − arr[left] ≤ arr[right] − x，那么将 left 减一，否则将 right 加一。相应地，如果 left 或 right 已经越界，那么不考虑对应部分的元素。

最后，区间 [left + 1, right − 1] 的元素就是我们所要获得的结果，返回答案既可。

代码：

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);}
};

执行用时：28 ms, 在所有 C++ 提交中击败了86.25%的用户
内存消耗：30.3 MB, 在所有 C++ 提交中击败了70.57%的用户
复杂度分析
时间复杂度： O(logn + k)，其中 n 是数组 arr 的长度。二分查找需要 O(logn)，双指针查找需要 O(k)。
空间复杂度：O(1)。返回值不计入空间复杂度。
author：LeetCode-Solution

【658. 找到 K 个最接近的元素】相关推荐

Java实现 LeetCode 658 找到 K 个最接近的元素（暴力）
658. 找到 K 个最接近的元素给定一个排序好的数组,两个整数 k 和 x,从数组中找到最靠近 x(两数之差最小)的 k 个数.返回的结果必须要是按升序排好的.如果有两个数与 x 的差值一样,优先 ...
【LeetCode通关全记录】658. 找到 K 个最接近的元素
[LeetCode通关全记录]658. 找到 K 个最接近的元素题目地址
leetcode 658. 找到 K 个最接近的元素
leetcode 658. 找到 K 个最接近的元素题目描述: 给定一个排序好的数组 arr ,两个整数 k 和 x ,从数组中找到最靠近 x(两数之差最小)的 k 个数.返回的结果必须要是按升序排 ...
LeetCode——658. 找到 K 个最接近的元素（双指针）
文章目录 658. 找到 K 个最接近的元素(双指针) 题目 1.双指针法思想:** 代码 2.二分法思想:** 代码 658. 找到 K 个最接近的元素(双指针) 来源:力扣(LeetCode) ...
LeetCode 每日一题——658. 找到 K 个最接近的元素
1.题目描述 658. 找到 K 个最接近的元素给定一个排序好的数组 arr ,两个整数 k 和 x ,从数组中找到最靠近 x(两数之差最小)的 k 个数.返回的结果必须要是按升序排好的. 整数 ...
658. 找到 K 个最接近的元素
给定一个排序好的数组 arr ,两个整数 k 和 x ,从数组中找到最靠近 x(两数之差最小)的 k 个数.返回的结果必须要是按升序排好的. 整数 a 比整数 b 更接近 x 需要满足: |a - ...
leetcode 658. Find K Closest Elements | 658. 找到 K 个最接近的元素（二分查找+双指针）
题目 https://leetcode.com/problems/find-k-closest-elements/ 题解在arr中找到第一个小于等于x的位置mid,然后再根据题意,用双指针分别向左. ...
LeetCode 658. 找到 K 个最接近的元素（二分查找）
1. 题目给定一个排序好的数组,两个整数 k 和 x,从数组中找到最靠近 x(两数之差最小)的 k 个数.返回的结果必须要是按升序排好的.如果有两个数与 x 的差值一样,优先选择数值较小的那个数. ...
力扣 658. 找到 K 个最接近的元素
题目给定一个排序好的数组 arr ,两个整数 k 和 x ,从数组中找到最靠近 x(两数之差最小)的 k 个数.返回的结果必须要是按升序排好的. 整数 a 比整数 b 更接近 x 需要满足: | ...

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

【658. 找到 K 个最接近的元素】相关推荐

最新文章

热门文章