力扣658.找到K个最接近的元素
题目链接:https://leetcode.cn/problems/find-k-closest-elements/description/
给定一个 排序好 的数组 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
arr
按 升序 排列-104 <= arr[i], x <= 104
想法:双向指针,缩小窗口长度到k
代码:
class Solution {
public:vector<int> findClosestElements(vector<int>& arr, int k, int x) {int left=0;int right=arr.size()-1;while(right-left+1>k){if(abs(arr[left]-x)<=abs(arr[right]-x)){right--;}else {left++;}}return vector<int>(arr.begin()+left,arr.begin()+right+1);}
};