题目描述
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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]}。
分析:
LinkedList<Integer> indexDeque = new LinkedList<>();//使用LinkedList做队列,存储的是num的下标
/**
用一个双端队列,队列第一个位置保存当前窗口的最大值,当窗口滑动一次
1.判断当前最大值是否过期
2.新增加的值从队尾开始比较,把所有比他小的值丢掉
*/
参考:https://zhuanlan.zhihu.com/p/34456480
- 窗口大小为w,数组大小为n,初始状态下窗口为空;
- 遍历数组中的前w-1个元素,对每一个元素,执行下列操作:
- 如果窗口最右端(也就是tail部分)的元素小于或等于当前新加入的元素,那么就将窗口最右端的元素移出窗口,直到窗口最右端元素大于当前新加入元素或者窗口为空为止
- 将当前元素加入窗口。
for(int i=0;i<size-1;i++){
while(!indexDeque.isEmpty()&&num[i]>num[indexDeque.getLast()]){//返回列表尾部的最后一个元素
indexDeque.removeLast();//java.util.LinkedList.removeLast() 方法移除并返回此列表的最后一个元素。
}
indexDeque.addLast(i);//addLast(E e) 方法将指定的元素E加到deque队列的末尾
}
上述操作保证了窗口中的最大值始终在窗口的最左端,也即是窗口的头部。下面我们继续描述算法:
- 遍历完前w个元素后,继续遍历数组中的剩余元素,执行下列操作:
- 将窗口中所有小于或等于当前新加入元素的元素移出,将当前元素加入窗口;
- 如果窗口最左端的元素的下标超出窗口的范围,则将其移出窗口;
- 每个时刻窗口中的元素最大值都在窗口的最左端,也就是头部,可以直接输出。
/** * 将窗口中所有小于或等于当前新加入元素的元素移出,将当前元素加入窗口; */ for(int i=size-1;i<num.length;i++){ while(!indexDeque.isEmpty()&& num[i]>num[indexDeque.getLast()]){ indexDeque.removeLast(); } indexDeque.addLast(i); /** * 如果窗口最左端的元素的下标超出窗口的范围,则将其移出窗口 */ if(i-indexDeque.getFirst()+1>size){ indexDeque.removeFirst(); } //每个时刻窗口中的元素最大值都在窗口的最左端,也就是头部,可以直接输出 result.add(num[indexDeque.getFirst()]); } return result; }
完整代码:
import java.util.ArrayList;
import java.util.LinkedList;
public class NewTest {
public static void main(String[] args) {
int[] num={2,3,4,2,6,2,5,1};
NewTest test= new NewTest();
test.maxInWindows(num,4);
}
public ArrayList<Integer> maxInWindows(int [] num, int size)
{
ArrayList<Integer> result = new ArrayList<>();
if(num == null) return result;
if(num.length<size||size<1) return result;
LinkedList<Integer> indexDeque = new LinkedList<>();//使用LinkedList做队列,存储的是num的下标
for(int i=0;i<size-1;i++){//先考虑前size-1 个数字
while(!indexDeque.isEmpty()&&num[i]>num[indexDeque.getLast()]){//返回列表尾部的最后一个元素
indexDeque.removeLast();//java.util.LinkedList.removeLast() 方法移除并返回此列表的最后一个元素。
}
indexDeque.addLast(i);//addLast(E e) 方法将指定的元素E加到deque队列的末尾
}
/**
* 将窗口中所有小于或等于当前新加入元素的元素移出,将当前元素加入窗口;
*/
for(int i=size-1;i<num.length;i++){
while(!indexDeque.isEmpty()&& num[i]>num[indexDeque.getLast()]){
indexDeque.removeLast();
}
indexDeque.addLast(i);
/**
* 如果窗口最左端的元素的下标超出窗口的范围,则将其移出窗口
*/
if(i-indexDeque.getFirst()+1>size){
indexDeque.removeFirst();
}
//每个时刻窗口中的元素最大值都在窗口的最左端,也就是头部,可以直接输出
result.add(num[indexDeque.getFirst()]);
}
return result;
}
}
补充:
Java.util.LinkedList.getLast() 方法用于从LinkedList或列表尾部的元素中获取或检索最后一个元素。
句法:
LinkedList.getLast()
参数:该方法不带任何参数。
返回值:该方法返回列表尾部的最后一个元素或元素。