leetcode-239-滑动窗口最大值

花花生

于 2023-10-02 20:11:37 发布

阅读量248

点赞数

分类专栏： # LeetCode小花园文章标签： leetcode 算法 javascript

本文链接：https://blog.csdn.net/A_D_E/article/details/133497472

版权

LeetCode小花园专栏收录该内容

61 篇文章 0 订阅

订阅专栏

题意描述：

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

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

示例：

输入：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

输入：nums = [1], k = 1
输出：[1]

解题思路：
Alice: 你咋想起来做这道了？
Bob: 我其实是在做另一道题目，多重背包的优化之单调队列优化，但是我不知道什么是单调队列，所以我需要先做一下这道题，学习一些单调队列。
Alice: 所有，有什么思路吗？
Bob: 学习的话，想个十分钟吧，然后看题解得了。
15-minutes-later
Alice: 好吧，题解看明白了吗？
Bob: 有点迷，我还是先上手写一下吧。
Alice: 没通过吧，有几个概念你要先了解。队列知道吧，先进先出。双端队列知道吗？两边都能进，两边都能出。你老用 JavaScript 的话，应该知道数组的 push pop shift unshift 这些方法吧，这就是双端队列。
Bob: 然后呢？
Alice：你还得知道单调队列，还有这题为啥需要单调队列来解。单调队列就是，队列中的值是单调的，单调就是数学上的那个单调，单调递增或者单调递减或者单调不增或者单调不减。
Bob: 那为啥要用单调队列呢？
Alice: 还是得读题啊。这题不是让求解滑动窗口的最大值吗？然后数组长度最大 10^5，k 的取值最大也是 10^5，按照最大的计算量 k取最大长度的一半也就是 5*10^4 然后每次都求最大值，计算量也就是 10^5 * (5 * 10^4)^2也就是 2.5 * 10^14, 时间限制是 1s 的话，一定会超时的。
Bob: 所以我们需要单调队列？
Alice: 差不多，单调队列非常适合这道题，这道题是让求 k 时间窗口的最大值，单调队列的头或尾一定是最大值。我们只需要在窗口滑动的过程中不断地维护队列就可以了。
Bob: 怎么维护呢？
Alice: 这里有两个点需要注意。1是要在维护队列的过程中不断干掉超出 k 时间窗口的值，2是要及时干掉那些不需要的值，比如说 nums[i] > nums[i-1] 的时候，nums[i-1] 就没必要继续留在队列中了，要求的是最大值 nums[i] 比 nums[i-1] 更大且更 ”年轻“ ，求最大值的时候 nums[i] 一定会覆盖 nums[i-1]。当然这里不止 i-1 前面 ”更老更小“ 的都应该干掉。
Bob: 所以我们的数据结构中还应该有下标，通过下标来计算某个值是否超出 k 时间窗口？
Alice: 对的，而另一个维护队列的时机就是在新元素入队之前，或者入队的时候。
Bob: 还有什么要注意的吗？
Alice: 有的，有两个点，一是初始化的时候也要维护单调队列，还有就是每次维护其实是在队首和队尾都要维护，队尾要干掉那些不必要的，队首要保证在 k 时间窗口内最大。
Bob: k 时间窗口最大我是这样理解的，队尾的维护会把比较小的干掉，这样队首元素出去之后，从队尾补上来的，应该总是剩下的最大的。
Alice: 对的。
Bob: 还是挺有技巧的，上代码吧。

代码：

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number[]}
 */
var maxSlidingWindow = function(nums, k) {

  const ans = []
  // 初始化 queue
  const queue = [];
  for(let i=0; i<k; ++i) {
    // 维护队尾，去除不必要的又老又小的值
    while (queue.length > 0 && queue[queue.length-1][0] <= nums[i]) {
      queue.pop();
    }
    // 新大值入队
    queue.push([nums[i], i]);
  }

  ans.push(queue[0][0]);

  for (let i=k; i<nums.length; ++i) {
    // 维护队尾，去除不必要的又老又小的值
    while (queue.length > 0 && queue[queue.length-1][0] <= nums[i]) {
      queue.pop();
    }
    // 新大值入队
    queue.push([nums[i], i]);
    // 维护队首
    while (queue.length > 0 && queue[0][1] + k <= i) {
      queue.shift();
    }

    ans.push(queue[0][0]);
  }

  return ans;
};