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

最新推荐文章于 2024-06-23 22:33:26 发布

禮樂_

最新推荐文章于 2024-06-23 22:33:26 发布

阅读量338

点赞数

分类专栏： LeetCode每日一题文章标签：算法 leetcode c++

本文链接：https://blog.csdn.net/qq_44238580/article/details/127542061

版权

LeetCode每日一题专栏收录该内容

21 篇文章 0 订阅

订阅专栏

输入：nums = [1,2], k = 4
输出：-1

题目描述：

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

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

示例一：

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

示例二：

输入：nums = [1,2], k = 4
输出：-1

实例三：

输入：nums = [2,-1,2], k = 3
输出：3

代码：

class Solution {
public:
    int shortestSubarray(vector<int> &nums, int k) {
        int n = nums.size(), ans = n + 1;
        long s[n + 1];
        s[0] = 0L;
        for (int i = 0; i < n; ++i)
            s[i + 1] = s[i] + nums[i]; // 计算前缀和
        deque<int> q;
        for (int i = 0; i <= n; ++i) {
            long cur_s = s[i];
            while (!q.empty() && cur_s - s[q.front()] >= k) {
                ans = min(ans, i - q.front());
                q.pop_front(); // 优化一
            }
            while (!q.empty() && s[q.back()] >= cur_s)
                q.pop_back(); // 优化二
            q.push_back(i);
        }
        return ans > n ? -1 : ans;
    }
};