[算法解腻] 剑指 Offer 59 - I. 滑动窗口的最大值

113 阅读1分钟

题目: 给定一个数组 nums 和滑动窗口的大小 k,请找出所有滑动窗口里的最大值。

leetcode.cn/problems/hu…

示例:

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

提示: 你可以假设 k 总是有效的,在输入数组 不为空 的情况下,1 ≤ k ≤ nums.length。

const maxSlidingWindow = (nums, k) => {
  const len = nums.length;
  const queue = [];
  const ans = [];
  
  // 创建初始窗口
  for (let i = 0; i < k; i++) {
     // 队列尾部元素对比,保持队列头部值最大
     while (queue.length > 0 && nums[i] >= nums[queue[queue.length - 1]]) {
        queue.pop();
     }
     queue.push(i);
  }
  // 获取第一个滑动窗口最大值
  ans.push(nums[queue[0]]);
  
  for (let j = k; j < len; j++) {
     // 窗口继续滑动
     while (queue.length > 0 && nums[j] >= nums[queue[queue.length - 1]]) {
        queue.pop();
     }
     // 追加元素
     queue.push(j);
     // 移除队列头部不在滑动窗口的元素
     while (j - k >= queue[0]) {
       queue.shift();
     }
     // 将队列头部也就是该区间最大的元素加入答案
     ans.push(nums[queue[0]]);
  }
  
  return ans;
}