LeetCode862 和至少为k的最短子数组

Keal、

已于 2023-10-10 16:44:59 修改

阅读量601

点赞数

文章标签：算法 java 数据结构

于 2023-10-10 16:32:22 首次发布

本文链接：https://blog.csdn.net/qq422243639/article/details/133749376

版权

本文介绍了一种使用单调队列实现的算法，用于解决给定整数数组nums和整数k，找到满足Sj-Si>=k时的最小子数组的长度。通过构造前缀和和维护一个滑动窗口，确保队列内的元素满足条件，最终返回符合条件的子数组的最小长度。

摘要由CSDN通过智能技术生成

题目：

在这里插入图片描述

解析：
1、先构造前缀和数组
2、单调队列存放滑动窗口，目的求Sj-Si >=k的情况下，窗口最小。

代码：

class Solution {
    public int shortestSubarray(int[] nums, int k) {
        int n = nums.length;
        long[] sums = new long[n + 1];      // 注意这里加和后，容易超过int，加和用long数组
        for (int i = 1; i <= n; i++) {
            sums[i] = sums[i - 1] + nums[i - 1];
        }

        Deque<Integer> deque = new LinkedList<>();
        deque.add(0);

        int res = Integer.MAX_VALUE;

        for (int i = 1; i <= n; i++) {
            while (!deque.isEmpty() && sums[i] - sums[deque.peek()] >= k) { // 注意边界，求最小这里应该 >=
                res = Math.min(res, i - deque.poll());
            }

			//将所有大于等于s[i]的数删掉 问题：这里不删能怎么样，上面求最小值会守影响么？
            while (!deque.isEmpty() && sums[deque.peekLast()] >= sums[i]) {  // 注意边界，求最小这里应该 >=
                deque.pollLast();
            }

            deque.add(i);
        }

        if (res == Integer.MAX_VALUE) {
            return -1;
        }

        return res;
    }
}