JS Bin

剑指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;
 
    }
}

  

 

posted @ 2020-03-21 17:09  左五六  阅读(227)  评论(0编辑  收藏  举报