LeetCode 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,1,2,3,4,5], k = 4, x = -1
输出:[1,1,2,3]
提示:
1 <= k <= arr.length
1 <= arr.length <= 104^44
arr 按 升序 排列
-104^44 <= arr[i], x <= 104^44
相向双指针,我们需要从arr里删去arr.size()-k个元素:
class Solution {
public:vector<int> findClosestElements(vector<int>& arr, int k, int x) {int left = 0;int right = arr.size() - 1;int deleteNum = 0;while (left < right) {if (deleteNum == arr.size() - k) {break;}if (abs(arr[left] - x) > abs(arr[right] - x)) {++left;} else {--right;}++deleteNum;}return vector<int>(arr.begin() + left, arr.begin() + right + 1);}
};
如果arr的长度为n,则此算法时间复杂度为O(n - k),空间复杂度为O(1)。