算法

栈与队列 239. 滑动窗口最大值(困难)

2024-03-20  本文已影响0人  宗驴

需求

给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。

返回 滑动窗口中的最大值 。

示例 1:

输入:nums = [1,3,-1,-3,5,3,6,7], k = 3
输出:[3,3,5,5,6,7]
解释:
滑动窗口的位置 最大值


[1 3 -1] -3 5 3 6 7 3
1 [3 -1 -3] 5 3 6 7 3
1 3 [-1 -3 5] 3 6 7 5
1 3 -1 [-3 5 3] 6 7 5
1 3 -1 -3 [5 3 6] 7 6
1 3 -1 -3 5 [3 6 7] 7

[图片上传失败...(image-cda983-1710751580816)]

思路

这个题最大的问题是怎么找出滑动窗口中的最大值,也就是怎么利用结构低成本的找饿到最大值。这里我们实现一个单调队列(单调队列缓存数据,当缓存的元素大于前面元素将小的元素全部移除,直到前面的元素大于当前缓存的元素。

滑动最大窗口.gif
/**
 * 239. 滑动窗口最大值 单调队列
 */

public class MaxSlidingWindow239 {
    public static int[] maxSlidingWindow(int[] nums, int k) {
        int length = nums.length;
        if (length == 1) {
            return nums;
        }

        //前k个元素产生一个最大值,
        //K后每移动一个产生一个最大值
        int[] res = new int[1 + length - k];

        // result数组下标
        int index = 0;
        MyQueue myQueue = new MyQueue();
        for (int i = 0; i < k; i++) {
            myQueue.push(nums[i]);
        }
        // 缓存第一个K 内的最大值
        res[index++] = myQueue.getMax();

        for (int i = k; i < length; i++) {
            // 将非本窗口元素删除
            myQueue.poll(nums[i - k]);
            // 将新的元素添加进去
            myQueue.push(nums[i]);
            res[index++] = myQueue.getMax();
        }

        return res;
    }

    public static void main(String[] args) {
        int[] nums = {1, 3, -1, -3, 5, 3, 6, 7};
        int k = 3;
        // 输出:[3,3,5,5,6,7]
        int[] result = maxSlidingWindow(nums, k);
        for (int i : result) {
            System.out.print(i + "   ");
        }
    }
}

class MyQueue {
    Deque<Integer> deque = new LinkedList<>();

    // 出口处数值是当前最大的元素
    //弹出元素时,比较当前要弹出的数值是否等于队列出口的数值,如果相等则弹出
    public void poll(int val) {
        if (!deque.isEmpty() && val == deque.peek()) {
            deque.poll();
        }
    }

    public void push(int val) {
        // 新增元素的时候会将所有小于这个元素的元素全部弹出
        while (!deque.isEmpty() && deque.peekLast() < val) {
            deque.removeLast();
        }
        deque.add(val);
    }

    public int getMax() {
        return deque.peek();
    }
}
image.png
上一篇 下一篇

猜你喜欢

热点阅读