剑指offer算法题:滑动窗口的最大值maxInWindows

给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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.LinkedList;
import java.util.Deque;
public class Solution {
    public ArrayList<Integer> maxInWindows(int [] num, int size)
    {
        ArrayList<Integer> result = new ArrayList<>();
        if(num == null || 
           num.length == 0 || 
           size <= 0 || 
           size > num.length)
            return result;
        Deque<Integer> deque = new LinkedList<>();//双端队列
        for(int i = 0; i < size;i++) {
            //待插入数据与队列中的数据比较,删除之前较小的数据(较小的数据肯定不能为接下来的滑动窗口的最大值)
            while(!deque.isEmpty() && num[i] >= num[deque.peekLast()])//从尾端比较,将所有小于的数据删除
                deque.pollLast();
            deque.add(i);//队列存的是数组的下标,方便将队列中滑出窗口的数据poll
        }
        for(int i = size;i < num.length;i++) {
            result.add(num[deque.peekFirst()]);
            while(!deque.isEmpty() && num[i] >= num[deque.peekLast()])
                deque.pollLast();
            //将队列中滑出窗口的数据poll
            if(!deque.isEmpty() && deque.peekFirst() <= i - size)
                deque.pollFirst();
            deque.add(i);
        }
        result.add(num[deque.peekFirst()]);
        return result;
    }
}
发布了292 篇原创文章 · 获赞 73 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_43777983/article/details/104411063
今日推荐