题目描述
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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]}。
算法解析:
这个算法利用双端队列实现,我们拿上边这个数组举例子,首先先塞满整个窗口,即我们首先先遍历到满足窗口大小的位置,然后将最可能称为最大值的放到队列的头,也就是说当我们遍历到一个数组值的时候,如果队头的值小于当前值,我们移除队头,队尾值小于当前值,我们移除队尾,队列头永远存放当前滑动窗口的最大值。
代码如下:
public ArrayList<Integer> maxInWindows(int [] num, int size)
{
ArrayList<Integer> result = new ArrayList<>();
if (num.length >= size && size >= 1){
Deque<Integer> doubleQueue = new ArrayDeque<>();
for (int i = 0; i < size; i++) {
while (!doubleQueue.isEmpty() && num[i] >= num[doubleQueue.peekFirst()]){
doubleQueue.pollFirst();
}
while (!doubleQueue.isEmpty() && num[i] >= num[doubleQueue.peekLast()]){
doubleQueue.pollLast();
}
doubleQueue.addLast(i);
}
for (int i = size; i < num.length; i++) {
result.add(num[doubleQueue.peekFirst()]);
while (!doubleQueue.isEmpty() && num[i] >= num[doubleQueue.peekFirst()]){
doubleQueue.pollFirst();
}
while (!doubleQueue.isEmpty() && num[i] >= num[doubleQueue.peekLast()]){
doubleQueue.pollLast();
}
if (!doubleQueue.isEmpty() && doubleQueue.peekFirst() <= (i - size)){
doubleQueue.pollFirst();
}
doubleQueue.addLast(i);
}
result.add(num[doubleQueue.peekFirst()]);
}
return result;
}