LibreOJ - 10135 祖孙询问（LCA查询最近公共祖先）

链接

题意：给出两个数，查询这两个数是否在同一个大的分支里
分析：模板题，最后LCA查询的时候如果一个是另一个的祖先，则满足条件。

#include<bits/stdc++.h>
using namespace std;
const int N=40010,D=50;
int head[N],deep[N],fa[N][1000];//1000代表层数，要尽量大点，否则结果出错 
int num; 
struct node
{
	int to,nex;
}e[2*N];
void add(int x,int y)
{
	num++;
	e[num].to=y;
	e[num].nex=head[x];
	head[x]=num;
}
void dfs(int u)
{
	for(int i=head[u];i;i=e[i].nex)
	{
		if(!deep[e[i].to])
		{
			deep[e[i].to]=deep[u]+1;
			fa[e[i].to][0]=u;
			dfs(e[i].to);
		}
	}
}
int lca(int x,int y)
{
	int flag=0;
	if(deep[x]<deep[y])//x较深 
	{
		flag=1;
		swap(x,y);
	}
	for(int i=D;i>=0;i--)
	{
		if(deep[y]<=deep[fa[x][i]])
		x=fa[x][i];
	}
	if(x==y)
	{
		
		if(flag) 
		return 1;
		return 2;
	}
	return 0;
}
int main()
{
	int x,y,s,q,n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d%d",&x,&y);
		if(y==-1)
		s=x;
		else
		{
		add(x,y);
		add(y,x);	
		}
	}
	deep[s]=1;
	dfs(s);
	for(int i=1;i<=D;i++) 
	for(int j=1;j<=40000;j++)
	fa[j][i]=fa[fa[j][i-1]][i-1];
	scanf("%d",&q);
	for(int i=1;i<=q;i++)
	{
		scanf("%d%d",&x,&y);
		printf("%d\n",lca(x,y));
	}
}