题目描述
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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]}。
思路:
这道题卡了我很久,搜了一下,答案都看了很久才看懂,下面跟大家详细的讲讲吧。
我是使用的一个双端队列来做的,
1. 判断下边界及特殊情况,size<num.length的情况
2.如果队尾的元素小于即将插入的元素,则删除队尾的元素,因为接下来的遍历中不要要他们了
3,看看队头的元素是不是还在我们这个双端队列中,如果不在,删除
4,当前最大的一个元素就是在队头,将它添加进就行了
下面贴上Java代码:
public ArrayList<Integer> maxInWindows(int [] num, int size)
{
ArrayList<Integer> result = new ArrayList<Integer>();
if(num == null){ return result; }
if(num.length < size || size < 1){ return result; }
LinkedList<Integer> indexDeque = new LinkedList<Integer>();
// 首先遍历窗口大小size个元素,然后再一次向后遍历
for (int i = 0; i < size-1; i++) {
while (!indexDeque.isEmpty() && num[i]>num[indexDeque.getLast()]){
indexDeque.removeLast();
}
indexDeque.addLast(i);
}
// 开始遍历之后的序列了,很开心,这时候可以一次想result中加入结果了,哈哈
for (int i = size-1; i < num.length; i++) {
// 每次先删除双端队列的结尾上比num[i]小的数
while (!indexDeque.isEmpty() && num[i]>num[indexDeque.getLast()]){
indexDeque.removeLast();
}
indexDeque.addLast(i);
// 删除已经不在我们这个窗口中的元素的下标
if(i-size+1>indexDeque.getFirst()){
indexDeque.removeFirst();
}
result.add(num[indexDeque.getFirst()]);
}
return result;
}