2022.10.26 leetcode每日一题 862.和至少为 K 的最短子数组

HeartRaindj

已于 2022-10-26 16:20:47 修改

阅读量165

点赞数

文章标签：算法 leetcode 数据结构

于 2022-10-26 11:07:10 首次发布

本文链接：https://blog.csdn.net/Morso/article/details/127528702

版权

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

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

子数组是数组中连续的一部分。

示例 1：

输入：nums = [1], k = 1
输出：1

解题思路：前缀和+单调队列

1. 前缀和的构造技巧

损失第一个元素，即令第一个元素为0，构造此等式pre[i + 1] = pre[i] + nums[i]
令pre[0] = nums[0]，构造pre[i + 1] = pre[i] + nums[i + 1]

2. 单调队列

利用deque 定义一个双端队列，如果维护的队列是严格单调的便称该队列为单调队列。

在求出前缀和的基础上，我们考虑两种优化：

i > j && pre[i] - pre[j] >= k，这种情况在 i 后面的pre不可能还会比 i - j 更优，
i > j && pre[i] <= pre[j]，此时pre[j] 也失去了最优，因为后面的 pre -pre[i] 要比 pre - pre[j] 大。

代码：

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