题目链接

牛客网

题目描述

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

例如，如果输入数组 {2, 3, 4, 2, 6, 2, 5, 1} 及滑动窗口的大小 3，那么一共存在 6 个滑动窗口，他们的最大值分别为 {4, 4, 6, 6, 6, 5}。

解题思路

PriorityQueue.remove(Object O)可以删除堆中的指定元素，只要维持堆的大小在size就可以

import java.util.*;
public class Solution {
   
    public ArrayList<Integer> maxInWindows(int [] nums, int size) {
   
        ArrayList<Integer> res = new ArrayList<>();
        if (size<1 || size>nums.length) return res;
        PriorityQueue<Integer> pq = new PriorityQueue<>((o1, o2)->o2-o1);
        for (int i=0;i<size;i++) 
            pq.add(nums[i]);
        res.add(pq.peek());
        for(int i=0, j=size;j<nums.length;i++, j++) {
   
            pq.remove(nums[i]);
            pq.add(nums[j]);
            res.add(pq.peek());
        }
        return res;
    }
}