T70138 寻找中点

T70138 寻找中点
题目描述

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

说明

【样例说明】

234是233的祖先，19是233的祖先。

【数据范围】

对30%的输入数据 ：n,m≤1000

对100%的输入数据 ：n,m≤40000，每个节点的编号大于0，都不超过 40000。

#include<bits/stdc++.h>
using namespace std;
const int N=100009;
int n,q,x,y,geng,first[N*2],next[N*2],to[N*2],dep[N],f[N][21],tot;
void add(int u,int v)
{
	next[++tot]=first[u];first[u]=tot;to[tot]=v;
	next[++tot]=first[v];first[v]=tot;to[tot]=u;
}
void Init(int u,int father)
{
	dep[u]=dep[father]+1;
	for(int i=0;i<=19;i++)
	{
		f[u][i+1]=f[f[u][i]][i];
	}
	for(int e=first[u];e;e=next[e])
	{
		int v=to[e];
		if(v==father) continue;
		f[v][0]=u;
		Init(v,u);
	}
}
int find(int x,int y)
{
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=20;i>=0;i--)
	{
		if(dep[f[x][i]]>=dep[y]) x=f[x][i];
		if(x==y) return x;
	}
	for(int i=20;i>=0;i--)
	{
		if(f[x][i]!=f[y][i])
		{
			x=f[x][i];
			y=f[y][i];
		}
	}
	return f[x][0];
}
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>x>>y;
		if(y==-1)
		{
			geng=x;
			continue;
		}
		add(x,y);
	}
	Init(geng,0);
	cin>>q;
	for(int i=1;i<=q;i++)
	{
		cin>>x>>y;
		int ans=find(x,y);
		if(ans==x) cout<<1<<endl;
		else if(ans==y) cout<<2<<endl;
		else cout<<0<<endl;
	}
	return 0;
}