题目
239. 滑动窗口最大值
给你一个整数数组 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]
示例 3:
输入:nums = [1,-1], k = 1
输出:[1,-1]
示例 4:
输入:nums = [9,11], k = 2
输出:[11]
示例 5:
输入:nums = [4,-2], k = 2
输出:[4]
解法
思路
单调递减双端队列。逻辑见代码注释
由于该队列需要从头弹出,从尾弹出,从尾推入,而且还需要递减,所以是单调递减双端队列。
代码如下
/**
* @param {number[]} nums
* @param {number} k
* @return {number[]}
*/
var maxSlidingWindow = function(nums, k) {
if(nums == null || nums.length == 1) return nums;
// 申明一个单调递减的双向队列,用于存储数据的下标
let queue = [];
// 结果数组
let result = [];
// 遍历nums数组
for(let i = 0;i < nums.length;i++){
// 比较当前元素nums[i]和队列的最后一项last,如果nums[i]>=nums[i],则弹出队列最后一项
// 循环比较,直到nums[i]小于最后一项为止
// 正常来说,我们应该推入一个数,并排序,然后弹出最小的一个数
// 但是由于我们要取最大的数,如果队列的数小于当前元素的话,这些数肯定不会是结果,在队列里还占着位置,直接弹出,时间复杂度还低。
while(queue.length>0 && nums[queue[queue.length-1]] <= nums[i]){
queue.pop();
}
// 然后将当前元素的下标推入队列
queue.push(i);
// 由于窗口每次会向右移动一格,我们需要判断我们队首最大的元素是否还在窗口内
// 如果不在则要弹出队首元素
if(queue[0] <= i-k){
queue.shift();
}
// 按我们这个遍历的模式,开始窗口长度为0,以此增加,那么只有当窗口长度等于k以后,我们才开始取队首的最大值当结果
if(i+1 >= k){
result.push(nums[queue[0]]);
}
}
return result;
};