poj 1330（LCA）

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <vector>
#include <set>
#include <queue>
#include <stack>
#include <climits>//形如INT_MAX一类的
using namespace std;

vector<int>lz[10005],que[10005];
int vis[10005],nopa[10005];
int n,m,pa[10005],s,e;

int find(int x)
{
if(pa[x] != x)
return pa[x] = find(pa[x]);
else
return x;
}

void LCA(int u)
{
int i,j;
for(i=0; i<lz[u].size(); i++)      //递归处理
{
LCA(lz[u][i]);
pa[lz[u][i]] = u;
}
vis[u] = 1;
for(i=0; i<que[u].size(); i++)     //询问处理
{
if(vis[que[u][i]])
{
cout << find(que[u][i]) << endl;
return;
}
}
}

int main()
{
int t,i,j,a,b;
cin >> t;
while(t--)
{
cin >> n;
for(i=1; i<=n; i++)       //初始化
{
lz[i].clear();
que[i].clear();
vis[i] = 0;
nopa[i] = 0;
pa[i] = i;
}
for(i=1; i<n; i++)
{
scanf("%d%d",&a,&b);
lz[a].push_back(b);      //算是单向边，a是b的父结点
nopa[b] = 1;
}
scanf("%d%d",&s,&e);
que[s].push_back(e);
que[e].push_back(s);         //询问两点连起关系
for(i=1; i<=n; i++)
{
if(!nopa[i]) //从总结点开始LCA
{
LCA(i);
}
}
}
return 0;
}