【栈和队列】滑动窗口的最大值

此题出自牛客网的剑指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]}。

题解如下

import java.util.ArrayList;
import java.util.LinkedList;
/*
思路：滑动窗口应当是队列，但为了得到滑动窗口的最大值，队列序可以从两端删除元素，因此使用双端队列。
 *     原则：
 *     对新来的元素k，将其与双端队列中的元素相比较
 *     1）前面比k小的，直接移出队列（因为不再可能成为后面滑动窗口的最大值了!）,
 *     2）前面比k大的X，比较两者下标，判断X是否已不在窗口之内，不在了，直接移出队列
 *     队列的第一个元素是滑动窗口中的最大值
*/
public class Solution {
    public ArrayList<Integer> maxInWindows(int [] num, int size)
    {
        ArrayList<Integer> result = new ArrayList<>();
        if(size>num.length || num == null || size <= 0){
            return result;
        }
        LinkedList<Integer> indexQueue = new LinkedList<>();
        //窗口的初始化
        for(int i=0;i<size-1;i++){
            //保证队列头永远是当前窗口最大值的下标
            while(!indexQueue.isEmpty() && num[i]>num[indexQueue.getLast()]){
                indexQueue.removeLast();
            }
            indexQueue.add(i);
        }
        
        //进行迭代
        for(int i=size-1;i<num.length;i++){
            //保证队列头永远是当前窗口最大值的下标
            while(!indexQueue.isEmpty() && num[i]>num[indexQueue.getLast()]){
                indexQueue.removeLast();
            }
            indexQueue.add(i);
            //判断是否已经不在窗口内
            if(i - indexQueue.getFirst() +1 >size){
                //如果已经不在此窗口，则将其移出
                indexQueue.removeFirst();
            }
            result.add(num[indexQueue.getFirst()]);
        }
        //返回结果集
        return result;
    }
}