线段树模板题,但是掉坑很多次:
(1)
普通的区间查询这里都是>=和<=,但是这里只能是==,因为如果是【1,3】进来的,tree[k]的范围是【1,8】就会直接进去了。
(2)
跨区间的时候有两种写法:
1
2
但是第二种写法在这题很容易崩,所以推荐用第一种写法。
(3)
查询的时候因为同时要查询maxn和minn,而可以:
在形参处加个flag,在返回的时候分类讨论,但是这样对于每一个窗口就要调用两次Query,如果数字一大搞不好容易超时,所以推荐用一个全局变量同时返回maxn和minn.
代码:
//F
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int MAXN=1e6+5;
const int INF=0x7f7f7f7f;
struct Node
{
int l,r;
int Max,Min;
}Tree[4*MAXN];
int Maxn[MAXN],Minn[MAXN];
int n,k;
int ans1,ans2;
void Build(int k,int l,int r)
{
Tree[k].l=l;
Tree[k].r=r;
if(l==r)
{
int w;
scanf("%d",&w);
Tree[k].Max=Tree[k].Min=w;
return;
}
int mid=(l+r)/2;
Build(2*k,l,mid);
Build(2*k+1,mid+1,r);
Tree[k].Max=max(Tree[2*k].Max,Tree[2*k+1].Max);
Tree[k].Min=min(Tree[2*k].Min,Tree[2*k+1].Min);
}
void Query(int k,int l,int r)
{
if(Tree[k].l==l&&Tree[k].r==r)
{
ans1=max(Tree[k].Max,ans1);
ans2=min(Tree[k].Min,ans2);
return;
}
int mid=(Tree[k].l+Tree[k].r)/2;
if(r<=mid)
Query(2*k,l,r);
else if(l>mid)
Query(2*k+1,l,r);
else
{
Query(2*k,l,mid);
Query(2*k+1,mid+1,r);
}
}
int main()
{
int i;
scanf("%d%d",&n,&k);
Build(1,1,n);
for(i=1;i<=n-k+1;i++)
{
ans1=-INF;
ans2=INF;
Query(1,i,i+k-1);//(节点序号,左边界,右边界,flag(flag=1表示求最大值,flag=2表示求最小值))
Maxn[i]=ans1;
Minn[i]=ans2;
}
printf("%d",Minn[1]);
for(i=2;i<=n-k+1;i++)
printf(" %d",Minn[i]);
printf("\n");
printf("%d",Maxn[1]);
for(i=2;i<=n-k+1;i++)
printf(" %d",Maxn[i]);
printf("\n");
return 0;
}