思路
利用优先队列每个位置存储一个int[2],第一个位置存放这个数的大小,第二个位置存放索引。
自定义排序规则为按照数值的大小从大到小排列,这样堆顶的数就永远是最大的。
需要注意的点是:我们并不需要在窗口每次移动的时候将最大值移出,除非最大值不在窗口内(这就是数组中索引的作用),利用while循环将不属于窗口的值移出.
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
int n = nums.length;
int[] res = new int[n-k+1];
PriorityQueue<int[]> pq = 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++){
pq.offer(new int[]{nums[i],i});
}
res[0] = pq.peek()[0];
int index = 0;
for(int i = k;i<nums.length;i++){
pq.offer(new int[]{nums[i],i});
while (pq.peek()[1]<i-k+1){
pq.poll();
}
res[++index] = pq.peek()[0];
}
return res;
}
}