LeetCode刷题系列 -- 862. 和至少为 K 的最短子数组

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

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

示例 1：

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

输出：1

示例 2：

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

输出：-1

示例 3：

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

输出：3

提示：

• 1 <= nums.length <= 105

• -10^5 <= nums[i] <= 105

• 1 <= k <= 10^9

862. 和至少为 K 的最短子数组 - 力扣（Leetcode）

思路：

定义前缀和数组 preSum
定义双端队列 de ，用户保存子数组起始的下标，其保持非递减序列，n压入 de, m 后压入 de,有 preSum[ m ] <= preSum[ n ] ，
遍历 前缀和数组 preSum。
1) 若是当前的前缀和 preSum[i] 减去 de 中的最小值 大于等于 k ，更新子数组最小长度
2) 清空 de 中大于等于 preSum[i] 的元素，因为 de中的元素作为 减数，显然减数越小越能实现子数组的和大于等于 k 的目的，即便是 de 中有元素preSum[m] 与 preSum[i] 相等，那么以 nums[m] 为起始的子数组的长度也大于 以 nums[i] 为起始的子数组

c++

class Solution {
public:
    int shortestSubarray(vector<int>& nums, int k) {
        vector<int64_t> preSum(nums.size(), 0); // 前缀和数组
        

        for(int i = 0; i < nums.size(); i++) {
            if(i == 0) {
                preSum[i] = nums[0];
            } else {
                preSum[i] = preSum[i-1] + nums[i];
            }
        }
        int result = nums.size() + 1;

        deque<int> de; // 用于存储前缀和数组的下标，其前缀和 为单调递增
        
        for(int i = 0; i< preSum.size(); i++) {
            int64_t curNum = preSum[i];
            if(curNum >= k) {
                result = result > (i+1) ? (i+1) : result;
            }
            // 若是当前的前缀和 preSum[i] 减去 de 中的最小值 大于等于 k ，更新子数组最小长度 
            while(!de.empty() && curNum - preSum[de.front()] >= k) {
                result = result > (i - de.front()) ? (i - de.front()) : result;
                de.pop_front();
            }

            // 清空 de 中大于等于  preSum[i] 的元素，因为 de中的元素作为 减数，显然减数越小越能实现子数组的和大于等于 k 的目的，即便是 de  中有元素preSum[m] 与 preSum[i] 相等，那么以 nums[m] 为起始的子数组的长度也大于 以 nums[i] 为起始的子数组
            while(!de.empty() && curNum <= preSum[de.back()]) {
                de.pop_back();
            }

            de.push_back(i);
        }

        return result == (nums.size() + 1) ? -1 : result;
    }
};