小白leetcode 22 10 02

题目:

给定一个 排序好 的数组 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);
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值