滑动窗口最大值

开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第12天，点击查看活动详情

239. 滑动窗口最大值 - 力扣（LeetCode）

给你一个整数数组 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

示例 2：

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

提示：

• 1 <= nums.length <= 10^5
• -10^4 <= nums[i] <= 10^4
• 1 <= k <= nums.length

思路

本题最直观的解法就是按照题目说的，不断滑动窗口，然后遍历窗口内的数字，找到最大值，这种方法虽然可行，但效率较低，不是最优解。

我们还可以使用单调队列求解。我们遍历nums，维护一个下标距离最大为k-1的单调递减队列，设当前移入窗口的元素为num，下标为i，我们先从队列头开始遍历，把下标小于i - k - 1的数字移除队列，然后再从队列尾开始遍历，把小于等于num的数字移出队列，最后把num放入队列尾。队首元素就是当前窗口的最大值。

解题

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number[]}
 */
var maxSlidingWindow = function (nums, k) {
  if (k === 1) return nums;
  const n = nums.length;
  let res = new Array(n - k + 1).fill(0);
  let idxs = [];
  for (let i = 0; i < n; i++) {
    const num = nums[i];
    let idx = i - k + 1;
    while (idxs.length && idxs[0] < idx) {
      idxs.shift();
    }
    while (idxs.length && nums[idxs.at(-1)] <= num) {
      idxs.pop();
    }
    idxs.push(i);
    if (idx >= 0) {
      res[idx] = nums[idxs[0]];
    }
  }
  return res;
};