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

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

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

示例 3:

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

题目链接

暴力

首先将前缀和算出来。然后通过前缀和去计算区间和。记录下最小的满足题意的区间长度即可

class Solution {
public:
    int shortestSubarray(vector<int>& nums, int k) {
        vector<int>sum(nums.size(),0);
        sum[0]=nums[0];
        int res=INT_MAX;
        //前缀和
        for(int i=1;i<nums.size();i++){
            sum[i]=sum[i-1]+nums[i];
        }
        for(int i=0;i<nums.size();i++){
            if(sum[i]>=k){
                res=min(res,i+1);
                //printf("sum[%d]\n",i);
                break;
            }
        }
        for(int idx=1;idx<nums.size();idx++){
            for(int j=idx;j<nums.size();j++){
                if(sum[j]-sum[idx-1]>=k){
                    res=min(res,j-idx+1);
                    //printf("sum[%d~%d]\n",idx,j);
                    break;
                }
            }
        }
        return res==INT_MAX?-1:res;
    }
};

很显然暴力是必然会TEL的

前缀和加队列

当然是学习了灵佬的题解才做出来的啦。孱弱垃圾博主怎么可能单吃困难呢

题解地址

找到最小j-i满足 j>i && presum[j+1]-presum[i]>=k

优化一

当presum[j+1]-presum[i]>=k已经成立的时候。需要将i出队。因为后面的数已经比j大了，区间长度不可能小于j-i+1

优化二

当presum[j]<=presum[i]的时候。这时，若后面存在x满足x-presum[i]>=k。那也必然满足x-presum[j]>=k。应该将i出队

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