[Leetcode] 239. 滑动窗口最大值
题目链接:239. 滑动窗口最大值
思路:
这是一道典型的单调队列题目,遍历数组,将下标存放在双向队列中,并用 L,R 来标记窗口的左边界和右边界。我们要保证队列中的下标对应的数组中的数要从大到小排序。如果当前遍历的数比队尾的值大,则需要弹出队尾值,直到队列重新满足从大到小的要求。刚开始遍历时,L 和 R 都为 0,有一个形成窗口的过程,此过程没有最大值,L 不动,R 向右移。当窗口大小形成时,L 和 R 一起向右移,每次移动时,判断队首的值的数组下标是否在 [L,R] 中,如果不在则需要弹出队首的值,当前窗口的最大值即为队首的数。
题目例子:
输入: nums = [1,3,-1,-3,5,3,6,7], 和 k = 3
输出: [3,3,5,5,6,7]
解释过程中队列中都是具体的值,方便理解,具体见代码。
初始状态:L=R=0,队列:{}
i=0,nums[0]=1。队列为空,直接加入。队列:{1}
i=1,nums[1]=3。队尾值为1,3>1,弹出队尾值,加入3。队列:{3}
i=2,nums[2]=-1。队尾值为3,-1<3,直接加入。队列:{3,-1}。此时窗口已经形成,L=0,R=2,result=[3]
i=3,nums[3]=-3。队尾值为-1,-3<-1,直接加入。队列:{3,-1,-3}。队首3对应的下标为1,L=1,R=3,有效。result=[3,3]
i=4,nums[4]=5。队尾值为-3,5>-3,依次弹出后加入。队列:{5}。此时L=2,R=4,有效。result=[3,3,5]
i=5,nums[5]=3。队尾值为5,3<5,直接加入。队列:{5,3}。此时L=3,R=5,有效。result=[3,3,5,5]
i=6,nums[6]=6。队尾值为3,6>3,依次弹出后加入。队列:{6}。此时L=4,R=6,有效。result=[3,3,5,5,6]
i=7,nums[7]=7。队尾值为6,7>6,弹出队尾值后加入。队列:{7}。此时L=5,R=7,有效。result=[3,3,5,5,6,7]
代码(Java):
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
if(nums == null) return null;
Deque<Integer> q = new ArrayDeque<>();
int[] res = new int[nums.length - k + 1];
int index = 0; // 结果数组的下标
for(int i = 0; i < nums.length; i ++) {
// 去除多出窗口的元素
if(!q.isEmpty() && q.peekFirst() <= i - k) q.pollFirst();
// 保持队列的单调性,依次从队尾弹出比当前小的数
while(!q.isEmpty() && nums[i] >= nums[q.peekLast()]) q.pollLast();
// 添加当前值对应的数组下标
q.addLast(i);
// 当窗口长度为k时,保存当前窗口中最大值,最大值就是队首
if(i >= k - 1) {
res[index ++] =nums[q.peekFirst()];
}
}
return res;
}
}