题目
给定一个数组和滑动窗口的大小,请找出所有滑动窗口里的最大值。
例如
输入数组{2,3,4,2,6,2,5,1}及滑动窗口大小3
输出最大值{4,4,6,6,6,5}
思路
思路一
遍历所有的滑动窗口,找出其中的最大值。
在滑动窗口找最大值的时间复杂度为O(k)
遍历所有的滑动窗口时间复杂度为O(n)
总的时间复杂度为O(nk)
思路二
由于滑动窗口的插入和删除符合先进先出的特点,所以可以考虑用队列来保存滑动窗口的值。
在遍历整个数组的过程中,只在队列总插入滑动窗口的最大值以及有可能是滑动窗口最大值的数。
例如,对于数组{2,3,4,2,6,2,5,1}
遍历到2,由于队列为空,将2插入队列
遍历到3,由于3大于2,所以将2移除队列,将3插入队列。
遍历到4,由于4是窗口最大值,将3出,将4插入。
遍历到2,由于4依然是窗口最大值,然而2有可能是下个窗口最大值,将2插入队尾。
遍历到6,由于6是窗口最大值,没有能是下个窗口的最大值。所以依次将2,4移出队列,将6插入队列。
遍历到2,由于2有可能是下个窗口最大值,将2插入队尾。
遍历到5,由于窗口最大值依然是6,5有可能是下个窗口最大值,将2移除队列,将5插入队尾。
遍历到1,由于此时窗口的最大值是5,1有可能是下个窗口最大值,所以将6从队列的头部移出,将1插入队列的尾部。
发现一个规律,窗口的最大值总是在双向队列头部。
代码
public static List<Integer> maxInWindows(int[] datas,int size){
List<Integer> result = new ArrayList<>();
if(datas.length >= size && size >= 1){
Deque<Integer> index = new ArrayDeque<>();
for(int i = 0;i < size;i++){
while(!index.isEmpty() && datas[i] >= datas[index.getLast()])
index.removeLast();
index.addLast(i);
}
for(int i = size;i < datas.length;i++){
result.add(datas[index.getFirst()]);
while(!index.isEmpty() && datas[i] >= datas[index.getLast()])
index.removeLast();
if(!index.isEmpty() && index.getFirst() <= (int)(i - size))
index.removeFirst();
index.addLast(i);
}
result.add(datas[index.getFirst()]);
}
return result;
}
总结
在Java中可以用ArrayDeque来当作双向队列。