单调队列名字逼格很高,但是其实很简单
比如,对于题目中样例的数据
1 3 -1 -3 5 3 6 7
我们要算最小值,我们看最小值是多少
假设现在我们有一个特殊的队列,双端队列,每一端都可以 push ,或者 pop ,的队列
首先我们看第一个元素,放入队列
1
此时最小值是 1
然后第二个
1 3
最小值还是 1
此时我们放入第三个 -1
1 3 -1
这个时候我们发现,因为 3 ≥ − 1 3 \ge -1 3≥−1 ,所以只要 3 和 -1 ,都在窗口里面,那么 3 就一定不是最小值,因为窗口一直会向右移动,所以如果 3 ,还在,那么 -1 ,也一定在,那么这个 3 有生之年再也不会是最小值了,不像 1 ,又年轻数还小,我们把它 pop 掉:
1 -1
1 也是一样,所以也剔除掉
-1
我们发现这样的操作就是每次把靠前的比它大的 pop 掉,然后将现在的元素放到队首,最后 pop 掉所有不在窗口里的元素,非常像单调栈,易得这个队列是单调递增的,所以队尾元素一定最小,每次输出队首元素就可以了
然后只剩下一个问题,那就是双端队列实现,推荐 STL deque
最后就是代码实现:
#include<deque>
#include<cctype>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define rp(i, e) for(int i=1;i<=(e);i++)
#define pr(i, e) for(int i=(e);i>=1;i--)
#define rp0(i, e) for(int i=0;i<(e);i++)
#define pr0(i, e) for(int i=(e-1);i>=0;i--)
#define rps(i, b, e) for(int i=(b);i<=(e);i++)
#define prs(i, e, b) for(int i=(e);i>=(b);i--)
#define rpg(i, x) for(int i=head[x];i;i=e[i].nxt)
#define opf(s) freopen(s".in", "r", stdin), freopen(s".out", "w", stdout)
using namespace std;
const int NR=1e6+10;
int n, k, a[NR];
deque<int>q1, q2;
inline int read()
{
int res=0;
bool f=false;
char c=getchar();
while(!isdigit(c))f|=(c=='-'), c=getchar();
while(isdigit(c))res=res*10+(c-'0'), c=getchar();
return (f?-res:res);
}
int main()
{
n=read(), k=read();
rp(i, n)a[i]=read();
rp(i, n)
{
while(!q1.empty() && a[q1.back()]>=a[i])q1.pop_back();
q1.push_back(i);
while(!q1.empty() && q1.front()<=i-k)q1.pop_front();
if(i>=k)printf("%d ", a[q1.front()]);
}
puts("");
rp(i, n)
{
while(!q2.empty() && a[q2.back()]<=a[i])q2.pop_back();
q2.push_back(i);
while(!q2.empty() && q2.front()<=i-k)q2.pop_front();
if(i>=k)printf("%d ", a[q2.front()]);
}
return 0;
}