力扣每日一题: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
-105 <= nums[i] <= 105
1 <= k <= 109

分析:刚开始看题没有注意到题目的下一行,子数组是连续的,我以为是不连续的,我说怎么可能这么简单,排一下序然后就出来了,信心满满,最后给个WA 。然后就仔细看了一下题目,先看一下数据,k>=1 ,说明长度最少是为1的。我们要求和,而且是要大于等于一个数,那前缀和不就来了嘛,先算出数组的前缀和,然后呢,就不会了,哈哈哈哈哈哈哈哈哈,不要急,先奖励自己一把,回来就是精神抖擞。开玩笑回归正题,我们求到了前缀和,第一想到的就是暴力,直接一个双for,时间复杂度O(n2),这个估计会爆了,现在差的就是优化了。在进行分析,坚持就是胜利。

我们可以用一个双端队列来保存一个下标值,这样我们就可以把时间复杂度降到O(n)了,如果一个序列相加起来已经满足要求了,那后面的数还有意义嘛,没有意义啊,所以就可以把它poll掉,然后在比较一下,那个更小,这个算是一个优化,后面如果前缀和j比前面下标的前缀和还更小,这个有意义嘛,没有意义,所以我们又可以把前面的那个下标给poll掉,这就算第二个优化了,我看到了两个图,可以更好的理解

 

AC代码

class Solution {
    public int shortestSubarray(int[] nums, int k) {
       int n = nums.length ;
        int res= n+1 ;
        long preSumArr[] = new long[n+1];

        //前缀和
        for (int i =0 ;i<n;i++){
            preSumArr[i+1]= preSumArr[i]+nums[i];
        }

        //双端队列
        Deque<Integer> queue = new ArrayDeque<>();

        for (int i =0 ;i<=n;i++){
            long cur = preSumArr[i];
            while (!queue.isEmpty()&&cur-preSumArr[queue.peek()]>=k){
                res= Math.min(res , i-queue.pollFirst());
            }

            while (!queue.isEmpty()&&preSumArr[queue.peekLast()]>=cur){
                queue.pollLast();
            }

            queue.offerLast(i);
        }



        return  res>n? -1 : res ;
    }
}

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

little Chen1

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值