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

文章介绍了一种解决和至少为K的最短子数组问题的方法,通过计算数组的前缀和并使用单调递增队列来优化搜索过程。当当前前缀和减去队首前缀和大于等于K时,更新答案。同时,如果当前前缀和小于队尾前缀和,则移除队尾元素,保持队列单调性。最终返回满足条件的最短子数组长度,若无解则返回-1。
摘要由CSDN通过智能技术生成
862. 和至少为 K 的最短子数组

看题目是与和有关的问题,那么就联想到前缀和,先把数组的前缀和算出来,在队列里插入下标,当当前前缀和减去队首前缀和满足条件时,说明有满足条件的数组出现,计算元素个数,同时把队首弹出,直到不满足条件,为了维护队列单调递增,当当前前缀和小于队尾前缀和,即将队尾弹出,直到符合递增队列。

为什么维护单调递增队列,因为要求和至少为k的子数组,队首存放最小的下标,与最小值的差值是最大的,更可能满足条件。 

class Solution {
public:
    int shortestSubarray(vector<int>& nums, int k) {
      
       int n=nums.size();
       vector<long long> sum(n+1,0);
       for(int i=0;i<n;i++)
       sum[i+1]=sum[i]+nums[i];
       int ans=INT_MAX;
       deque<int> que;
       for(int i=0;i<sum.size();i++)
       {
            while(!que.empty()&&sum[i]-sum[que.front()]>=k)
            {
                ans=min(ans,i-que.front());
                que.pop_front();
            }

            while(!que.empty()&&sum[i]<=sum[que.back()])
            que.pop_back();

            que.push_back(i);
       }
       if(ans==INT_MAX)
       return -1;
       else
       return ans;
    }
    
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值