题目
前缀和 越大还越靠前的肯定要被淘汰。
对于每个i,前面满足sum[i]-sum[q[head]]>=K的,++head就可以了
因为 后面的i,肯定不会用这些head。与其这些head被 i 后面的使用,为啥不被 i 使用,所以head被后面的 i 用了也没用,不是最优,所以不用。
class Solution {
public:
int *sum;
int getsum(int x){return (x==-1)?0:sum[x];}
int shortestSubarray(vector<int>& A, int K) {
int n=A.size();sum=new int[n];
int *q=new int[n+2],head=0,tail=1,ans=n+1;q[++head]=-1;
for(int i=0;i<n;++i){
sum[i]=i?(sum[i-1]+A[i]):A[i];
while(head<=tail&&getsum(q[tail])>=getsum(i)) --tail;
while(head<=tail&&getsum(i)-getsum(q[head])>=K) ans=min(ans,i-q[head]),++head;
q[++tail]=i;
}
return (ans==n+1)?-1:ans;
}
};