祖孙询问4.11

题目描述

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

输入样例#1：

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：

1
0
0
0
2

说明

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

lca模板题 模板可见我的博客点击打开链接

#include<bits/stdc++.h>
#define V (to[i])
#define N 500010
using namespace std;
int tot=0,ti=0;
int next[N],head[N],to[N],in1[N],out1[N],fa[N][25],dis[N],n,m;
inline void add(int x,int y) {
	tot+=1;
	next[tot]=head[x];
	to[tot]=y;
	head[x]=tot;
}
inline void dfs(int x) {
	in1[x]=++ti;
	for(int i=head[x]; i; i=next[i]) {
		if(fa[x][0]^V) {
			fa[V][0]=x;
			dis[V]=dis[x]+1;
			dfs(V);
		}
	}
	out1[x]=++ti;
}
inline bool pd(int x,int y) {
	return (in1[x]<=in1[y]&&out1[x]>=out1[y]);
}
inline int get_lca(int x,int y) {
	if(x==y)
		return x;
	if(dis[x]<dis[y])
		swap(x,y);
	for(int i=20; ~i; i--)
		if(!pd(fa[x][i],y))
			x=fa[x][i];
	x=fa[x][0];
	return x;
}
int main() {
	int x,y,root;
	scanf("%d",&n);
	for(int i=1; i<=n; i++) {
		scanf("%d%d",&x,&y);
		if(y==-1) {
			root=x;
			continue;
		}//记录根节点 
		add(x,y);
		add(y,x);
	}
	fa[root][0]=root;
	dfs(root);
	for(int i=1; i<=20; i++)
		for(int j=1; j<=40005; j++)//注意(点的编号不超过40000)
			fa[j][i]=fa[fa[j][i-1]][i-1];
	scanf("%d",&m);
	for(int i=1; i<=m; i++) {
		scanf("%d%d",&x,&y);
		int oo=get_lca(x,y);//找到最近公共祖先 
		if(oo==x)puts("1");
		else if(oo==y)puts("2");
		else puts("0");
	}
}