题目描述
给你一个整数数组 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
var maxSlidingWindow = function(nums, k) {
const q = []; // 定义一个队列,为了取值方便
const n = nums.length;
for(let i = 0; i < k; i++){
// 队列不为空,并且当前元素是大于数组里面其他元素
while(q.length && nums[i] >= nums[q[q.length - 1]]){
q.pop();
}
q.push(i);
}
const ans = [nums[q[0]]];
for(let i = k; i < n; i++){
// 保证最大的值的下标在q内
while(q.length && nums[i] >= nums[q[q.length - 1]]){
q.pop(); // [循环]删除数组的最后一个元素
}
q.push(i);
console.log('q', q);
// 判断当前最大值是否在滑动窗口里面
while(q[0] <= i - k){
q.shift(); // 删除数组的第一个元素
}
// console.log('ans', ans);
ans.push(nums[q[0]]);
}
return ans;
};
代码2
var maxSlidingWindow = function(nums, k) {
let c = [];
for(let i = 0; i < nums.length - k + 1; i++){
let ans = nums.slice(i, i + k);
c.push(Math.max(...ans));
}
return c;
}