代码随想录-栈与队列 | 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. 一个队列:随着窗口的移动,队列也一进一出;每次移动后,队列告诉我们这里面的最大值是什么(详见上述分析)

总结

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

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

相关推荐
hsling松子1 小时前
使用PaddleHub智能生成,献上浓情国庆福
人工智能·算法·机器学习·语言模型·paddlepaddle
dengqingrui1232 小时前
【树形DP】AT_dp_p Independent Set 题解
c++·学习·算法·深度优先·图论·dp
C++忠实粉丝2 小时前
前缀和(8)_矩阵区域和
数据结构·c++·线性代数·算法·矩阵
ZZZ_O^O3 小时前
二分查找算法——寻找旋转排序数组中的最小值&点名
数据结构·c++·学习·算法·二叉树
CV-King3 小时前
opencv实战项目(三十):使用傅里叶变换进行图像边缘检测
人工智能·opencv·算法·计算机视觉
代码雕刻家3 小时前
数据结构-3.9.栈在递归中的应用
c语言·数据结构·算法
雨中rain3 小时前
算法 | 位运算(哈希思想)
算法
Kalika0-05 小时前
猴子吃桃-C语言
c语言·开发语言·数据结构·算法
代码雕刻家5 小时前
课设实验-数据结构-单链表-文教文化用品品牌
c语言·开发语言·数据结构
sp_fyf_20245 小时前
计算机前沿技术-人工智能算法-大语言模型-最新研究进展-2024-10-02
人工智能·神经网络·算法·计算机视觉·语言模型·自然语言处理·数据挖掘