题面
题目描述
给你一个长度为 N 的数组,一个长为 K 的滑动的窗体从最左移至最右端, 你只能见到窗口的K个数,每次窗体向右移动一位,如下表:
你的任务是找出窗口在各位置时的 max value,min value.
输入格式
第 1 行 n,k, 第 2 行为长度为 n 的数组
输出格式
2 行, 第 1 行每个位置的 min value, 第 2 行每个位置的 max value
题解
单调队列算法。
这个算法很骚。。。
我们需要维护两个单调队列,一个递增,一个递减。
每次查询只要查询两个队列的队头就可以了。
具体怎么维护呢?
递增队列我们称为
qmin
q
m
i
n
,递减队列我们称为
qmax
q
m
a
x
.
开始先让
k
k
<script id="MathJax-Element-133" type="math/tex">k</script>个数进队,然后再每次读入一个数再让它进队。
进队的时候先保证这个队列是单调的,然后再弹出队头与当前进队的元素距离超过k个的。
然后进队。
code
#include<bits/stdc++.h>
using namespace std;
inline int read(){
int num=0;char c=' ';bool flag=true;
for(;c>'9'||c<'0';c=getchar())
if(c=='-')
flag=false;
for(;c>='0'&&c<='9';num=(num<<3)+(num<<1)+c-48,c=getchar());
return flag ? num : -num;
}
const int maxn=1000020;
int n,k;
struct node{
int num,pos;
}qmax[maxn],qmin[maxn];
//qmax鍗曡皟閫掑噺
//qmin鍗曡皟閫掑
int maxhead,maxtail,minhead,mintail;
void init(){
n=read();
k=read();
for(int i=1;i<=k;i++){
int t=read();
while(maxhead<=maxtail&&qmax[maxtail].num<t)maxtail--;
qmax[++maxtail]=(node){t,i};
while(minhead<=mintail&&qmin[mintail].num>t)mintail--;
qmin[++mintail]=(node){t,i};
}
}
int maxans[maxn],minans[maxn],top=0;
void work(){
for(int i=k+1;i<=n;i++){
int t=read();
maxans[++top]=qmax[maxhead].num;
minans[top]=qmin[minhead].num;
while(maxhead<=maxtail&&qmax[maxtail].num<t)maxtail--;
while(maxhead<=maxtail&&i-qmax[maxhead].pos>=k)maxhead++;
qmax[++maxtail]=(node){t,i};
while(minhead<=mintail&&qmin[mintail].num>t)mintail--;
while(minhead<=mintail&&i-qmin[minhead].pos>=k)minhead++;
qmin[++mintail]=(node){t,i};
}
maxans[++top]=qmax[maxhead].num;
minans[top]=qmin[minhead].num;
}
void print(){
for(int i=1;i<top;i++)
printf("%d ",minans[i]);
printf("%d\n",minans[top]);
for(int i=1;i<top;i++)
printf("%d ",maxans[i]);
printf("%d\n",maxans[top]);
}
int main(){
init();
work();
print();
return 0;
}