剑指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.PriorityQueue;
import java.util.Comparator;
public class Solution {
//大顶堆,从大到小的排列
private PriorityQueue<Integer> heap=new PriorityQueue<Integer>(15,new Comparator<Integer>(){
@Override
public int compare(Integer o1,Integer o2){
return o2-o1;
}
});
public ArrayList<Integer> maxInWindows(int [] num, int size)
{
//存放结果集
ArrayList<Integer> result=new ArrayList<Integer>();
if(num==null||num.length==0||num.length<size||size<=0){
return result;
}
for(int i=0;i<size;i++){
heap.offer(num[i]);
}
result.add(heap.peek());
//i从1到num.length-size
for(int i=1;i<num.length+1-size;i++){
heap.remove(num[i-1]);
heap.offer(num[i+size-1]);
result.add(heap.peek());
}
return result;
}
}
解决办法二:
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> maxInWindows(int [] num, int size)
{
ArrayList<Integer> result = new ArrayList<Integer>();
if(num == null||size == 0 ||size>num.length){
return result;
}
int max = num[0];//记录最大值
int index = -1;//记录最大值下标,防止最大值过期
int len = num.length;
int plow = 0,phight = size-1;//滑动窗口边界
while(phight<len){//此处不要条件plow<phight,因为有可能size=1
if((index>=plow)&&(index<=phight)){//保证最大值未过期
if(max<num[phight]){
max = num[phight];
index = phight;
}
}else{
max = num[plow];
index = plow;
for(int i=plow;i<=phight;i++){
if(max<num[i]){
max = num[i];
index = i;
}
}
}
result.add(max);
plow++;
phight++;
}
return result;
}
}

浙公网安备 33010602011771号