滑动窗口的最大值
Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情。
点击 这里 可以查看更多算法面试相关内容~
一、题目描述:
给定一个长度为 n 的数组 num 和滑动窗口的大小 size ,找出所有滑动窗口里数值的最大值。 例如,如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小3,那么一共存在6个滑动窗口,他们的最大值分别为{4,4,6,6,6,5}; 针对数组{2,3,4,2,6,2,5,1}的滑动窗口有以下6个: {[2,3,4],2,6,2,5,1}, {2,[3,4,2],6,2,5,1}, {2,3,[4,2,6],2,5,1}, {2,3,4,[2,6,2],5,1}, {2,3,4,2,[6,2,5],1}, {2,3,4,2,6,[2,5,1]}。
二、思路分析:
创建一个size大小的大顶推,移动一个窗格,删除左边元素,堆顶入队列
三、AC 代码:
package cn.com.codingce.栈队列堆.滑动窗口的最大值;
import java.util.ArrayList;
import java.util.PriorityQueue;
public class Solution {
public static void main(String[] args) {
System.out.println(new Solution().maxInWindows(new int[]{1, 2, 5, 9}, 3));
}
public ArrayList<Integer> maxInWindows(int [] num, int size) {
ArrayList<Integer> ret = new ArrayList<>();
if(size > num.length || size < 1) {
return new ArrayList<>();
}
// 大顶堆
PriorityQueue<Integer> priorityQueue = new PriorityQueue<>((o1, o2) -> o2 - o1);
// 维护一个大小为 size 的大顶堆
for (int i = 0; i < size; i++) {
priorityQueue.add(num[i]);
}
ret.add(priorityQueue.peek());
for (int i = 0, j = i + size; j < num.length; i++, j ++) {
priorityQueue.remove(num[i]);
priorityQueue.add(num[j]);
ret.add(priorityQueue.peek());
}
return ret;
}
}
总结
每天进步一点点