poj 1330

先贴代码：

#include<iostream>
#include<cstdio>
#include<vector>
#include<string>
#include<string.h>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = (int)1e4+5;
int r[maxn];
int f[maxn];
int ansc[maxn];
int dgr[maxn];
bool visit[maxn];
vector<int> c[maxn];
vector<int> q[maxn];
int findf(int x){
	if(x != f[x]){
		f[x] = findf(f[x]);
	}
	return f[x];
}

int unify(int x,int y){
	int fx = findf(x);
	int fy = findf(y);
	if(fx == fy){
		return 0;
	}
	if(r[fx] < r[fy]){
		f[fx] = fy;
		r[fy] += r[fx];
	}
	else{
		f[fy] = fx;
		r[fx] += r[fy];
	}
	return 1;
}
int n;
void init(){
	for(int i = 0; i < n; ++i){
		r[i] = 1;
		f[i] = i;
		dgr[i] = 0;
		visit[i] = false;
		c[i].clear();
		q[i].clear();
		ansc[i] = 0;
	}
}
int LCA(int x){
	ansc[x] = x;
	int size = c[x].size();
	for(int i = 0; i < size; ++i){
		LCA(c[x][i]);
		unify(x,c[x][i]);
		ansc[findf(x)] = x;
	}
	visit[x] = true;
	size = q[x].size();
	for(int i = 0; i < size; ++i){
		if(visit[q[x][i]]){
			printf("%d\n",ansc[findf(q[x][i])] + 1);
			return 0;
		}
	}
	return 0;
}

int main(){
	int t;
	scanf("%d", &t);
	while(t--){
		scanf("%d", &n);
		init();
		int a, b;
		for(int i = 0; i < n - 1; ++i){
			scanf("%d%d",&a,&b);
			--a;
			--b;
			c[a].push_back(b);
			dgr[b]++;
		}
		scanf("%d%d",&a,&b);
		--a; --b;
		q[a].push_back(b);
		q[b].push_back(a);
		for(int i = 0; i < n; ++i){
			if(dgr[i] == 0){
				LCA(i);
				break;
			}
		}
	}
	return 0;
}

Tarjan是解决树中lca问题的离线算法。

基于深度优先周游和并查集。

转载于:https://my.oschina.net/u/1421373/blog/379569