单调队列定义
队内元素是单调的,递增或递减。
单调队列性质
1、队尾既可以执行入队操作,也可以执行出队操作, 但队头只能执行出队操作
单调队列解决滑动窗口最大最小值问题
题目:
有一个长为 n的序列 a,以及一个大小为 k 的窗口。现在这个从左边开始向右滑动,每次滑动一个单位,求出每次滑动后窗口中的最大值和最小值。
输入格式
输入一共有两行,第一行有两个正整数 n,k。 第二行 n 个整数,表示序列 a
输出格式
输出共两行,第一行为每次窗口滑动的最小值
第二行为每次窗口滑动的最大值
样例输入:
8 3
1 3 -1 -3 5 3 6 7
样例输出:
-1 -3 -3 -3 3 3
3 3 5 5 6 7
题目分析:
首先给出一个数组,有n个数字,然后给出滑动区间长度为k,每次滑动窗口右移动一个单位,每次找出窗口内的最大和最小值。
复杂度分析:
如果我用两个for循环,内存for循环找出最大值,或最小值,那么复杂度为O(nm)
如果我用两个优先队列,每次取出最大值,最小值。那么复杂度为O(nlogn)
使用单调队列,每次操作O(1),一共n次操作复杂度O(n);
代码实现
我们用一个数组q,存放队列中元素下标,然后我们用head,和tail维护单调队列!
1、队首出队:
当队首元素从窗口滑出时,队首元素出队head++
2、队尾入队:
当新元素滑入窗口时,要把新元素从队尾插入,此时分为两种情况:
(以每次滑出最大值为例)
1、直接插入:如果新元素小于队尾元素,直接插入,tail++
2、先删除后插:如果新元素大于等于队尾元素,那就先删除队尾元素,删除方法:tail–。用一个循环一直删到队尾元素比新插入的元素大为止!
完整代码:
#include <bits/stdc++.h>
using namespace std;
int q1[1000100],q2[1000100];
int a[1000010];
int n,m;
void min_deque()
{
//初始化,队头在前,队尾在后
int h=1,t=0;
for(int i=1;i<=n;i++)
{
//如果队头下标,不在窗口内,队头出队
while(h<=t&&q1[h]+m<=i) h++;
//队尾入队
while(h<=t&&a[i]<a[q1[t]])t--;
q1[++t]=i;//队尾插入
if(i>=m){
printf("%d ",a[q1[h]]);
}
}
cout<<endl;
}
void max_deque()//与min_deque相同
{
int h=1,t=0;
for(int i=1;i<=n;i++)
{
while(h<=t&&q2[h]+m<=i)h++;
while(h<=t&&a[i]>a[q2[t]])t--;//这里换成>号
q2[++t]=i;
if(i>=m)
{
printf("%d ",a[q2[h]]);
}
}
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
min_deque();
max_deque();
return 0;
}