方法一:优先队列
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
//使用大顶堆,当两个元素值不同时,按值降序排列;值相同时,按下标降序排列
Queue<int[]> queue = new PriorityQueue<>((o1, o2) -> o1[0] != o2[0] ? o2[0] - o1[0] : o2[1] - o1[1]);
//将前k个元素入堆
for (int i = 0; i < k; i++) {
queue.offer(new int[]{nums[i], i});
}
int n = nums.length;
int[] ans = new int[n - k + 1];
ans[0] = queue.peek()[0];
for (int i = k; i < n; i++) {
queue.offer(new int[]{nums[i], i});
//如果最大值的位置在滑动窗口左边界外,将其从大顶堆中移除
//易错点:应该循环判断位置,而不是只判断一次
//例如[9,10,8,-7,-4,-8,2] k = 5,在最后一个窗口时,
//此时数组中的元素均在大顶堆中,而10,9需要先后被移除
while (queue.peek()[1] <= i - k) {
queue.poll();
}
//记录每个滑动窗口中的最大值
ans[i - k + 1] = queue.peek()[0];
}
return ans;
}
}
方法二:单调队列
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
//使用双端队列按照升序存储元素下标
//并且元素下标在数组中对应的值严格单调递减
Deque<Integer> deque = new LinkedList<>();
//将前 k 个元素的下标入队
for (int i = 0; i < k; i++) {
//如果队列不为空且当前入队元素大于等于队尾元素的值,将队尾元素的下标出队
while (!deque.isEmpty() && nums[i] >= nums[deque.peekLast()]) {
deque.pollLast();
}
//如果队列为空或当前入队元素小于队尾元素的值,将当前元素的下标入队
deque.offerLast(i);
}
int n = nums.length;
//n-k+1个窗口,对应n-k+1个值
int[] ans = new int[n - k + 1];
//将第一个窗口中的最大值加入结果集中
ans[0] = nums[deque.peekFirst()];
for (int i = k; i < n; i++) {
//如果队列不为空且当前入队元素大于等于队尾元素的值,将队尾元素的下标出队
while (!deque.isEmpty() && nums[i] >= nums[deque.peekLast()]) {
deque.pollLast();
}
//如果队列为空或当前入队元素小于队尾元素的值,将当前元素的下标入队
deque.offerLast(i);
//如果最大值的位置在滑动窗口左边界外,将其从队列中移除
while (deque.peekFirst() <= i - k) {
deque.pollFirst();
}
//记录每个滑动窗口中的最大值
ans[i - k + 1] = nums[deque.peekFirst()];
}
return ans;
}
}