P1886 滑动窗口 /【模板】单调队列 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
单调队列版子题
为研究树网的核的第四种写法,我去学习了单调队列。单调队列顾名思义,要建立一个队列,并且使这个队列单调,建立单调队列的目是及时排除一定不是最优解的情况。比如这道题:
让你寻找区间K内的极值(不妨设极小值),建立一个队列p为单调队列,存K区间内有可能作为答案的数,队列q存对应队列p的编号。当新加入一个数,判断和队首的编号是否超过k,若超过,则弹出队首元素。此时由于找极小值,若此元素比队尾的小,则队尾的元素不再具有价值,因为当前区间不可能选它,之后新加入数也不会选,因此把队尾所有大于等于新加入元素的元素弹出,这就叫做及时排除一定不是最优解的选择。
此队列为双端队列,用STL的deque书写较为方便(有个细节是删去队尾元素时注意队列清空的情况),代码如下:
#include <bits/stdc++.h>
using namespace std;
const int N=1000001;
deque<int> p,q;
int n,k,a[N];
void minn()
{
q.push_front(1);
p.push_front(a[1]);
if(k==1) cout<<a[1]<<" ";
for(int i=2;i<=n;i++)
{
if(i-q.front()>=k)
{
q.pop_front();
p.pop_front();
}
while(a[i]<=p.back()&&q.size())
{
q.pop_back();
p.pop_back();
}
q.push_back(i);
p.push_back(a[i]);
if(i>=k) cout<<p.front()<<" ";
}
}
void maxx()
{
q.clear();
p.clear();
q.push_front(1);
p.push_front(a[1]);
if(k==1) cout<<a[1]<<" ";
for(int i=2;i<=n;i++)
{
if(i-q.front()>=k)
{
q.pop_front();
p.pop_front();
}
while(a[i]>=p.back()&&q.size())
{
p.pop_back();
q.pop_back();
}
q.push_back(i);
p.push_back(a[i]);
if(i>=k) cout<<p.front()<<" ";
}
}
int main()
{
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
minn();
cout<<endl;
maxx();
}