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

总结

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

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

相关推荐
Codeking__24 分钟前
链表算法综合——重排链表
网络·算法·链表
minji...1 小时前
数据结构 堆(4)---TOP-K问题
java·数据结构·算法
技术卷1 小时前
详解力扣高频SQL50题之610. 判断三角形【简单】
sql·leetcode·oracle
AI_Keymaker2 小时前
一句话生成3D世界:腾讯开源混元3D模型
算法
Leon_vibs2 小时前
当 think 遇上 tool:深入解析 Agent 的规划之道
算法
旧时光巷2 小时前
【机器学习-2】 | 决策树算法基础/信息熵
算法·决策树·机器学习·id3算法·信息熵·c4.5算法
落羽的落羽2 小时前
【C++】论如何封装红黑树模拟实现set和map
数据结构·c++·学习
落了一地秋2 小时前
4.5 优化器中常见的梯度下降算法
人工智能·算法·机器学习
前端伪大叔2 小时前
第 5 篇:策略参数怎么调优?Freqtrade hyperopt 快速入门
算法·代码规范
Code季风2 小时前
深入理解令牌桶算法:实现分布式系统高效限流的秘籍
java·算法·微服务