题目描述
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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]}。
思路1:
考虑到每一个滑动窗口都遍历一遍取最大值,使得复杂度过大。每进入下一个滑动窗口,都让前一个窗口中最大值与其起始值进行比较,若最大值>起始值,则说明最大值是产生在后面这些值之间,则只需让该最大值与本窗口中末尾值进行比较,较大者即为本窗口中最大值;若最大值<=起始值,则说明最大值是起始值,则需对本窗口进行遍历寻找最大值.
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> maxInWindows(int [] num, int size)
{
ArrayList<Integer> al=new ArrayList<>();
//size==0的情况容易忽视
if(size==0||num.length<size)return al;
int max=-10000000,len=num.length;
for(int i=0;i<size;i++){
if(num[i]>=max){
max=num[i];
}
}
al.add(max);
for(int i=1;i<=len-size;i++){
if(max>num[i-1]){
max=max>num[i+size-1]?max:num[i+size-1];
}else{
max=-100000000;
for(int j=i;j<i+size;j++){
if(num[j]>=max){
max=num[j];
}
}
}
al.add(max);
}
return al;
}
}
思路2:
用一个双端队列,保证队列第一个位置是当前窗口的最大值,当窗口滑动一次:
1.判断当前最大值是否为前一窗口的首元素,若是,表示该最大值已过期,将其丢掉
2.新增加的值从队尾开始比较,把所有比他小的值丢掉
3.只有滑动窗口大小满足给定大小时,才将队首元素存入列表
import java.util.ArrayList;
import java.util.ArrayDeque;
public class Solution {
public ArrayList<Integer> maxInWindows(int [] num, int size)
{
ArrayList<Integer> res=new ArrayList<>();
if(size==0||size>num.length)return res;
ArrayDeque<Integer> ad=new ArrayDeque<>();
for(int i=0;i<num.length;i++){
int begin=i-size+1;
if(!ad.isEmpty()){
if(ad.peekFirst()<begin){//判断当前最大值是否为前一窗口的首元素
ad.pollFirst();
}
while(!ad.isEmpty()&&num[ad.peekLast()]<=num[i]){//注意点:比较的是数组中两元素
ad.pollLast();
}
}
ad.add(i);
if(begin>=0){
res.add(num[ad.peekFirst()]);//从窗口大小满足给定大小时开始添加最大值.注意添加的不是下标
}
}
return res;
}
}
优化思路二:
省略if(!ad.isEmpty())
import java.util.ArrayList;
import java.util.ArrayDeque;
public class Solution {
public ArrayList<Integer> maxInWindows(int [] num, int size)
{
ArrayList<Integer> res=new ArrayList<>();
int len=num.length;
if(len==0||size==0||size>len)return res;
ArrayDeque<Integer> ad=new ArrayDeque<>();
for(int i=0;i<len;i++){
int begin=i-size+1;
while(!ad.isEmpty()&&num[ad.peekLast()]<num[i]){
ad.pollLast();
}
ad.add(i);
if(ad.peekFirst()<begin){
ad.pollFirst();
}
if(begin>=0){
res.add(num[ad.peekFirst()]);
}
}
return res;
}
}