hdu 4605 线段树与二叉树遍历

时间:2024-10-07 16:35:50

思路:

首先将所有的查询有一个vector保存起来。我们从1号点开始dfs这颗二叉树,用线段树记录到当前节点时,走左节点的有多少比要查询该节点的X值小的,有多少大的,

同样要记录走右节点的有多少比X小的,多少比X大的。小和大的x,y值题目给了。当要进行左儿子时,建该节点值插入走左的线段树,回退的时候将其删除,进入右儿子时将其插入走右的线段树,同样回退时删除。遍历完一个树,整个查询就做完了,最后输出。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#define Maxn 200100
#define lson(x) (x<<1)
#define rson(x) ((x<<1)+1)
#define mid ((tree[po].l+tree[po].r)>>1)
#define inf 100000000
using namespace std;
struct Query{
int x,i;
};
struct AA{
int x,y;
}ans[Maxn];
struct Tree{
int l,r,lnum,rnum;
}tree[Maxn*];
int head[Maxn],e,w[Maxn],Index[Maxn],flag=,L,R,Lsum,Rsum;
vector<Query> query[Maxn];
struct Edge{
int u,v,next;
}edge[Maxn];
void buildTree(int l,int r,int po)
{
tree[po].l=l,tree[po].r=r;
tree[po].lnum=,tree[po].rnum=;
if(l==r)
return ;
buildTree(l,mid,lson(po));
buildTree(mid+,r,rson(po));
}
void Insert(int val,int po,int type)
{
if(tree[po].l==tree[po].r)
{
if(type)
tree[po].rnum++;
else
tree[po].lnum++;
return ;
}
if(type)
tree[po].rnum++;
else
tree[po].lnum++;
if(Index[mid]>=val)
Insert(val,lson(po),type);
else
Insert(val,rson(po),type);
}
void del(int val,int po,int type)
{
if(tree[po].l==tree[po].r)
{
if(type)
tree[po].rnum--;
else
tree[po].lnum--;
return ;
}
if(type)
tree[po].rnum--;
else
tree[po].lnum--;
if(Index[mid]>=val)
del(val,lson(po),type);
else
del(val,rson(po),type);
}
void get_Num(int val,int po)
{
if(Index[tree[po].l]>val)
return ;
if(Index[tree[po].r]<val)
{
R+=tree[po].rnum;
L+=tree[po].lnum;
return ;
}
if(tree[po].l==tree[po].r)
{
if(tree[po].rnum||tree[po].lnum)//如果图中有相同数值,则不能到达
flag=;
return ;
}
get_Num(val,lson(po));
get_Num(val,rson(po));
}
void init()
{
e=;
Lsum=Rsum=;
memset(head,-,sizeof(head));
for(int i=;i<=;i++)
{
query[i].clear();
ans[i].x=ans[i].y=;
}
}
void add(int u,int v)
{
edge[e].u=u;edge[e].v=v;edge[e].next=head[u];head[u]=e++;
}
void dfs(int u)//进行二叉树遍历
{
int i,j,temp,Size,pos;
Size=query[u].size();
for(i=;i<Size;i++)
{
L=R=flag=;
get_Num(query[u][i].x,);
if(flag)
{
ans[query[u][i].i].y=-;
continue;
}
ans[query[u][i].i].y+=L*;
ans[query[u][i].i].y+=Lsum-L;
ans[query[u][i].i].y+=R*;
ans[query[u][i].i].y+=Rsum-R;
ans[query[u][i].i].x+=R;
}
int f=;
for(i=head[u];i!=-;i=edge[i].next)
{
if(f)
{
Insert(w[u],,);
Rsum++;
}
else
{
Insert(w[u],,);
Lsum++;
}
f++;
dfs(edge[i].v);
if(f==)
{
del(w[u],,);
Rsum--;
}
else
{
del(w[u],,);
Lsum--;
}
}
}
int main()
{
int n,m,q,t,i,j,a,b,u,num;
scanf("%d",&t);
while(t--)
{
init();
num=;
scanf("%d",&n);
for(i=;i<=n;i++)
{
scanf("%d",w+i);
Index[i]=w[i];
}
sort(Index+,Index+n+);
num=;
for(i=;i<=n;i++)//将坐标离散化
if(Index[i]>Index[num])
Index[++num]=Index[i];
buildTree(,num,);//建立线段树
scanf("%d",&m);
for(i=;i<=m;i++)
{
scanf("%d%d%d",&u,&a,&b);
add(u,b);
add(u,a);
}
scanf("%d",&q);
Query temp;
int v;
for(i=;i<=q;i++)//将所有查询保存起来
{
scanf("%d%d",&v,&temp.x);
temp.i=i;
query[v].push_back(temp);
}
int f=;//标记走左还是右
for(i=head[];i!=-;i=edge[i].next)//从1号节点开始遍历
{
if(f)//若走右
{
Insert(w[],,);//将该值插入右线段树,并且总数加1
Rsum++;
}
else
{
Insert(w[],,);//插入左线段树,总数加1
Lsum++;
}
f++;
dfs(edge[i].v);
if(f==)
{
del(w[],,);//从右儿子退出,将其删除,总数减1
Rsum--;
}
else
{
del(w[],,);//从左儿子退出
Lsum--;
}
}
for(i=;i<=q;i++)
{
if(ans[i].y!=-)
printf("%d %d\n",ans[i].x,ans[i].y);
else
printf("0\n");
}
}
return ;
}