【堆 优先队列】offer 59-1 滑动窗口的最大值

题目

给定一个数组 nums 和滑动窗口的大小 k，请找出所有滑动窗口里的最大值。

示例:

输入: nums = [1,3,-1,-3,5,3,6,7], 和 k = 3
输出: [3,3,5,5,6,7]
解释:

滑动窗口的位置 最大值

---

[1 3 -1] -3 5 3 6 7 3
1 [3 -1 -3] 5 3 6 7 3
1 3 [-1 -3 5] 3 6 7 5
1 3 -1 [-3 5 3] 6 7 5
1 3 -1 -3 [5 3 6] 7 6
1 3 -1 -3 5 [3 6 7] 7

思路

建立大根堆，每次的堆顶就是最大元素
堆的容量是k的大小，当堆的大小大于等于k时，弹出窗口的第一个值
当i遍历3个数以及以上后，就可以将每次往堆中添加数据后的堆顶存入结果数组中。

代码

public int[] maxSlidingWindow(int[] nums, int k) {
        if(nums.length==0) return new int[0];
        Queue<Integer> queue = new PriorityQueue<>((a, b) -> b-a);
        int len = nums.length;
        int []res = new int[len-k+1];
        int index = 0;
        for(int i = 0;i<len;i++){
            //数量超过窗口大小，移出窗口第一个数字
            //因为是先判断再添加，所以判断的是上一个数添加进去后的结果
            if(queue.size()>=k){
                queue.remove(nums[i-k]);
            }
            queue.offer(nums[i]);
            //更新结果
            if(i>=k-1){
                res[index++] = queue.peek();
            }
        }
        return res;



    }