代码随想录-栈与队列 | 239滑动窗口最大值

代码随想录-栈与队列 | 239滑动窗口最大值

LeetCode 239-滑动窗口最大值

题目链接
代码随想录

题目描述

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

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

示例

输入:nums = [1,3,-1,-3,5,3,6,7], k = 3

输出:[3,3,5,5,6,7]

解题思路

判断

  • 使用单调序列:队列中的元素一定要排序,并将最大值放在出队口。
  • 问题在于:窗口移动时,队列需要弹出元素,如何在排序后的队列中弹出正确的元素呢?(这个元素显然不一定就是最大/小值)
  • 其实队列只需要维护可能成为窗口里最大值的元素即可,同时保证队列里的元素数值是由大到小的。

代码

java 复制代码
//利用双端队列手动实现单调序列
/*
 *  用一个单调队列来存储对应的下标,每当窗口滑动的时候,直接取队列的
 *头部指针对应的值放入结果集即可。
 *  单调队列类似 (tail -->) 3 --> 2 --> 1 --> 0 (--> head) 
 *(右边为头节点,元素存的是下标)
 */
class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        ArrayDeque<Integer> deque = new ArrayDeque<>();
        int n = nums.length;
        int[] res = new int[n - k + 1];
        int idx = 0;
        for(int i = 0; i < n; i++) {
            //根据题意,i为nums下标,要在[i-k+1,i]中选到最大值。只需保证两点:
            //1.队列头节点需要在[i-k+1,i]范围内,不符合则弹出
            while(!deque.isEmpty() && deque.peek() < i - k + 1) {
                deque.poll();
            }
            //2.既然是单调,就要保证每次放进去的数字要比末尾的都大,否则也弹出
            while(!deque.isEmpty() && nums[deque.peekLast()] < nums[i]) {
                deque.pollLast();
            }
            deque.offer(i);

            //因为单调,当i增长到符合第一个k范围的时候,每滑动一步都将队列头节点放入结果里即可
            if(i >= k - 1){
                res[idx++] = nums[deque.peek()];
            }
        }
        return res;
    }
}

也可以用自定义数组进行实现。这个以后再看。

复杂度

  • 时间复杂度
    ∵ nums中的每个元素最多被push和pop各一次,没有任何多余操作
    ∴ 时间复杂度为O(n)
  • 空间复杂度
    ∵ 我们定义了一个辅助队列
    ∴ O(k)

难点

  • 如何求一个区间里的最大值
    1. 暴力方法
    2. 一个队列:随着窗口的移动,队列也一进一出;每次移动后,队列告诉我们这里面的最大值是什么(详见上述分析)

总结

学习到新知识:单调序列的特点与使用。

本题的写法仅适用于本题。单调队列不是一成不变的,不同场景不同写法,总之要保证队列里单调递增或递减的原则,所以叫做单调队列。

相关推荐
暴力求解6 分钟前
C++类和对象(上)
开发语言·c++·算法
JKHaaa13 分钟前
几种简单的排序算法(C语言)
c语言·算法·排序算法
让我们一起加油好吗19 分钟前
【基础算法】枚举(普通枚举、二进制枚举)
开发语言·c++·算法·二进制·枚举·位运算
FogLetter37 分钟前
微信红包算法揭秘:从随机性到产品思维的完美结合
算法
YGGP1 小时前
吃透 Golang 基础:数据结构之 Map
开发语言·数据结构·golang
BUG收容所所长1 小时前
二分查找的「左右为难」:如何优雅地找到数组中元素的首尾位置
前端·javascript·算法
weixin_419658311 小时前
数据结构之栈
数据结构
图先1 小时前
数据结构第一章
数据结构
itsuifengerxing2 小时前
python 自定义无符号右移
算法
猎板PCB厚铜专家大族2 小时前
高频 PCB 技术发展趋势与应用解析
人工智能·算法·设计规范