题目描述:
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。
示例:
如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小3, 针对数组{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]}
他们的最大值分别为{4,4,6,6,6,5};
思路:
(1)我自己的解法:
- 假定原数组为a,定义一个长度为3的数组b,将数组a每次向右移动一位,从当前位起到后2位赋值给数组b,
- 通过Math.max()找到b数组最大值并保存下来。
public ArrayList<Integer> MaxInWindows(int [] num, int size){ int b[]=new int[size]; int a=0,c=0; ArrayList<Integer> list=new ArrayList(); int n=num.length; for(int i=0;i<n-(size-1);i++){ for(int j=0;j<size;j++){ b[j]=num[i+j]; c=Math.max(b[j],c); } list.add(c); c=0; } return list; }
(2)
我们可以使用双向队列(Linklist),队列中只存放当前元素的下标,设新来的元素为k,如果前面的元素比k小,直接把前面的删除(因为不可能成为后面窗口的最大值),如果前面的元素比k大,判断是否还在窗口范围内,不在则移除
①先判断当前队列是否为空,如果不空而且当前元素比队列中尾端的元素大,将队列元素的尾端弹出;
②判断队列头元素(存的是下标)是否还在滑动窗口范围中,不在则把头元素移除;
public ArrayList<Integer> maxInWindows2(int [] num, int size) {
if (num == null || num.length == 0 || size <= 0 || num.length < size) {
return new ArrayList<Integer>();
}
ArrayList<Integer> result = new ArrayList<>();
//双端队列,用来记录每个窗口的最大值下标
LinkedList<Integer> qmax = new LinkedList<>();
for (int i = 0; i < 8; i++) {
while (!qmax.isEmpty() && num[qmax.peekLast()] < num[i]) {
qmax.pollLast(); //获取列表最后一个元素,但是元素被删除。
}
qmax.addLast(i);
//判断队首元素是否过期
if (qmax.peekFirst() == i - 3){
qmax.pollFirst();
}
//向result列表中加入元素
if (i >= 3 - 1) {
result.add(num[qmax.peekFirst()]);
}
}
return result;
}