堆与滑动窗口的结合（算法村第十六关黄金挑战）

滑动窗口最大值

239. 滑动窗口最大值 - 力扣（LeetCode）

给你一个整数数组 nums，有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。

返回 滑动窗口中的最大值 。

示例 1：

输入：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

示例 2：

输入：nums = [1], k = 1
输出：[1]

提示：

• 1 <= nums.length <= 105
• -104 <= nums[i] <= 104
• 1 <= k <= nums.length

堆与滑动窗口

参考题解：239. 滑动窗口最大值 - 力扣（LeetCode）

public int[] maxSlidingWindow(int[] nums, int k)
{
    //创建一个大根堆，节点由nums中元素的值pair[0]及其下标pair[1]组成
    PriorityQueue<int[]> maxHeap = new PriorityQueue<>((int[] pair1, int[] pair2) -> (pair2[0] - pair1[0]));

    //将前k个元素加入大根堆
    for (int i = 0; i < k; i++)
        maxHeap.offer(new int[]{nums[i], i});

    //答案数组,存滑动窗口中的最大值
    int[] ans = new int[nums.length - k + 1];
    //存滑动窗口的第一个最大值
    ans[0] = maxHeap.peek()[0];

    //滑动窗口开始移动
    for (int i = k; i < nums.length; i++)
    {
        maxHeap.offer(new int[]{nums[i], i});

        //不断移除堆顶最大值，直到堆顶元素的索引出现在滑动窗口中
        //i - k是滑动窗口（闭区间）的左边界
        while (maxHeap.peek()[1] <= i - k)
            maxHeap.poll();

        //滑动窗口移动后，从ans[1]开始存
        ans[i - k + 1] = maxHeap.peek()[0];
    }

    return ans;
}