优先队列解决滑动窗口最大值问题

leetcode239:. - 力扣（LeetCode）
 

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        PriorityQueue<int[]> queue = new PriorityQueue<>(new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                return o1[0] != o2[0] ? o2[0] - o1[0] : o2[1] - o1[1];
            }
        });

        for (int i = 0; i < k; i++) {
            queue.add(new int[]{nums[i], i});
        }

        ArrayList<Object> resList = new ArrayList<>();
        resList.add(queue.peek()[0]);

        int index = 1;

        for (int i = k; i < nums.length; i++) {
            queue.add(new int[]{nums[i], i});
            //删除过期的头部，只要头部不是过期的就好
            while (queue.peek()[1] <= i - k) {
                queue.poll();
            }
            resList.add(queue.peek()[0]);
            index++;
        }
        int[] res = new int[resList.size()];
        for (int i = 0; i < resList.size(); i++) {
            res[i] = (int) resList.get(i);
        }
        return res;
    }
}

        Java的优先队列支持根据自定义的comparator来排序，我们先将值大的放在队首，值相同元素越靠后越需要排在前面，后续每轮在添加一个元素后，我们需要选取窗口中的最大值，先将队首元素不在窗口内的弹出，然后直接取队首元素即可，不用担心队列后面还有不在窗口内的过期元素，因为并不影响我们取队首窗口内的最大值。遍历添加完所有元素，窗口也就移到了最后，这样就可以避免每次都对 窗口内元素进行排序。