力扣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
arr
按 升序 排列-104 <= arr[i], x <= 104
代码:
class Solution {
public:
vector<int> findClosestElements(vector<int>& arr, int k, int x) {
int left = 0, right = k-1;
int cost = 0, mincost = 99999999999999, minleft = 0;
vector<int> res;
while(right < arr.size()){
cost = 0;
for(int i = left; i <= right; i++){
cost += abs(arr[i] - x);
}
if(cost < mincost){
minleft = left;
mincost = cost;
}
if(cost > mincost){
break;
}
left++;
right++;
}
for(int i = minleft; i <= minleft+k-1; i++){
res.push_back(arr[i]);
}
return res;
}
};
解题思路:
(1)暴力暴力暴力。
(2)定义 k 大小的窗口,计算每个元素与 x 的差。
(3)找到最小差值的窗口。
(4)还可以使用双指针从头尾开始遍历。