剑指offer-----滑动窗口的最大值(java版)

来源:互联网 发布:软件打印价格牌 编辑:程序博客网 时间:2024/06/01 07:13

一 题目

给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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]}。一共存在6个滑动窗口,他们的最大值分别为{4,4,6,6,6,5};


三 思路

第1种方法:

使用java的优先队列(也就是堆),我们可以使用最大堆(PriorityQueue默认是最小堆,构建最大堆只需要在构建时传入一个comparator即可),在滑动窗口滑动时,将当前元素加入最大堆(PriorityQueue)中,堆顶的元素即是最大值,同时还要判断当前元素是否存在于当前窗口中,不存在的话弹出,最后就把该元素添加入Arraylist


第2种方法:

我们可以使用双向队列(Linklist),队列中只存放当前元素的下标,设新来的元素为k,如果前面的元素比k小,直接把前面的删除(因为不可能成为后面窗口的最大值),如果前面的元素比k大,判断是否还在窗口范围内,不在则移除

①先判断当前队列是否为空,如果不空而且当前元素比队列中尾端的元素大,将队列元素的尾端弹出;

②判断队列头元素(存的是下标)是否还在滑动窗口范围中,不在则把头元素移除;


四 程序源码

第一种方法:

import java.util.ArrayList;import java.util.Comparator;import java.util.PriorityQueue;/** * @author tanzj * @create 2016-12-24 19:14 */class tmp{    public tmp() {    }    public tmp(Integer num,Integer pos) {        this.pos = pos;        this.num = num;    }    public Integer pos;    public Integer num;    public Integer getNum() {        return num;    }    public void setNum(Integer num) {        this.num = num;    }    public Integer getPos() {        return pos;    }    public void setPos(Integer pos) {        this.pos = pos;    }}public class Solution {    public ArrayList<Integer> maxInWindows(int [] num, int size) {        ArrayList<Integer> arr = new ArrayList<>();        if (num==null)            return arr;        if (num.length<size||size<=0)            return arr;        PriorityQueue<tmp> q = new PriorityQueue<>(100, new Comparator<tmp>() {            @Override            public int compare(tmp o1, tmp o2) {                return o2.num-o1.num;            }        });        for (int i=0;i<size-1;i++)            q.offer(new tmp(num[i],i));        for (int i=size-1;i<num.length;i++){            q.offer(new tmp(num[i],i));            tmp p = q.peek();            while (p.getPos()<i-(size-1)){                q.poll();                p = q.peek();            }            arr.add(p.getNum());        }        return arr;    }}
第二种方法:

import java.util.ArrayList;import java.util.Deque;import java.util.LinkedList;public class Solution {    public ArrayList<Integer> maxInWindows(int [] num, int size) {        ArrayList<Integer> arr = new ArrayList<>();        if (num==null)            return arr;        if (num.length<size||size<=0)            return arr;        Deque<Integer> queue = new LinkedList<>();        for (int i=0;i<num.length;i++){             while (!queue.isEmpty()&&num[i]>=num[queue.getLast()])                queue.pollLast();            while (!queue.isEmpty()&&queue.getFirst()<i-(size-1))                queue.pollFirst();            queue.offerLast(i);             if (i+1>=size)                arr.add(num[queue.getFirst()]);        }        return arr;    }}



1 0
原创粉丝点击