栈和队列是重要的数据结构,应用在程序中很多底层结构中(比如栈是递归的底层实现),本次由队列来引出一个重要的数据结构(单调队列)。
例题:239滑动窗口
代码:
//解法一
//自定义数组
class MyQueue {
Deque<Integer> deque = new LinkedList<>();
//弹出元素时,比较当前要弹出的数值是否等于队列出口的数值,如果相等则弹出(因为这样可以保证单调队列的逻辑)
//同时判断队列当前是否为空
void poll(int val) {
if (!deque.isEmpty() && val == deque.peek()) {
deque.poll();
}
}
//添加元素时,如果要添加的元素大于入口处的元素,就将入口元素弹出
//保证队列元素单调递减
//比如此时队列元素3,1,2将要入队,比1大,所以1弹出,此时队列:3,2
void add(int val) {
while (!deque.isEmpty() && val > deque.getLast()) {
deque.removeLast();
}
deque.add(val);
}
//队列队顶元素始终为最大值
int peek() {
return deque.peek();
}
}
class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
if (nums.length == 1) {
return nums;
}
int len = nums.length - k + 1;
//存放结果元素的数组
int[] res = new int[len];
int num = 0;
//自定义队列
MyQueue myQueue = new MyQueue();
//先将前k的元素放入队列
for (int i = 0; i < k; i++) {
myQueue.add(nums[i]);
}
res[num++] = myQueue.peek();
for (int i = k; i < nums.length; i++) {
//滑动窗口移除最前面的元素,移除时判断该元素是否等于单调队列的头元素,若等于则移除,否则不移 除(保证单调队列的逻辑)
myQueue.poll(nums[i - k]);
//滑动窗口加入最后面的元素
myQueue.add(nums[i]);
//记录对应的最大值
res[num++] = myQueue.peek();
}
return res;
}
}
总结点
1.单调队列
单调的理解很简单,初中数学就应该学过单调的概念:元素从小到大排列(单调增);元素从大到小排列(单调减)。单调队列的理论逻辑就是这样,那么针对本题如何使用单调队列呢?
针对本题,我们一开始想到的是创建一个队列来模拟滑动窗口,这个队列呢,放进去窗口里的元素,然后随着窗口的移动,队列也一进一出,每次移动之后,队列告诉我们里面的最大值是什么。但是我们会发现这个队列在实际实现起来是有困难的,所以我们改变思路:其实队列没有必要维护窗口里的所有元素,只需要维护有可能成为窗口里最大值的元素就可以了,同时保证队里里的元素数值是由大到小的。
通过动画的形式来队列元素是如何变化的:
对于窗口里的元素{2, 3, 5, 1 ,4},单调队列里只维护{5, 4} 就够了,保持单调队列里单调递减,此时队列出口元素就是窗口里最大元素。(本题为什么选择用单调队列呢?因为使用单调队列方便我们每次滑动窗口放回当前窗口元素的最大值,解决了返回值的问题剩下的问题就在于如何让单调队列的push和pop的逻辑和题目对应起来(同时又不破坏单调队列本身的逻辑))。
设计单调队列的时候,pop,和push操作要保持如下规则:
-
pop(value):如果窗口移除的元素value等于单调队列的出口元素,那么队列弹出元素,否则不用任何操作(判断弹窗元素是否为当前窗口所有元素的最大值,若是队列应当把这个最大元素pop来维护滑动窗口的逻辑,若不是不操作来保证单调队列的逻辑(因为队列维护的是单调队列的逻辑所以不要纠结于题目中的滑动窗口的逻辑了))
-
push(value):如果push的元素value大于入口元素的数值,那么就将队列入口的元素弹出,直到push元素的数值小于等于队列入口元素的数值为止
保持如上规则,每次窗口移动的时候,只要使用que.front()(返回队首元素操作)就可以返回当前窗口的最大值。
将单调队列用于本题的整体过程如下图:
注意:题解中单调队列里的pop和push接口,仅适用于本题。单调队列不是一成不变的,而是不同场景不同写法,总之要保证队列里单调递减或递增的原则,所以叫做单调队列。不要以为本地中的单调队列实现就是固定的写法。
讲解出处:代码随想录卡尔老师