题目大意是给你一个长度为n的序列,长度为k的窗口,窗口不断滑动,求每次移动时窗口内的最大值和最小值。
据说是单调队列裸题。我用G++语言交就会TLE,用C++交就A掉了…
我们开两个双端队列来分别维护一个单调递减的队列和一个单调递增的队列。以维护递增队列为例,因为他前面出现的值小,所以肯定不会是最优解,就可以把它pop掉。这是正确的。每次读入时我们把它push进去,push的时候要不断维护更新这个队列,同时我们要判断一下队列有没有超出长度限制。这样的队头就是这个窗口中最大或最小的值了。
代码如下:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
int n,k;
struct Num{
int val,num;
};
deque<Num>q;
void push(int x,int y)
{
while(!q.empty()&&q.back().val<=x)
q.pop_back();
Num a;
a.val=x;
a.num=y;
q.push_back(a);
}
int a[1000000+10];
deque<Num>q2;
void push2(int x,int y)
{
while(!q2.empty()&&q2.back().val>=x)
q2.pop_back();
Num a;
a.val=x;
a.num=y;
q2.push_back(a);
}
int main()
{
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
push2(a[i],i);
while(i-k>=q2.front().num)
q2.pop_front();
if(i>=k)
printf("%d ",q2.front().val);
}
puts("");
for(int i=1;i<=n;i++)
{
push(a[i],i);
while(i-k>=q.front().num)
q.pop_front();
if(i>=k)
printf("%d ",q.front().val);
}
return 0;
}
因为是区间最大值和最小值,而且没有更新操作,所以也比较容易的考虑到用ST表,但是MLE了…我觉得正确性应该没什么问题吧。还是不要脸的把代码放上来吧,请不要盲目复制粘贴哦>_<
#include<cstdio>
#include<cmath>
#include<iostream>
using namespace std;
const int MAXN=1000000+10;
int a[MAXN],g[MAXN][20],f[MAXN][20];
int main()
{
int n,k;
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(int i=1;i<=n;i++)
g[i][0]=f[i][0]=a[i];
for(int i=1,j=2;j<=n;i++,j*=2)
{
for(int s=1;s<=n;s++)
{
f[s][i]=max(f[s][i-1],f[s+(j>>1)][i-1]);
g[s][i]=min(g[s][i-1],g[s+(j>>1)][i-1]);
}
}
int l=1,r=k;
while(r<=n)
{
int p=int(log(r-l+1)/log(2));
int minn=min(g[l][p],g[r-(1<<p)+1][p]);
printf("%d ",minn);
l++;
r++;
}
puts("");
l=1,r=k;
while(r<=n)
{
int p=int(log(r-l+1)/log(2));
int maxx=max(f[l][p],f[r-(1<<p)+1][p]);
printf("%d ",maxx);
l++;
r++;
}
return 0;
}