题目
描述
给定一个长度为 n 的数组 nums 和滑动窗口的大小 size ,找出所有滑动窗口里数值的最大值。
例如,如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小3,那么一共存在6个滑动窗口,他们的最大值分别为{4,4,6,6,6,5}; 针对数组{2,3,4,2,6,2,5,1}的滑动窗口有以下6个: {[2,3,4],2,6,2,5,1}, {2,[3,4,2],6,2,5,1}, {2,3,[4,2,6],2,5,1}, {2,3,4,[2,6,2],5,1}, {2,3,4,2,[6,2,5],1}, {2,3,4,2,6,[2,5,1]}。
数据范围:1≤size≤n≤10000,数组中每个元素的值满足∣val∣≤10000
要求:空间复杂度 O(n)O(n),时间复杂度 O(n)O(n)
示例1
输入:[2,3,4,2,6,2,5,1],3
返回值:[4,4,6,6,6,5]
示例2
输入:[9,10,9,-7,-3,8,2,-6],5
返回值:[10,10,9,8]
示例3
输入:[1,2,3,4],3
返回值:[3,4]
思路
通过维护一个值递减的存数组下标的双头队列,队首下标的值始终为最大值,队首后的下标的值为以后可能出现的最大值
- 定义一个双头队列que和一个结果数组res
- 循环遍历此数组num
- 当维护的最大值不在滑动窗口中时,将队首元素出队
- 判断后续进入滑动窗口的数是否大于维护的最大值(即que的队首元素),若是,则清空队列
- 判断后续进入滑动窗口的数是否大于队首后的值(即后续可能成为的最大值),若是,则将小于此值的值出队,即pop
- 将可能成为滑动窗口最大值的值入队(即push)
- 将每个滑动窗口中维护的最大值加入结果队列中
代码
function maxInWindows(num, size)
{
// write code here
let que = []//定义一个存最大值下标的双头队列,队首为最大值
let res = []//存窗口最大值的数组
const n = num.length
for(let i =0;i<n;i++){
if(i-que[0] >= size){//当维护的最大值不在窗口范围内时,将队首出队
que.shift()
}
while(que.length>0 && num[que[0]]<=num[i]){//当新入队的值为最大值时,将队列清空
que.shift()
}
while(que.length>0 && num[que[que.length-1]]<=num[i]){//当新入队的值比可能成为最大值的值大时,将比此值小的值出队
que.pop()
}
que.push(i)//将此值入队
if(i>=size-1){//将移动窗口中维护的最大值加入结果数组中
res.push(num[que[0]])
}
}
return res
}
module.exports = {
maxInWindows : maxInWindows
};