poj1330,JDOJ3055O(nlogn)做法

昨天发了个暴力做法,今天发个nlogn做法.

poj1330需要稍作改动.

#include<stdio.h>
int f[100001][18];//倍增,f[i][j]表示以i为起点跳2^j步的父亲.
int level[100001];//表示i点的层数是多少
int ans[100001][50];//ans[i][j]表示第i点有第j个儿子.
int idx[100001],tmp;
int dfs(int i)
{
	level[i]=level[f[i][0]]+1;//倍增更新层数
	for(int j=1;j<17;j++)
		f[i][j]=f[f[i][j-1]][j-1];//动规方程(核心)
	for(int j=idx[i];j>0;j--)
		dfs(ans[i][j]);//注意调用dfs那里,如果f[i][0]=0,说明这个点是总根.
}
int lca(int a,int b)
{
	if(level[a]<level[b])
		tmp=a,a=b,b=tmp;
	int deep=level[a]-level[b];
	for(int i=0;i<17;i++)
	{
		if((deep&(1<<i))!=0)
			a=f[a][i];
	}//先让层数浅的跳到层数深的地方.
	if(a==b)
		return a;//一样的话,a或b就是LCA;
	for(int j=16;j>=0;j--)
	{
		if(f[a][j]!=f[b][j])
			a=f[a][j],b=f[b][j];
	}
	return f[a][0];//这次跳到不是他们共同父亲的地方,所以是a或b的下一个.
}
int main()
{
	int n,m,a,b,x,y;
	scanf("%d%d",&n,&m);
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&a,&b);
		f[b][0]=a;
		ans[a][++idx[a]]=b;
	}
	for(int i=1;i<=n;i++)
	{
		if(f[i][0]==0)
		{
			dfs(i);
			break;
		}
	}
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		printf("%d\n",lca(x,y));//下面的就不解释了
	}
}