单调队列
我们从最简单的问题开始:
eg. 现在有一个数列{An},假设它有n项,问题是:求出每个区间长度为k的连续区间内数列的最大值。
数学表达式: f(i) = max{a(i-k+1),a(i-k+2),…, a(i)},i = 0,1,…,N-1。
解法一:
很直观的一种解法,那就是从数列的开头,将窗放上去,然后找到这最开始的k个数的最大值,然后窗最后移一个单元,继续找到k个数中的最大值。
这种方法每求一个f(i),都要进行k-1次的比较,复杂度为O(N*k)。
那么有没有更快一点的算法呢?
解法二:
我们知道,上一种算法有一个地方是重复比较了,就是在找当前的f(i)的时候,i的前面k-1个数其它在算f(i-1)的时候我们就比较过了。那么我们能不能保存上一次的结果呢?当然主要是i的前k-1个数中的最大值了。答案是可以,这就要用到单调递减队列。
单调递减队列是这么一个队列,它的头元素一直是队列当中的最大值,而且队列中的值是按照递减的顺序排列的。我们可以从队列的末尾插入一个元素,可以从队列的两端删除元素。同理单调递增队列。
1.首先看插入元素:为了保证队列的递减性,我们在插入元素v的时候,要将队尾的元素和v比较,如果队尾的元素不大于v,则删除队尾的元素,然后继续将新的队尾的元素与v比较,直到队尾的元素大于v,这个时候我们才将v插入到队尾。
2.队尾的删除刚刚已经说了,那么队首的元素什么时候删除呢?由于我们只需要保存i的前k-1个元素中的最大值,所以当队首的元素的索引或下标小于i-k+1的时候,就说明队首的元素对于求f(i)已经没有意义了,因为它已经不在窗里面了。所以当index[队首元素]<i-k+1时,将队首元素删除。
为了让读者更明白一点,我举个简单的例子。
假设数列为:8,7,12,5,16,9,17,2,4,6.N=10,k=3.
那么我们构造一个长度为3的单调递减队列:
首先,那8和它的索引0放入队列中,我们用(8,0)表示,每一步插入元素时队列中的元素如下:
队列的状态
第1个数入队:8
第2个数入队:7
第3个数入队: 7 12
第4个数入队: 5
第5个数入队:5 16
第6个数入队:5 9
第7个数入队:9 17
第8个数入队:2
第9个数入队:2 4
第10个数入队:2 4 6
取队首 8
7
7
5
5
9
9
2
2
2
区间为3,所以前两个数产生的最小值舍弃,从第3个数开始,每入队后取队首正好是区间为3 的最小值,
现在队列里刚好没有超过区间的,如果超过区间就的
把从队首删除。
代码如下:
#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
using namespace std;
const int maxn=100007;
int a[maxn],q[maxn],s[maxn];//s保存每个区间的最小值,q是模拟队列数组
int main()
{
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
int head=1,tail=0;
q[0]=0;
for(int i=1;i<=n;i++)
{
while(head<=tail&&a[q[tail]]>a[i])
{
tail--;//不满足单增队列
}
q[++tail]=i;//i入队
if(i-q[head]>=m)
//当前数下标减去队首存储下标大于等于m,说明超出区间
//注意q保存的是下标而不是数值
{
head++;//队列长度超出m,队头出队
}
for(int j=head;j<=tail;j++)
printf("%d ",a[q[j]]);
printf("\n");
s[i]=a[q[head]] ;//拿队头出来,就是拿区间最小值,根据算法需要把队头拿出来计算
}
for(int i=3;i<=n;i++)
printf("%d ",s[i]);
return 0;
}
//单调队列,一定要注意,不超区间,队首就不删