2016夏季练习——LCA

来源：POJ1330

这个是倍增法的写法，回头将Tarjan算法的方式写上

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <vector>
using namespace std;
const int MAXN = 10000 +10;
int n,rt;
int l,r;
vector<int>tree[MAXN];
bool root[MAXN];
int depth[MAXN];
int parent[MAXN][25];
void dfs(int cur,int pre,int d){
//	cout<<cur<<" "<<pre<<" "<<d<<endl;
	depth[cur]=d;//当前的层数
	parent[cur][0]=pre;//想上一层，就是其父亲节点
	for(int i=0;i<tree[cur].size();i++){
		if(tree[cur][i]!=pre) dfs(tree[cur][i],cur,d+1);//这一步也是显然的
	}
}
void ini(){
	dfs(rt,-1,0);//搜索，先得到所有的深度
	for(int k=0;k<=22;k++){
		for(int i=1;i<=n;i++){
			if(parent[i][k]<0) parent[i][k+1]=-1;//到了根节点，在怎么上去都是根节点
			else parent[i][k+1]=parent[parent[i][k]][k];//如果不是，就上去一半的长度
		}
	}
}
int LCA(int l,int r){
//	cout<<"l:r "<<l<<" "<<r<<endl;
	if(depth[l]>depth[r]) swap(l,r);
//	cout<<"l:r "<<l<<" "<<r<<endl;
	for(int i=0;i<22;i++){
//		cout<<i<<" l:r "<<l<<" "<<r<<endl;
//		cout<<"ans= "<<((depth[r]-depth[l])>>i&1)<<endl;
		if((depth[r]-depth[l])>>i&1) r=parent[r][i];//用二进制的差值来计算我们向上走的步数的组合
	}
//	cout<<"l:r "<<l<<" "<<r<<endl;
	if(r==l) return l;//如果就是同一个节点了，直接返回
	for(int i=21;i>=0;i--){
		if(parent[r][i]!=parent[l][i]){//否则一起向上
			l=parent[l][i];
			r=parent[r][i];
		}
	}
	//注意停止条件是不想等，但是由于我是逐步减小的做法，所以最后会停下来
//	cout<<"ddd2"<<endl;
	return parent[l][0];//向上一层，就是答案
}
int main(){
	int T,d;
	scanf("%d",&T);
	while(T--){
		scanf("%d",&n);
		memset(root,false,sizeof(root));
		memset(depth,0,sizeof(depth));
		//memset(parent,0,sizeof(parent));
		for(int i=1;i<=n;i++) tree[i].clear();
		int son,fa;
		for(int i=1;i<n;i++)
		{
			scanf("%d%d",&fa,&son);
			root[son]=1;
			tree[fa].push_back(son);
			tree[son].push_back(fa);
		}
		rt=0;
		for(int i=1;i<=n;i++){
			if(!root[i]){
				rt=i;
				break;
			}
		}
		ini();
		scanf("%d%d",&l,&r);
		cout<<LCA(l,r)<<endl;
	}
	return 0;
}