《剑指offer》面试题59:队列的最大值

题目:滑动窗口的最大值

给定一个数组和滑动窗口的大小,请找出所有滑动窗口的最大值。例如,输入数组{2,3,4,2,6,2,5,1}和数字3,那么一共存在6个滑动窗口,他们的最大值分别为{4,4,6,6,6,5}。


思路:

把可能成为最大值数字的下标放入双端队列deque,从而减少遍历次数。首先,所有在没有查看后面数字的情况下,任何一个节点都有可能成为某个状态的滑动窗口的最大值,因此,数组中任何一个元素的下标都会入队。关键在于出队,以下两种情况下,该下标对应的数字不会是窗口的最大值需要出队:(1)该下标已经在窗口之外,比如窗口长度为3,下标5入队,那么最大值只可能在下标3,4,5中出现,队列中如果有下标2则需要出队;(2)后一个元素大于前面的元素,那么前面的元素出对,比如目前队列中有下标3、4,data[3] = 50,data[4]=40,下标5入队,但data[5] = 70,则队列中的3,4都需要出队。

数组{2,3,4,2,6,2,5,1}的长度为3的滑动窗口最大值求解步骤如下

步骤    插入数字    滑动窗口    队列中的下标   最大值
1       2          2           0(2)          N/A          
2       3          2,3         1(3)          N/A   
3       4          2,3,4       2(4)          4   
4       2          3,4,2       2(4),3(2)     4   
5       6          4,2,6       4(6)          6   
6       2          2,6,2       4(6),5(2)     6   
7       5          6,2,5       4(6),6(5)     6   
8       1          2,5,1       6(5),7(1)     5

时间复杂度在o(n)~o(nk)之间,空间复杂度o(k)。

基于以上思路,java参考代码如下:

import java.util.ArrayDeque;
import java.util.Deque;

public classMaxInSlidingWindow {
    //把可能会成为最大值的下标存储下来,从而降低扫描次数
    public static int[] maxInWindows(int[] data,final int size){
        if(data==null ||data.length==0||data.length<size)
            return new int[0];
        int[] result = new int[data.length-size+1];//result的长度?
        Deque<Integer> deque = new ArrayDeque<>();
        //deque.getFirst()---获取,但不移除此双端队列的第一个元素。
        //deque.getLast()---获取,但不移除此双端队列的最后一个元素。
        //deque.addLast(E e)---将指定元素插入此双端队列的末尾。
        //removeFirst()---获取并移除此双端队列第一个元素。
        //removeLast()--- 获取并移除此双端队列的最后一个元素。       
        
        //deque中所存储的都是数组的下标,下标能方便的判断某个数是否还在窗口内
        for(int i=0;i<size-1;i++){
            while (!deque.isEmpty()&&data[i]>=data[deque.getLast()])//后一个元素大于前面的元素,那么前面的元素出队列
                deque.removeLast();
            deque.addLast(i);
        }
        for(int i=size-1;i<data.length;i++){
            while (!deque.isEmpty()&&i-deque.getFirst()+1>size)//该下标已经在窗口之外,那么前面的元素出队列
                deque.removeFirst();
            while (!deque.isEmpty()&&data[deque.getLast()]<=data[i])//后一个元素大于前面的元素,那么元素从后面出列,主要是后面出列多
                deque.removeLast();
            deque.addLast(i);
            result[i-(size-1)] = data[deque.getFirst()];
        }
        return result;
    }
    public static void main(String[] args){
        int[] data = new int[]{2,3,4,2,6,2,5,1};
        int[] result = maxInWindows(data,3);
        for(int i=0;i<result.length;i++){
            System.out.print(result[i]);
            //System.out.print("\t");//中间隔一个tab距离
            System.out.print(" ");
        }
    }
}

测试用例:

a.功能测试(输入数组的数字大小无序;输入数组的数字单调递增;输入数组的数字单调递减)。
b.边界值测试(滑动窗口的大小为0,1,等于输入数组的长度,大于输入数组的长度)。
c.特殊输入测试(输入数组为空)。

参考:

https://www.jianshu.com/p/95e4e819761b

猜你喜欢

转载自blog.csdn.net/qq_43502142/article/details/87894236