返回 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 <= A.length <= 50000
-10 ^ 5 <= A[i] <= 10 ^ 5
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;
}
};