给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回滑动窗口中的最大值。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sliding-window-maximum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
优先队列
import java.util.Comparator;
import java.util.PriorityQueue;
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
int n = nums.length;
PriorityQueue<int[]> pq = new PriorityQueue<int[]>(new Comparator<int[]>() {
public int compare(int[] pair1, int[] pair2) {
return pair1[0] != pair2[0] ? pair2[0] - pair1[0] : pair2[1] - pair1[1];
}
});
int[] ans = new int[n - k + 1];
for (int i = 0; i < n; ++i) {
pq.offer(new int[]{nums[i], i});
if (i >= k - 1) {
ans[i - k + 1] = pq.peek()[0];
while (!pq.isEmpty() && pq.peek()[1] <= i - k + 1) {
pq.poll();
}
}
}
return ans;
}
}
单调队列
import java.util.Deque;
import java.util.LinkedList;
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
int[] ret = new int[nums.length - k + 1];
Deque<Integer> queue = new LinkedList<>();
for (int i = 0; i < nums.length; ++i) {
while (!queue.isEmpty() && nums[i] >= nums[queue.peekLast()]) {
queue.pollLast();
}
queue.offerLast(i);
if (i >= k - 1) {
ret[i - k + 1] = nums[queue.peekFirst()];
if (queue.peekFirst() == i - k + 1) {
queue.pollFirst();
}
}
}
return ret;
}
}