题目(来源力扣):
给定一个数组 nums 和滑动窗口的大小 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.最大值还在窗口内,这时候只要比较临时变量和刚加进窗口的值即可得到新的窗口的最大值
2.另外一种情况是,最大值刚好被移出窗口,这时候我们就丢失了当前窗口的最大值,如果我们知道当前窗口的第二大的最大值,就可以解决此问题。
因此,将临时变量改为单调递减队列,且是双端队列(即队列两边都可以移出元素),队列的首元素记录的一直是最大值,窗口移动时,加入窗口的元素比队尾元素小,则直接进队,如果比队尾元素大,则遍历将队列种比进队元素小的元素出队,再将加入窗口的元素加入队列,保持队列的单调递减性,对于窗口移出的元素,则判断其是否等于队首元素,是则队首元素移除,窗口的最大值也随之变化。
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
if(nums.length == 0 || k == 0) return new int[0];
Deque<Integer> deque = new LinkedList<>();
int[] res = new int[nums.length - k + 1];
// 未形成窗口
for(int i = 0; i < k; i++) {
while(!deque.isEmpty() && deque.peekLast() < nums[i])
deque.removeLast();
deque.addLast(nums[i]);
}
res[0] = deque.peekFirst();
// 形成窗口后
for(int i = k; i < nums.length; i++) {
if(deque.peekFirst() == nums[i - k])
deque.removeFirst();
while(!deque.isEmpty() && deque.peekLast() < nums[i])
deque.removeLast();
deque.addLast(nums[i]);
res[i - k + 1] = deque.peekFirst();
}
return res;
}
}