滑动窗口最大值

滑动窗口最大值

题目

给你一个整数数组 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]

思路

• 可以使用优先队列，每个元素是一个两维数组，一个是元素的值，一个是元素的下标，重写他的排序规则，如果两个值不一样大，就按值排序，如果值一样大，就按下标排序，下标大的在前面，
• 然后每访问一个元素，将元素加入进优先队列，然后从队列中将对头元素取出来，如果他的下标在 i - k + 1中，就是还在滑动数组范围内的，添加到结果集中，如果不在其中就将该元素弹出，取下一个对头元素。

代码实现

public int[] maxSlidingWindow(int[] nums, int k) {
    List<Integer> result = new ArrayList<>();
    PriorityQueue<int[]> priorityQueue = new PriorityQueue<>(new Comparator<int[]>() {
        @Override
        public int compare(int[] o1, int[] o2) {
            return o1[0] != o2[0] ? o2[0] - o1[0] : o1[1] - o2[1];
        }
    });
    for (int i = 0; i < k; i++ ) {
        priorityQueue.offer(new int[]{nums[i], i});
    }
    result.add(priorityQueue.peek()[0]);
    for (int i = k; i < nums.length; i++) {
        priorityQueue.offer(new int[]{nums[i], i});
        int[] peek = priorityQueue.peek();
        while (peek[1] < i - k + 1) {
            priorityQueue.poll();
            peek = priorityQueue.peek();
        }
        result.add(peek[0]);
    }
    return result.stream().mapToInt(i -> i).toArray();
}