862. 和至少为 K 的最短子数组

返回 A 的最短的非空连续子数组的长度,该子数组的和至少为 K

如果没有和至少为 K 的非空子数组,返回 -1 。

 

示例 1:

输入:A = [1], K = 1
输出:1

示例 2:

输入:A = [1,2], K = 4
输出:-1

示例 3:

输入:A = [2,-1,2], K = 3
输出:3

 

提示:

  1. 1 <= A.length <= 50000
  2. -10 ^ 5 <= A[i] <= 10 ^ 5
  3. 1 <= K <= 10 ^ 9

思路:这题我知道用滑动窗口的思想但是具体到怎么去做还是有点不清晰,于是参考了dicuss里面的解法

第一步 求出该数组的前缀和数组 B 值得注意的是B[0]=0; B[j] 表示 A[0]-A[j-1]的值

对于B中的某个位置j来说,它希望找到目前最大的i使得满足 B[J]-B[i]>=K

我们用一个双端队列deque来维护一个递增的在j之前的B中的某些位置,这样一来的目的在于,如果排在最左边的必然是最有可能使得K最大的,如果它不满足,那就直接把j插入到deque中就行了,如果它满足,那就看第二个元素满不满足,如果第二个元素也满足,那么最短长度就会更新。

以下是一些我自己的理解性解释:

1.我们删除第一个元素的保证在于,在j后面的数就算减去第一个数也满足条件,但是它相对于j来说长度长了1.所以我们还是会选择j时候条件成立的最小长度。

2.每次插入都要将大于等于B[j]的值弹出来,最后再将B[j]放进去以保持队列的递增性。

3.为什么要保持递增性的序列呢

我们假设 有x1,x2   B[x1]>B[x2]  ,如果B[j]-B[x1]>=K,那么B[j]-B[x2]也一定>=K,而且x2离j的距离更短,那么我们会选择x2而不是x1。x2的sum值是小于x1的,所以x2会在队列中,x1会被弹出!

Code:

class Solution {
public:
       int shortestSubarray(vector<int> A, int K) {
        int N = A.size(), res = N + 1;
        vector<int> B(N + 1, 0);
        for (int i = 0; i < N; i++) B[i + 1] = B[i] + A[i];
        deque<int> d;
        for (int i = 0; i < N + 1; i++) {
           
            while (d.size() > 0 && B[i] - B[d.front()] >= K)
                res = min(res, i - d.front()), d.pop_front();
            while (d.size() > 0 && B[i] <= B[d.back()]) d.pop_back();
            d.push_back(i);
       
            
        }
        return res <= N ? res : -1;
    }
};

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值