leetcode862_和至少为K的最短子数组_单调双端队列&前缀和

1. 这题感觉真的很难, 首先求和的方法转换为前缀和的形式,避免重复计算.想求[i,j]区间则为sum[i+1]-sum[j].

2. 运用单调双端队列解决应该选择哪个区间的问题.因为如果存在负数,则需要把包含负数区间的求和去掉,因为负数后面的数肯定也符合条件>=K,而且长度也比较小.

class Solution {
public:
    int shortestSubarray(vector<int>& A, int K) {
        if(A.size()==0) return -1;
        //定义A.size()+1长度的数组,包含sum[0].
        //解决sum[1]=sum[0]+A[0]情况.
        //前缀和sum[i]代表[0,i-1]区间的求和.
        vector<int> sum(A.size()+1, 0);
        deque<int> dq;
        int res = INT_MAX;
        for(int i=1;i<=A.size();i++) {
            sum[i] = sum[i-1]+A[i-1];
        }
        //单调双端队列保证单增,如果存在小于等于的情况,
        //说明里面一定包含负数或0,去掉dq最后一个值.
        //从dq最前寻找符合>=K条件的长度,并计算最短长度.
        for(int i=0;i<=A.size();i++) {
            if(i!=0) {
                while(dq.size()>0 && sum[dq.back()]>=sum[i]) dq.pop_back();
                while(dq.size()>0 && sum[i]-sum[dq.front()]>=K) {
                    res = min(res, i-dq.front());
                    dq.pop_front();
                }
            }
            dq.push_back(i);
        }
        //有可能不存在,则返回-1.
        return res==INT_MAX ? -1:res;
    }
};