滑动窗口的最大值
题目描述
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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]}。
分析
窗口相当于一个队列,每滑动一次,队首减去一个元素,队尾加上一个元素。
使用一个双端队列q来辅助记录最大值的下标。
- 当q为空时,新元素肯定是最大的,直接入队;
- 当q不为空时,将新元素与q中的元素从后向前比较,如果q中的元素小于新元素则出队,直到q中的元素大于新元素或者q为空后,新元素入队;
- 判断队首的元素是否已过期,这就是为什么在队列中保存下标而不是保存数值的作用。若队首的元素的下标 + 窗口尺寸size = 新元素的下标i,说明队首元素已不在窗口中(尾-头=size-1),故从队列q中删除队首元素;
- 如果新元素i >= size-1,说明窗口中是满元素的,此时q首的元素就是最大值。
示例如下:
输入数组{2,3,4,2,6,2,5,1},输出为{4,4,6,6,6,5}。
Java代码
import java.util.LinkedList;
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> maxInWindows(int [] num, int size)
{
ArrayList<Integer> res = new ArrayList<Integer>();
if(num==null || num.length==0 || size<=0 || size>num.length)
return res;
LinkedList<Integer> q = new LinkedList<Integer>();
for(int i=0; i<num.length; i++){
// 新元素入队
while(!q.isEmpty() && num[q.peekLast()]<num[i])
q.pollLast();
q.addLast(i);
// 判断队首元素是否过期
if(i == q.peekFirst()+size)
q.pollFirst();
// 将队首元素保存为最大值
if