图论 tarjan 求 LCA 祖孙询问

【题目描述】：

已知一棵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

【时间限制、数据范围及描述】：

时间：1s 空间：128M

对于30%的数据，n,m≤1000。

对于100%的.据,n,m≤40000，每个节点的编号都不超过40000。

AC代码：

#include<cstdio>
#include<iostream>
using namespace std;
struct node{
	int u,v,next,lca;
}q[40005*2],e[40005*2];
int n,fa[40005*2],cnt,hq[40005*2],he[40005*2],p,m;
bool vis[40005*2];
void addq(int u,int v){
	cnt++;
	q[cnt].next=hq[u];
	hq[u]=cnt;
	q[cnt].v=v;
	q[cnt].u=u; 
}
void adde(int u,int v){
	cnt++;
	e[cnt].next=he[u];
	he[u]=cnt;
	e[cnt].v=v;
}
int find(int x){
	if(fa[x]==x)return x;
	else return fa[x]=find(fa[x]);
}
int dfs(int x){
	fa[x]=x;
	vis[x]=1;
	for(int i=he[x];i;i=e[i].next){
		if(!vis[e[i].v]){
			dfs(e[i].v);
			fa[e[i].v]=x;
		}
	}
	for(int i=hq[x];i;i=q[i].next){
		if(vis[q[i].v]){
			q[i].lca=find(q[i].v);
			if(i%2){
				q[i+1].lca=q[i].lca;
			}
			else{
				q[i-1].lca=q[i].lca;
			}
		}
	}
}
int query(int i){
	if(q[i].lca==q[i].u)return 2;
	if(q[i].lca==q[i].v)return 1;
	return 0;
}
int main(){
	int u,v;
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%d%d",&u,&v);
		if(v==-1){
			p=u;continue;
		}
		adde(u,v);
		adde(v,u);
	}
	cnt=0;scanf("%d",&m);
	for(int i=1;i<=m;i++){
		scanf("%d%d",&u,&v);
		addq(u,v);
		addq(v,u);
	}
	dfs(p);
	for(int i=1;i<=m;i++){
		printf("%d\n",query(i*2));
	}
	return 0;
}