给你一个整数数组 nums 和一个整数 k ,找出 nums 中和至少为 k 的 最短非空子数组 ,并返回该子数组的长度。如果不存在这样的 子数组 ,返回 -1 。
子数组 是数组中 连续 的一部分。
class Solution {
public:
int shortestSubarray(vector<int>& nums, int k) {
int n = nums.size();
vector<long> prefix(n + 1);
for (int i = 0; i < n; i++) {
prefix[i + 1] = prefix[i] + nums[i];
}
int res = n + 1;
deque<int> q;
for (int i = 0; i <= n; i++) {
long curSum = prefix[i];
// 前面的某个前缀和第一次出现prefix[i]−prefix[x]≥k
while (!q.empty() && curSum - prefix[q.front()] >= k) {
res = min(res, i - q.front());
q.pop_front();
}
// 放入当前值前,更难满足的数可以直接弹出
while (!q.empty() && prefix[q.back()] >= curSum) {
q.pop_back();
}
q.emplace_back(i);
}
return res < n + 1 ? res : -1;
}
};