剑指:滑动窗口的最大值

题目描述

给定一个数组和滑动窗口的大小,请找出所有滑动窗口里的最大值。

例如,如果输入数组 [2, 3, 4, 2, 6, 2, 5, 1] 及滑动窗口的大小 3,那么一共存在 6 个滑动窗口,它们的最大值分别为 [4, 4, 6, 6, 6, 5]

注意:

  • 数据保证 k 大于 0,且 k 小于等于数组长度。

样例

输入:[2, 3, 4, 2, 6, 2, 5, 1] , k=3

输出: [4, 4, 6, 6, 6, 5]


解法

利用双向队列,保证队列头部存放的是最大值的下标,当队列头部下标过期时弹出。

细节:

  • 当数组元素小于队列头部下标对应的元素时,在队列尾部中插入数组元素下标。(如果队列尾部有比该元素小的元素,先弹出,再插入。)
  • 当数组元素大于或等于队列头部下标构成的元素时,弹出元素直至队列为空,再插入数组元素下标。
import java.util.*;
public class Solution {
    /**
用一个双端队列,队列第一个位置保存当前窗口的最大值,当窗口滑动一次
1.判断当前最大值是否过期
2.新增加的值从队尾开始比较,把所有比他小的值丢掉
*/
    public ArrayList<Integer> maxInWindows(int [] num, int size)
    {
        if(num==null || num.length==0 || size<=0 || size>num.length){
            return new ArrayList<Integer>();
        }
        
        ArrayList<Integer> res = new ArrayList<>();
        
        LinkedList<Integer> qmax = new LinkedList<>();
        
        for(int i=0; i<num.length; i++){
            while(!qmax.isEmpty() && num[qmax.peekLast()] < num[i]){
                qmax.pollLast();
            }
            qmax.addLast(i);
            
            //判断队首元素是否过期
            if(qmax.peekFirst() == i-size){
                qmax.pollFirst();
            }
            
            if(i+1>=size){
                res.add(num[qmax.peekFirst()]);
            }
        }
        
        return res;
    }
}
 
原文地址:https://www.cnblogs.com/lisen10/p/11415610.html