POJ 1470 Closest Common Ancestors LCA离线Tarjin算法

http://poj.org/problem?id=1470

题意：给定一颗有根树，并且有Q次询问，格式是(u，v)，求任意一次询问的两个结点的最近公共祖先。

算法：裸的LCA ，这是LCA的第一题，首先说下LCA离线Tarjin算法，其实质就是dfs ＋并查集。首先将所有的查询都先记录下面（因为是离线算法），然后就是从根结点对树进行一次ｄｆｓ，复杂度为O(E) ，注意树用邻接矩阵存储。当我们从一个结点假设为u开始对树进行dfs的时候，首先选出一个未扩展的结点v对其进行扩展，在回溯的时候，询问的两个结点都在以结点v为根的子树中的询问都已经有了结果了，有结点不在v的子树中的询问的结果还没有得出结果，这时候我们利用并查集将v所在的子树和u结点合并（即将v的father设置为u），这样在我们下去找到这样一个结点：该结点属于某次询问，而且这次询问的另外一个结点已经被遍历过了，这时候我们只要找到这个另外结点所在集合的father就是这次查询的LCA。这个证明可以由dfs的性质和结点合并（也就是将结点合并到其父结点的集合中去）的时候决定。这样当dfs遍历完了整棵树之后，也就得到了所有查询的结果。时间复杂度为：O(N + Q)，N为结点树（PS：有些地方这里是说E，其实是一样的， 因为这是一颗树），Q为询问次数。

代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
const int MAXN = 900+10 ;
int N ,Q ;
struct Node{
	int num ;
	int next ;
}edge[MAXN];
int cnt1,root[MAXN] ;
bool use[MAXN] ;

struct Node2{
	int num,next ;
	int f ;
}e2[500000] ;
int cnt2,root2[MAXN] ;
int ans[500000] ;
bool vis[MAXN] ;
int f[MAXN] ;
int res[MAXN] ;

void Init(){
	cnt1 = cnt2 = 0 ;
	memset(root ,-1 ,sizeof(root));
	memset(use, 0 ,sizeof(use));	
	memset(root2 , -1, sizeof(root2)) ;
	memset(vis,  0 ,sizeof(vis));
	for(int i=1;i<=N;i++)	f[i] = i ;
	memset(res ,0,sizeof(res));
}
void add(int a, int b){
	edge[cnt1].num = b;
	edge[cnt1].next = root[a] ;
	root[a] = cnt1 ++ ;
}
void add2(int a, int b ,int f){
	e2[cnt2].num = b ;
	e2[cnt2].f = f ;
	e2[cnt2].next = root2[a] ;
	root2[a] = cnt2++ ;
}
int find(int u){
	if(u != f[u]){
		f[u] = find( f[u] ) ;
	}
	return f[u] ;
}
void tarjin(int u){
	vis[u] = 1 ;
	for(int i=root2[u] ;i!=-1;i=e2[i].next){
		int f = e2[i].f;
		int v = e2[i].num ;
		if(vis[v]){
			ans[f] = find(v);
		}	
	}
	for(int i=root[u] ;i!=-1;i=edge[i].next){
		int v = edge[i].num;
		if(!vis[v]){
			tarjin(v) ;
			f[v] = u ;
		}
	}
}
int main(){
	int a ,b ,c ,i,j;
	while(scanf("%d",&N) == 1){
		Init() ;
		for(i=1;i<=N;i++){
			scanf("%d",&a);
			while(getchar()!='(') ;
			scanf("%d",&b);
			while(getchar() != ')') ;
			for(j=1;j<=b;j++){
				scanf("%d",&c);
				add(a,c) ;
				use[c] = 1 ;
			}
		}
		scanf("%d",&Q);
		for(i=1;i<=Q;i++){
			while(getchar() != '(') ;
			scanf("%d %d",&a,&b);
			add2(a,b,i) ;
			add2(b,a,i) ;
			while(getchar()!=')') ; 
		}
		for(i=1;i<=N;i++){
			if(use[i] == 0)	break ;
		}
		tarjin(i);
		for(int i=1;i<=Q;i++){
			res[ ans[i] ]++ ;
		}
		for(int i=1;i<=N;i++){
			if(res[i] == 0)	continue ;
			printf("%d:%d\n",i,res[i]);
		}
	}	
	return 0;
}