【问题描述】
已知一棵n个节点的有根树。有m个询问。每个询问给出了一对节点的编号x和y,询问x与y的祖孙关系。
【输入格式】
输入第一行包括一个整数n表示节点个数。
接下来n行每行一对整数对a和b表示a和b之间有连边。如果b是-1,那么a就是树的根。
第n+2行是一个整数m表示询问个数。
接下来m行,每行两个正整数x和y。
【输出格式】
对于每一个询问,输出1:如果x是y的祖先,输出2:如果y是x的祖先,否则输出0。
【样例输入】
10
234 -1
12 234
13 234
14 234
15 234
16 234
17 234
18 234
19 234
233 19
5
234 233
233 12
233 13
233 15
233 19
【样例输出】
1
0
0
0
2
【数据规模】
对于30%的数据,n,m≤1000。
对于100%的.据,n,m≤40000,每个节点的编号都不超过40000。
题解
我写了LCA,BFS序更快。
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
int n,m,zz,head[40002],root;
struct bian{int to,nx;} e[80002];
int fa[40002][20],h[40002];//2^16=65536
void insert(int x,int y)
{
zz++; e[zz].to=y; e[zz].nx=head[x]; head[x]=zz;
zz++; e[zz].to=x; e[zz].nx=head[y]; head[y]=zz;
}
void init()
{
int i,x,y;
scanf("%d",&n);
for(i=1;i<=n;i++)
{scanf("%d%d",&x,&y);
if(y==-1) root=x;
else insert(x,y);
}
}
void dfs(int x)
{
int i,p;
for(i=1;i<=15;i++)
{if(h[x]<(1<<i)) break;
fa[x][i]=fa[fa[x][i-1]][i-1];
}
for(i=head[x];i;i=e[i].nx)
{p=e[i].to;
if(p==fa[x][0]) continue;
h[p]=h[x]+1; fa[p][0]=x;
dfs(p);
}
}
int lca(int x,int y)
{
if(h[x]<h[y]) swap(x,y);
int i,c;
c=h[x]-h[y];
for(i=15;i>=0;i--)
{if(c&(1<<i)) x=fa[x][i];}
for(i=15;i>=0;i--)
{if(fa[x][i]!=fa[y][i])
{x=fa[x][i]; y=fa[y][i];}
}
if(x==y) return x;
else return fa[x][0];
}
void work()
{
int i,x,y,t;
scanf("%d",&m);
dfs(root);
for(i=1;i<=m;i++)
{scanf("%d%d",&x,&y);
if(x==y) printf("0\n");
else
{t=lca(x,y);
if(t==x) printf("1\n");
else if(t==y) printf("2\n");
else printf("0\n");
}
}
}
int main()
{
freopen("tree.in","r",stdin);
freopen("tree.out","w",stdout);
init(); work();
return 0;
}