入门力扣自学笔记185 C++ （题目编号：862）（多看看：双端队列）

862. 和至少为 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 <= 105
-105 <= nums[i] <= 105
1 <= k <= 109

---

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

---

思路：

这道题我们利用一个双端队列来进行解题。

首先，我们要计算原数组的前缀和，用来快速找到区间的子数组和。

其次，枚举每个前缀，维护到一个单调递增的双端队列中。

最后，对比当前前缀与双端队列的第一个元素（最小元素），通过差值算出区间和是否达到k，如果达到k则将区间长度添加到答案中。

我们返回的时候，只需要找到最优解即可。

---

代码：

class Solution {
public:
    int shortestSubarray(vector<int>& nums, int k) {
        int n = nums.size();
        vector<long long> p{0};
        for(auto a : nums)
            p.push_back(p.back() + a);
        int ans = n + 1;
        deque<int> q;
        for(int i = 0;i < n+1;i++)
        {
            long long a = p[i];
            //cout << a <<endl;
            while(!q.empty() && a <= p[q.back()])
                q.pop_back();
            while(!q.empty() && a - p[q.front()] >= k)
            {
                ans = min(ans,i-q.front());
                q.pop_front();
            }
            q.push_back(i);
        }
        return ans < n + 1 ? ans : -1; 
    }
};