和至少为 K 的最短子数组

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

题目描述

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

题解

主要是“前缀和+单调双端队列”的解法。首先，对于这种需要多次求数组某段区间之和，甚至求数组每一个区间的和，的情况而言，前缀和是一个很有效的方法。我们定义一个pre数组，$pre[i]$代表位置$i$之前的数值之和。其中，$pre[0]=0$。那么数组从下标$i$开始长度为$L$的子数组之和便是$pre[i+L]-pre[i]$。
有了前缀和之和，我们其实可以枚举所以情况的子数组之和，最终得出满足条件的最短子数组，但这依旧不够高效。
不难发现以下两个特点：

• 当对于$pre[j]$来说，如果已经有$pre[j]-pre[i]>=k$，那么就算在下标$i$之前的另一个下标$i^{{}^{\prime }}$也满足$pre[j]-pre[i^{{}^{\prime }}]>=k$，我们也不需要计算了，因为$j-i$肯定比$j-i^{{}^{\prime }}$短。而且这个长度也是起始下标为$i$的子数组满足大于$k$的最短情况，所以$i$以后便不用再参与计算。
• 假如对于下标$j$来说，在其后下标$k$不满足$pre[k]-pre[j]>=k$，那么对于在下标$j$之前，但是值却大于$nums[j]$的下标来说，肯定也不需要进行考虑了。所以在每次比较完下标，再将其插入需要进行遍历的容器中时，可以先将其中比它大的值弹出，保持该容器单调递增。
  满足上述两个特点，我们需要在该容器两端删除数据，在一端加入数据，所以我们选择双端队列。
  代码如下：

int shortestSubarray(vector<int>& nums, int k) {
    int n = nums.size();
    vector<long long> pre(n + 1, 0);
    # 求前缀和
    for (int i = 0; i < n; ++i){
        pre[i + 1] = pre[i] + nums[i];
    }
    int ret = n + 1;
    deque<int> q;
    for (int i = 0; i <= n; ++i){
        long long cur = pre[i];
        # 从队头开始遍历，遇到满足条件的下标便于记录的ret进行比较，并且将改下标弹出，直至不满足>=k停止循环（队列单调递增）
        while (!q.empty() && cur - pre[q.front()] >= k){
            ret = min(ret, i - q.front());
            q.pop_front();
        }
        # 当前遍历到的下标比较完之后也需要插入队列，为了保持单调递增，则将队尾大于当前值弹出
        while (!q.empty() && pre[q.back()] >= cur){
            q.pop_back();
        }
        q.push_back(i);
    }
    return ret < n + 1 ? ret : -1;
}