美文网首页
【教3妹学算法-每日3题(3)】 和至少为 K 的最短子数组

【教3妹学算法-每日3题(3)】 和至少为 K 的最短子数组

作者: 程序员小2 | 来源:发表于2022-07-18 22:33 被阅读0次

    插: 前些天发现了一个巨牛的人工智能学习网站,通俗易懂,风趣幽默,忍不住分享一下给大家。点击跳转到网站。
    坚持不懈,越努力越幸运,大家一起学习鸭~~~

    3妹

    3妹:小呀么小二郎呀, 背着那书包上学堂。
    2哥:不怕太阳晒, 不怕那风雨打。
    3妹:就怕老师说我懒呀,没有学问,无脸见爹娘。
    2哥:3妹, 周杰伦又发新专辑了,you know? 你的曲库该更新一下了。
    3妹:yeah, I know. 我可是听着我伦的歌长大的。
    2哥:是的, 记得那时还是上高中的时候……
    3妹:2哥,又开始回忆你的青春岁月了,哈哈
    2哥:3妹也会取笑人了,不跟你说了,我继续做题了。

    讲课

    题目:

    给你一个整数数组 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 <= 10^5
    -105 <= nums[i] <= 10^5
    1 <= k <= 10^9

    java代码:

    class Solution {
        public int shortestSubarray(int[] A, int K) {
            int N = A.length;
            long[] P = new long[N+1];
            for (int i = 0; i < N; ++i)
                P[i+1] = P[i] + (long) A[i];
    
            // Want smallest y-x with P[y] - P[x] >= K
            int ans = N+1; // N+1 is impossible
            Deque<Integer> monoq = new LinkedList(); //opt(y) candidates, as indices of P
    
            for (int y = 0; y < P.length; ++y) {
                // Want opt(y) = largest x with P[x] <= P[y] - K;
                while (!monoq.isEmpty() && P[y] <= P[monoq.getLast()])
                    monoq.removeLast();
                while (!monoq.isEmpty() && P[y] >= P[monoq.getFirst()] + K)
                    ans = Math.min(ans, y - monoq.removeFirst());
    
                monoq.addLast(y);
            }
    
            return ans < N+1 ? ans : -1;
        }
    }
    

    相关文章

      网友评论

          本文标题:【教3妹学算法-每日3题(3)】 和至少为 K 的最短子数组

          本文链接:https://www.haomeiwen.com/subject/hxklirtx.html