给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回滑动窗口中的最大值。
示例 1:
输入:nums = [1,3,-1,-3,5,3,6,7], k = 3 输出:[3,3,5,5,6,7] 解释:滑动窗口的位置 最大值
[1 3 -1] -3 5 3 6 7 3 1 [3 -1 -3] 5 3 6 7 3 1 3 [-1 -3 5] 3 6 7 5 1 3 -1 [-3 5 3] 6 7 5 1 3 -1 -3 [5 3 6] 7 6 1 3 -1 -3 5 [3 6 7] 7 示例 2:
输入:nums = [1], k = 1 输出:[1] 示例 3:
输入:nums = [1,-1], k = 1 输出:[1,-1] 示例 4:
输入:nums = [9,11], k = 2 输出:[11] 示例 5:
输入:nums = [4,-2], k = 2 输出:[4]
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/sliding-window-maximum
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
// 定义一个优先队列,存放元素的值和坐标,优先队列的最顶就是最大值。
// 将队列的顶部 的值赋值到结果中。
int numsLength = nums.length;
// 定义返回值数组
int[] ans = new int[numsLength - k + 1];
PriorityQueue<int[]> priorityQueue = new PriorityQueue<int[]>(new Comparator<int[]>(){
public int compare(int[] pari1, int[] pari2) {
// 第一个元素存放的是nums里面的元素
if (pari1[0] != pari2[0]){
// 值做比较
return pari2[0] - pari1[0];
} else {
// 下标做比较
return pari2[1] - pari1[1];
}
}});
// 将前k个元素入队
for (int i = 0; i < k; i++) {
priorityQueue.offer(new int[] {nums[i], i});
}
// 第一个结果
ans[0] = priorityQueue.peek()[0];
// 处理k以后的元素
for (int i = k; i < numsLength; i++) {
// 将当前元素入队,如果当前的最大值是落在了滑动窗口的左侧,就出队
priorityQueue.offer(new int[]{nums[i], i});
while (priorityQueue.peek()[1] < i - k + 1) {
priorityQueue.poll();
}
ans[i - k + 1] = priorityQueue.peek()[0];
}
return ans;
}
}