第六十三题:滑动窗口的最大值
题目描述
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。
例如,如果输入数组{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]}。
思路
①双端队列 队列头始终保持为最大值
②大顶堆 建立大小为 k 的大顶堆,将元素添加到 maxHeap中,在取值的同时保持窗口大小不变
具体图解如下(双端队列)
具体实现代码如下(双端队列)
public class Solution{
// 滑动窗口最大值
public static ArrayList<Integer> maxInWindows_1(int [] num, int size) {
ArrayList<Integer> ret = new ArrayList<>();
// 鲁棒性
if (num == null || num.length < size || size == 0){
return ret;
}
LinkedList<Integer> indexDeque = new LinkedList<>();
for (int i = 0; i < num.length; i++) {
while (!indexDeque.isEmpty() && num[i] > num[indexDeque.getLast()]) {
indexDeque.removeLast();
}
indexDeque.addLast(i);
//在头部移除元素
if (indexDeque.getFirst() == i - size) {
indexDeque.removeFirst();
}
//输出结果
if (i >= size - 1) {
ret.add(num[indexDeque.getFirst()]);
}
}
return ret;
}
}
具体图解如下(大顶堆)
具体实现代码如下(大顶堆)
public class Solution{
// 滑动窗口最大值
public ArrayList<Integer> maxInWindows(int [] num, int size){
ArrayList<Integer> res = new ArrayList<>();
if (size > num.length || size < 1){
return res;
}
// 构建最大堆,即堆顶元素是堆的最大值。
PriorityQueue<Integer> maxHeap = new PriorityQueue<>(4, new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return o2.compareTo(o1);
}
});
for (int i = 0; i < num.length; i++) {
// 添加元素
maxHeap.add(num[i]);
// 移出元素
if (i >= size -1){
res.add(maxHeap.peek());
maxHeap.remove(num[i-size+1]);
}
}
return res;
}
}
NowCoder(Online Coding, Please Click)