剑指offer:求滑动窗口的最大值

题目叙述:

给定一个数组和滑动窗口的大小，找出所有滑动窗口里数值的最大值。例如，如果输入数组{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]}。

方法一：

很简单得到结果集的大小是num.length-size+1。使用双层循环求每一个滑动窗口的最大值。

import java.util.ArrayList;
public class Solution {
    public ArrayList<Integer> maxInWindows(int [] num, int size)
    {
       //结果集合的个数是num.length-size+1
       ArrayList<Integer> res = new ArrayList<>();
       if(num.length == 0 || size == 0) return res;
       if((num.length-size+1)<=0) return res;
       //Stack<Integer> s = new Stack<Integer>();
       for(int i=0;i<num.length-size+1;i++){
           int end = i+size;
           int max = num[i];
           for(int j =i+1;j<end;j++){
               if(num[j]>max) max = num[j];
           }
           res.add(max);
       }
       return res;
    }
}

方法二:

方法一的时间复杂度是O($n^2$)。考虑使用双端队列
本题也可以使用两个栈来维持一个队列。

import java.util.ArrayList;
import java.util.Deque;
import java.util.LinkedList;
public class Solution {
    public ArrayList<Integer> maxInWindows(int [] num, int size)
    {
       //结果集合的个数是num.length-size+1
       ArrayList<Integer> res = new ArrayList<>();
       if(num.length == 0 || size == 0) return res;
       if((num.length-size+1)<=0) return res;
       //考虑使用双端队列：Deque
       //队列中存放的是每个数组元素的下标
       Deque<Integer> deque = new LinkedList<>();
       for(int i=0;i<num.length;i++){
           //从后面依次弹出队列中比当前num值小的元素，同时也能保证队列首元素为当前窗口最大值下标
           while(deque.size()>0&&num[deque.peekLast()]<=num[i])
               deque.pollLast();
           //当前窗口移出队首元素所在的位置，即队首元素坐标对应的num不在窗口中，需要弹出
           while(deque.size()>0&&i-deque.peekFirst()+1>size)
               deque.pollFirst();
           deque.offerLast(i);//把每次滑动的num下标加入队列
           if(size>0 && i+1>=size) res.add(num[deque.peekFirst()]);//当滑动窗口首地址i大于等于size时才开始写入窗口最大值
        }         
       return res;       
    }
}