Leetcode862. 和至少为 K 的最短子数组

给你一个整数数组 nums 和一个整数 k ，找出 nums 中和至少为 k 的 最短非空子数组 ，并返回该子数组的长度。如果不存在这样的 子数组 ，返回 -1 。

子数组 是数组中 连续 的一部分。

示例 1：

输入：nums = [1], k = 1
输出：1
示例 2：

输入：nums = [1,2], k = 4
输出：-1
示例 3：

输入：nums = [2,-1,2], k = 3
输出：3

提示：

1 <= nums.length <= 10^5
-10^5 <= nums[i] <= 10^5
1 <= k <= 10^9

想法：
和最少为k的子数组，首先求出该数组的前缀和，后续以 O1 时间复杂度获得任一子数组的和。

long[] s = new long[n + 1];
for(int i = 1; i <= n; i++) s[i] = s[i - 1] + nums[i - 1];

如果数组中不存在负数，则前缀和单调递增，直接如果滑动窗口得出答案。
但数组中存在负数，前缀和呈以下形式:

对于每个点，要找到离自己最近，且满足和至少为 k 的点。
例如如果节点 ⑥ 与节点 ③ 的差值 >= k，节点 ⑥ 与节点 ② 的差值也 >= k。
显而易见的，节点③更适合作为答案输出，节点③的存在使得节点②不可能作为"最短子数组的左边界"。
由此推出，如果当前节点 x 值小于前面某些节点，则前面的节点可以丢掉不要。那么上图可以修改为

题解：
遍历子数组的右边界，并维护一个单调递增队列，丢掉前面比当前前缀和大的节点后，插入队尾。遍历队头，重复判断是否存在符合条件的子数组左边界，记录答案。由于队头队尾都需要具有 poll 方法，使用 Deque 双端队列。
代码：

class Solution {
    public int shortestSubarray(int[] nums, int k) {
        int n = nums.length;
        Deque<long[]> q = new ArrayDeque<>();
        long[] s = new long[n + 1];
        int ret = 0x3f3f3f3f;
        for(int i = 1; i <= n; i++) s[i] = s[i - 1] + nums[i - 1];
        for(int i = 0; i <= n; i++){
            long sv = s[i];
            //如果当前前缀和小于前面某个前缀和，则前面的前缀和，不可能是满足条件的最短子数组
            while(!q.isEmpty() && q.peekLast()[0] >= sv) q.pollLast();
            //单调递增队列，最小值在队头
            while(!q.isEmpty() && sv - q.peekFirst()[0] >= k){
                ret = Math.min(ret,(int) (i - q.pollFirst()[1]));
            }
            q.offer(new long[]{sv,i});
        }
        return ret == 0x3f3f3f3f ? -1 : ret;
    }
}