题目:
给定一个 排序好 的数组 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里离x最近的k个数,那其实x如果放在这个arr里面的话他也是在一定的位置的,那么对于最接近x来说就要从左右两边去比较,可能是左边也可能是右边,最后比较出k个数即可。
基于这个思路,提出一个滑动窗口的概念。
计算窗口内元素与目标x的差的绝对值的和sum;
向右滑动窗口,并不断比较当前的sum与之前的大小;
在当前sum更小时更新ans,即左窗口边界;sum值相等时不更新(由于本题要返回其自身更小的值);
由于arr是排好序的,根据单调性可以在当前sum更大时直接退出循环。
class Solution {
public:
vector<int> findClosestElements(vector<int>& arr, int k, int x) {
int sum = 0;
int ans = 0;
for(int i = 0; i < k; ++i)
sum += abs(arr[i] - x);
for(int left = 1; left <= arr.size() - k; ++left)
{
int curr_sum = sum - abs(arr[left - 1] - x) + abs(arr[left + k - 1] - x);
if(curr_sum < sum)
ans = left;
else if(curr_sum > sum)
break;
sum = curr_sum;
}
return vector<int>(arr.begin() + ans, arr.begin() + ans + k);
}
};