祖孙询问

【问题描述】
已知一棵 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。

开始没想那么多= =直接用的并查集，结果超时只得了60分。

实际上这道题应该用bfs+lca（dfs也可以，但是本人觉得bfs要好一点）

#include<cstdio>//lca+bfs/dfs
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=400005;
struct edge
{
	int to;
	int next;
}e[maxn];
int n;
int m;
int x,y,a,b;
int k=0;
int head[maxn];
int root;
int sb[16];
int deep[maxn];//记录结点深度
int queue[maxn];//
int father[maxn][16];//i的10^j次方的父亲 
bool visit[maxn];

void add(int u,int v)
{
	k++;
	e[k].to=v;
	e[k].next=head[u];
	head[u]=k;
}
void bfs()
{
	int qhead=0;
	int tail=1;
	queue[qhead]=root;
	visit[root]=1;
	while(qhead<tail)
	{
		int now=queue[qhead++];//
		for(int i=1;i<=15;i++)
		{
			if(sb[i]<=deep[now])
			{
				father[now][i]=father[father[now][i-1]][i-1];
			//	printf("i=%d f=%d\n",i,father[now][i]);
			}
			else break;
		}
		int p=head[now];
		while(p!=0){
			if(!visit[e[p].to])
			{
			   // printf("hehe=%d\n",e[p].to);
				deep[e[p].to]=deep[now]+1;
				father[e[p].to][0]=now;
			//	printf("now=%d\n",now);
				visit[e[p].to]=1;
				queue[tail++]=e[p].to;
				
			}
			p=e[p].next;
		}
	}
}
int lca(int a,int b)//查找最近公共祖先 
{
	if(deep[a]<deep[b])//默认a的深度大于b所以这里做一个判定 
	{
		int temp=a;
		a=b;
		b=temp;
	}
//	printf("a=%d b=%d\n",a,b);
	int t=deep[a]-deep[b];
	//printf("t=%d\n",t);
	for(int i=0;i<=15;i++)
	{
		if(t&sb[i])//
		{
			a=father[a][i];
		//	printf("father=%d\n",father[x][i]);
		}
	}
	for(int i=15;i>=0;i--)
	{
		if(father[a][i]!=father[b][i])
		{
			a=father[a][i];
			b=father[b][i];
		}
	}
//	printf("a=%d b=%d\n",a,b);
	if(a==b) return b;
	return father[a][0];
}
int main()
{
	freopen("tree.in","r",stdin);
	freopen("tree.out","w",stdout);
	sb[0]=1;
	for(int i=1;i<=15;i++)
	{
		sb[i]=sb[i-1]<<1;//倍增 bin[i-1]*2; 
	}
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d %d",&a,&b);
		if(b==-1) root=a;//找到根节点 
		else//如果不是根节点则建立祖先和儿子的关系 
		{
			add(a,b);
			add(b,a);
		}
	}
	memset(visit,0,sizeof(visit));
	
	bfs();//printf("-------------------");
	scanf("%d",&m);
	for(int i=1;i<=m;i++)
	{
		scanf("%d %d",&x,&y);
		int t=lca(x,y);//最近公共祖先 
	//	printf("t=%d\n",t);
		if(x==t&&t!=y) printf("1\n"); 
		else if(y==t&&x!=t) printf("2\n");
		else printf("0\n");
	}
	return 0;
}