单调队列的基本概念
单调队列,顾名思义就是单调递增或者单调递减的队列。事实上,我们通常用到的单调队列并不是通常所说的遵循FIFO原则的队列,而是一个双端的队列,也就是通常说的Deque,它既可以从队列的头取走元素,也可以从队列的尾取走元素。一般用于动态规划中或者用于动态维护子数组的最值(最小值或者最大值)。这里一个非常典型的问题就是滑动窗口问题。
滑动窗口问题
滑动窗口是计算机网络中一个非常重要的概念,但是这里讨论的是算法问题,所以直接上题目吧。 题目链接:239. Sliding Window Maximum
给定一个整数数组nums,给定一个整数k,k为窗口大小,滑动窗口,每次向右滑动一格,求每一个窗口中的元素的最大值。
解题思路
这个问题最简单的办法自然是暴力求解,每次从一个窗口的起点开始向后遍历k个数字,然后求出最大值,很容易知道这个解法的时间复杂度为O(n^2),但是题目给定的要求是线性的时间复杂度,也就是O(n)。所以这里自然不能用暴力求解。
实际上,我们可以用一个双端队列来存储当前窗口的最大值。我们创建一个双端队列,这里队列中存储的是窗口最大值的下标。如果队列为空,那么就将当前元素的下标添加到队列中。如果队列不为空,那么就有两种情况。如果当前元素比队尾元素所指向的元素还要大,那么当这个元素进入到滑动窗口,队列尾那些比他小的元素所指向的元素,一定不可能是窗口中的最大值,所以就将队尾这些元素删去,直到队尾元素指向的元素比当前的元素大或者相等,然后将当前元素的下标加入队列。如果当前元素比队尾元素所指向的元素要小或者相等,那么就直接将当前元素的下标加入到队列尾。取的时候,只需要取队列头的元素就可以了,因为那就代表着当前窗口的的最大值。取的时候要判断一下,如果队头元素已经是窗口之外了,那么就要将队头出队。否则直接取队头元素就可以了,它就是窗口中最大值对应的下标。文字很多很抽象,用图中的样例来模拟一下好了。遍历完第一个窗口的时候,队列的元素为1(3)、2(-1),括号中为最大元素,队列元素为元素的下标。所以当前窗口最大值为3
第二个窗口,-3比-1小,所以将3入队,队列元素为1(3)、2(-1)、3(-3),当前窗口最大值为3
第三个窗口,5比队列中所有元素都要大,所以队列中所有元素被清空,队列元素为4(5),当前窗口最大值为5
第四个窗口,队列元素4(5)、5(3)
第五个窗口,队列元素6(6)
.......
通过上面的模拟思路应该很清晰了,那么就很容易写出对应的代码了,下面贴出AC代码
public class Solution {
public int[] maxSlidingWindow(int[] nums, int k) {
if(nums == null || nums.length == 0){
return new int[0];
}
int[] result = new int[nums.length-k+1];
LinkedList queue = new LinkedList<>();
for (int i = 0; i < nums.length; i++) {
while (!queue.isEmpty() && nums[queue.peekLast()] < nums[i]){
queue.pollLast();
}
queue.offerLast(i);
//判定队列头下标是否在窗口之内
if(queue.peekFirst() == i-k){
queue.pollFirst();
}
if(i >= k-1){
result[i-k+1] = nums[queue.peekFirst()];
}
}
return result;
}
}
简单总结一下,队列里面存储的元素,是一组下标,这一组下标对应的元素是单调递减的,所以称之为单调队列。因为下标对应的元素递减,所以每次队列头所对应的元素就是窗口中最大的元素。就酱紫。
最后吐槽一下知乎的编辑器,真是难用得要死!