双向队列解决滑动窗口最大值问题

这个问题让我死在了双向队列的使用和算法逻辑上面,所以记录一下,以备后面查看所需。

题目描述:

    给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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]}。

解题思路:

    对于size>length 和size<=0 的情况都返回空的数组

    先计算出第一个滑动窗口的最大值。

    对数组进行遍历,保证当前遍历到的数等于或小于其前面的数,这样就可以保证每次队列前面都是本次滑动窗口最大值

    队列里面保存滑动窗口大小的数的下标,如果队列前面所保存的下标不在范围之内,则出队列

    数组记录队列第一个下标所对应的数


代码实现:

    注意双向队列的使用

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> array = new ArrayList<Integer>();
        int length = num.length;
        if(size<=0||size>length){
            return array;
        }
        Deque<Integer> deque = new LinkedList<>();
        for(int j = 0; j < size; j++){
            while(!deque.isEmpty()&&num[j]>num[deque.peekFirst()])
                deque.pollFirst();
            deque.push(j);
        }
        for(int j = size; j < length; j++){
            array.add(num[deque.peekLast()]);
            while(!deque.isEmpty()&&num[j]>num[deque.peekFirst()])
                deque.pollFirst();
            if(!deque.isEmpty()&&deque.peekLast()<=j-size)
                deque.pollLast();
            deque.push(j);
        }
        array.add(num[deque.peekLast()]);
        return array;
    }
}

猜你喜欢

转载自blog.csdn.net/zz0129/article/details/80819534