题目:滑动窗口最大值
思路:双端队列。
队列的两端均可以删除元素。
我们需要确保队列中的元素是递减的,这样每次可以从队列头部拿到滑动窗口的最大值。
同时,随着窗口的移动,我们需要在队列里删除不在窗口的元素,因此,在队列里,我们保存的是元素的下标。
遍历数组中的元素,重复以下过程:
- 从队列尾部删除比当前元素小的所有元素;
- 当前元素入队列;
- 如果队列头部元素不在窗口内,删除它(队列中存的是索引);
- 如果滑动窗口已经形成,把队头元素加入结果集。
代码:
import java.util.*;
public class Solution {
public ArrayList<Integer> maxInWindows(int [] num, int size) {
int len = num.length;
ArrayList<Integer> res = new ArrayList<>();
if (size > len || size <= 0) {
return res;
}
Deque<Integer> deque = new LinkedList<>();
for (int i = 0; i < len; i ++) {
while (!deque.isEmpty() && num[i] > num[deque.peekLast()]) {
deque.pollLast();
}
deque.offerLast(i);
if (deque.peekFirst() < i - size + 1) {
deque.pollFirst();
}
if (i >= size - 1) {
res.add(num[deque.peekFirst()]);
}
}
return res;
}
}