滑动窗口的最大值

描述

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

解析

用一个大顶堆，保存当前滑动窗口中的数据。滑动窗口每次移动一格，就将前面一个数出堆，后面一个数入堆。

代码

public ArrayList<Integer> maxInWindows(int [] num, int size) {
        ArrayList<Integer> resList = new ArrayList<>();
        if(null == num || num.length <= 0 || size <= 0 || size > num.length){
            return resList;
        }
        PriorityQueue<Integer> queue = new PriorityQueue<>((o1, o2) -> o2 - o1);//大顶堆
        int index = 0;//当前访问的数组index
        for (index = 0; index < size; index++) {
            queue.offer(num[index]);//初始化堆，最大的数就在堆顶
        }
        int numSize = num.length;
        while (index < numSize) {//依次右移，得到最大数，然后出堆一个数，再入堆一个新的数
            resList.add(queue.peek());
            queue.remove(num[index - size]);
            queue.offer(num[index]);
            index++;
        }
        resList.add(queue.peek());//最后一次入堆，没有找出最大值，补偿
        return resList;
    }