单调队列:满足单调性的队列结构
单调队列使用场景较少,一般只用在求滑动窗口与解决求一个数左(右)边第一个小(大)于他的元素(元素的下标)的问题上。
模板为滑动窗口问题。
暴力做法使用两层循环求解的时间复杂度是O(n^2)。使用单调队列优化时间复杂度可以达到O(n)。
在指针i从左向右遍历的过程中,我们可以用一个队列来保存a数组所有下标(不包括i指向元素),越大的元素越靠近队头,越小的元素越靠近队尾。所以我们每次从队尾开始遍历,当队列不为空且队列元素大于a[i]时,tt --(弹出该元素),直到队头元素小于a[i]时,且把不属于窗口内的队头元素弹出(hh ++ ),输出队头元素,并将a[i]压入队尾。这样循环遍历数组我们就可以得到一个单调增队列,即队头元素最大,队尾元素最小。
和单调栈不一样,单调队列队头可以插入和弹出,队头也可以弹出。
由于每个元素一定会进入队列且至多弹出一次,因此操作次数至多为2n,故时间复杂度为O(n)。
#include<iostream>
using namespace std;
const int N = 1000010;
int a[N],q[N]; //此题单调队列里存的为数组下标,方便判断窗口滑动到哪里以便于弹出队头元素
int main(){
int n,k;
cin >> n >> k;
for(int i = 0;i < n;i ++ ){
cin >>a[i];
}
//找最小值
int hh = 0,tt = -1; //清空队列
for(int i = 0;i < n;i ++ ){ //枚举序列
while(hh <= tt && a[q[tt]] >= a[i]) tt --; //队尾出队(队列不为空且新元素更优)
q[ ++ tt] = i; //队尾入队,(储存下标,方便队头出队)
if(q[hh] < i - k + 1) hh ++ ; //队头出队
if(i >= k - 1) cout << a[q[hh]] << ' '; //输出最值
}
puts("");
//找最大值
hh = 0,tt = -1;
for(int i = 0;i < n;i ++ ){
while(hh <= tt && a[q[tt]] <= a[i]) tt --;
q[ ++ tt] = i;
if(q[hh] < i - k + 1) hh ++ ;
if(i >= k - 1) cout << a[q[hh]] << ' ';
}
return 0;
}