题目描述
给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回 滑动窗口中的最大值 。
示例图
输入: 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
解法
1.首先建立一个数组ans存储最后的答案,建立一个双端队列deque用来存储数据的位置,且位置对应的数据是单调递减的,位置下标是单调递增的。
2.先遍历第一个窗口,每遍历一个值,在将他的位置插入前,从队尾开始判断,小于等于它的值的位置均弹出,最后得到一个初始的deque,队首就是第一个窗口的最大值的位置,将其对应的值插入到ans。
3.假设一个窗口的两边指针是start、end,则遍历下一个窗口时,起始位置改成start+1,所以如果start位置存在于队列,那他一定是在队首,这是要将start位置弹出。然后在插入end+1位置前,也要先弹出前方比他小的数的位置。此时队首位置对应的数据就是当前窗口最大的值的位置。
过程
nums = [1,3,-1,-3,5,3,6,7], k = 3
代码
var maxSlidingWindow = function(nums, k) {
const deque = []
const ans = []
for(let i = 0; i < k; i++) {
while(deque.length > 0 && nums[i] >= nums[deque[deque.length-1]]) deque.pop()
deque.push(i)
}
ans.push(nums[deque[0]])
for(let i = k; i < nums.length; i++) {
while(deque.length > 0 && nums[i] >= nums[deque[deque.length-1]]) deque.pop()
if(deque.length > 0 && i-k >= deque[0]) deque.shift()
deque.push(i)
ans.push(nums[deque[0]])
}
return ans
};