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

最新推荐文章于 2024-09-27 11:50:10 发布

NakamotoItachi

最新推荐文章于 2024-09-27 11:50:10 发布

阅读量91

点赞数

分类专栏： 21天打卡文章标签：算法 leetcode c++

本文链接：https://blog.csdn.net/i_muyu/article/details/127540842

版权

21天打卡专栏收录该内容

20 篇文章 0 订阅

订阅专栏

在这里插入图片描述
抄题解

class Solution {
public:
    int shortestSubarray(vector<int>& nums, int k) {
        int n = nums.size();
        vector<long> preSumArr(n + 1);
        for (int i = 0; i < n; i++) {
            preSumArr[i + 1] = preSumArr[i] + nums[i];
        }
        int res = n + 1;
        deque<int> qu;
        for (int i = 0; i <= n; i++) {
            long curSum = preSumArr[i];
            while (!qu.empty() && curSum - preSumArr[qu.front()] >= k) {
                res = min(res, i - qu.front());
                qu.pop_front();
            }
            while (!qu.empty() && preSumArr[qu.back()] >= curSum) {
                qu.pop_back();
            }
            qu.push_back(i);
        }
        return res < n + 1 ? res : -1;
    }
};