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

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

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 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 的最短子数组)