给定一个数组和滑动窗口的大小,请找出所有滑动窗口里的最大值。
例如,如果输入数组 [2, 3, 4, 2, 6, 2, 5, 1]
及滑动窗口的大小 3,那么一共存在 6 个滑动窗口,它们的最大值分别为 [4, 4, 6, 6, 6, 5]
。
注意:
样例
输入:[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