剑指offer-63.滑动窗口的最大值

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/zxm1306192988/article/details/82026101

牛客

题目描述
给定一个数组和滑动窗口的大小，找出所有滑动窗口里数值的最大值。例如，如果输入数组{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.Comparator;
import java.util.PriorityQueue;
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;
        }
        PriorityQueue<Integer> heap = new PriorityQueue<>(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {// 最大堆
                return o2 - o1;
            }
        });
        for (int i = 0; i < size; i++) {// 现将第一个滑动窗口内的数字加入堆中
            heap.add(num[i]);
        }
        result.add(heap.peek());
        for (int start = 1, end = size; end < num.length; start++, end++) {
            heap.remove(num[start - 1]);
            heap.add(num[end]);
            result.add(heap.peek());
        }
        return result;
    }
}