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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/shortest-subarray-with-sum-at-least-k
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

class Solution {
public:
    //使用单调的双端队列 从队头到队尾是递增的
    //首先计算前缀和sum[i+1]
    //我们要找的是sum[x2]-sum[x1]>=k 且 x2-x1最小
    //每次入队时 先比较队尾元素是否比自己大 如果比自己大就将队尾元素pop
    //然后从队头开始比较与自己比较 若差大于等于k则符合要求 res=min(res,自己的坐标减去队头元素的坐标) 将当前队头pop
    int shortestSubarray(vector<int>& A, int K) {
        deque<int> dq;
        int n = A.size();
        vector<int> sum(n + 1, 0);
        for (int i = 1; i < n + 1; i++)sum[i] = sum[i - 1] + A[i-1];
        int res = n + 1;
        for (int i = 0; i < n + 1; i++)
        {
            while (!dq.empty() && sum[dq.back()] > sum[i])dq.pop_back();
            while (!dq.empty() && sum[i] - sum[dq.front()] >= K)
            {
                res = min(res, i - dq.front());
                dq.pop_front();
            }
            dq.push_back(i);
        }
        return res == n + 1 ? -1 : res;
    }
};