LeetCode862. 和至少为 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 <= A.length <= 50000
-10 ^ 5 <= A[i] <= 10 ^ 5
1 <= K <= 10 ^ 9

维护一个前缀和数组sum和一个双端队列que。
若x<y ,sum[y] <= sum[x],则x出队。
若sum[y] - sum[x] >= k,则更新答案，x出队。因为后面的值比y大，这种情况下得到的值不会比刚刚更新过的值小。

class Solution {
    int que[int(1e5 + 10)] = {0};
    int sum[int(1e5 + 10)] = {0};
public:
    int shortestSubarray(vector<int>& A, int K) {
        for(int i = 0; i < A.size(); i++)
            sum[i + 1] = A[i] + sum[i];
        int head = 0, tail = 0, ans = 1e9;
        for(int i = 0; i < A.size() + 1; i++) {
            while(head < tail && sum[i] < sum[que[tail - 1]])   tail--;
            while(head < tail && sum[i] - sum[que[head]] >= K) {
                ans = min(ans, i - que[head]);
                head++;
            }
            que[tail++] = i;
        }
        return ans <= A.size() ? ans : -1;
    }
};