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

hhhcbw

已于 2022-10-26 11:22:34 修改

阅读量350

点赞数

分类专栏：力扣刷题笔记文章标签： leetcode 算法 c++

于 2022-10-26 11:19:51 首次发布

本文链接：https://blog.csdn.net/weixin_44491423/article/details/127528497

版权

力扣刷题笔记专栏收录该内容

14 篇文章 1 订阅

订阅专栏

文章目录

题目链接
题目大意
解题思路
代码(C++)

题目链接

点我(^_^)

题目大意

解题思路

这题主要用到前缀和+双向单调队列，前缀和 $s u m [i]$ 表示 ${\textstyle \sum_{j=0}^{i-1}} nums[j]$ ，那么使用 $s u m [a] - s u m [b]$ 就表示 ${\textstyle \sum_{j=b}^{a-1}} nums[j]$ ，就可以 O(1) 查询子数组之和。

我们从前往后遍历 $s u m$ 数组，双向单调队列 $q$ 保存了前面遍历过的 $s u m$ 数组下标，那么对于当前 $s u m [i]$ ，如果其在 $q$ 中找到了满足 $s u m [i] - s u m [j] >= k$ 的 $j$ ，那么 $j$ 就可以从 $q$ 中删去，原因是就算之和还有满足条件的 $i$ ，其对答案也不会产生贡献(因为长度一定大于当前情况)。又因为单调队列(后面说明为什么会成为单调队列)，所以如果当前的 $j$ 不满足，那么 $j$ 之后在 $q$ 中保存的也一定不满足，直接结束当前循环。原因是单调队列，保证了 $j$ 之后的 $s u m$ 一定大于 $s u m [j]$ ，那么 $s u m [j]$ 作为减数已经是当前最小的了，那之后更大的 $s u m$ 作为减数肯定更不满足。

至于为什么会成为单调队列，因为对于当前遍历的 $i$ ，完成上述在 $q$ 中遍历之后，要将其加入到 $q$ 中，而如果 $q$ 中存在下标 $k$ 使得 $s u m [k] >= s u m [i]$ ，那么因为前面遍历过的 $s u m$ 只能作为减数，那么减数肯定是越小越好，如果之后 $i$ 都不满足，那么 $k$ 一定不会满足，对于相等的情况， $i$ 又在后面，会更好(因为要最短的子数组)。所以要把 $k$ 从 $q$ 中删去，至此 $q$ 成为单调队列。

代码(C++)

class Solution {
public:
    int shortestSubarray(vector<int>& nums, int k) {
        int ans = nums.size() + 1;
        deque<int> q;
        long long sum[100005];
        sum[0] = 0;
        for(int i=0; i<nums.size(); ++i) {
            sum[i+1] = sum[i] + nums[i];
        }
        q.push_back(0);
        for(int i=0; i<nums.size(); ++i) {
            while(!q.empty()) {
                if(sum[i+1] - sum[q.front()] >= k) {
                    ans = min(i+1-q.front(), ans);
                    q.pop_front();
                } else {
                    break;
                }
            }
            while(!q.empty()) {
                if(sum[i+1] <= sum[q.back()])
                    q.pop_back();
                else
                    break;
            }
            q.push_back(i+1);
        }
        if(ans == nums.size()+1)
            return -1;
        else
            return ans;
    }
};