题目描述
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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]}。
思路
核心思想就是qmax队头维护的是最大元素的下标,如果后一个元素大小超过qmax的最后索引对应的,就把qmax的最后一个扔掉**【排序是从大到小排的】**
两个if的条件理解,i-size表示窗口的第一个值的前一个,说明此时qmax的第一个元素已经出窗口了
size-1表示从第一个窗口开始向result中加入值,刚开始并没有达到窗口的大小
代码
import java.util.ArrayList;
import java.util.LinkedList;
public class Solution {
public ArrayList<Integer> maxInWindows(int [] num, int size)
{
//保存结果
ArrayList<Integer> res = new ArrayList<>();
//双端队列,用于保存每个窗口的最大值下标
LinkedList<Integer> qmax = new LinkedList<>();
if(num == null || num.length == 0 || size <= 0 || size > num.length) return res;
for(int i = 0;i < num.length;i++){
while(!qmax.isEmpty() && num[qmax.peekLast()] < num[i]){
qmax.pollLast();//这一步保证qmax的第一个元素是最大值的索引
}
qmax.addLast(i);
//判断队首元素(滑动窗口最大的元素)是否已经出窗口了
if(qmax.peekFirst() == i - size){
qmax.pollFirst();
}
//向res中加入元素,一旦达到窗口大小就开始加元素
if(i >= size - 1){
res.add(num[qmax.peekFirst()]);
}
}
return res;
}
}