题目大意:
给你一棵树,找出两点的最近公共祖先LCA。
输入:第1行包含一个整数T,表示测试用例的数量。每个测试用例的第1行都包含整数N(2≤N≤10,000),表示树中的节点数。节点用1~N标记。接下来N-1行,每行都包含一对边的整数,第1个整数是第2个整数的父节点(有N个节点的树有N-1条边)。每个测试用例的最后一行都包含两个不同的整数,求其LCA。
输入样例:
2
16
1 14
8 5
10 16
5 9
4 6
8 4
4 10
1 13
6 15
10 11
6 7
10 2
16 3
8 1
16 12
16 7
5
2 3
3 4
3 1
1 5
3 5
输出样例:
4
3
题解:由于本题数据量不大,所以可以暴力求解最近公共祖先LCA。
#include<cstdio>
using namespace std;
const int maxn=10010;
int fa[maxn];//树的存储,使用父亲表示法,把一条边的信息存储在子节点上,fa[i]=j,表示子节点的父亲是i
bool flag[maxn];
void Init(int n){
for(int i=1;i<=n;i++){
fa[i]=i;
flag[i]=0;
}
}
int LCA(int u,int v){
if(u==v)
return u;
flag[u]=1;
while(fa[u]!=u){//u向上走到根,并标记沿途进过的节点
u=fa[u];
flag[u]=1;
}
if(flag[v]) //如果v节点被访问过,那么v就是u,v的最近公共祖先
return v;
while(fa[v]!=v){//v向上,找沿途被访问过的节点,访问过的节点就是u,v的最近公共祖先
v=fa[v];
if(flag[v])
return v;
}
return 0;
}
int main(){
int n,u,v,T;
scanf("%d",&T);
while(T--){
scanf("%d",&n);
Init(n);
for(int i=1;i<n;i++){
scanf("%d%d",&u,&v);
fa[v]=u;
}
scanf("%d%d",&u,&v);
printf("%d\n",LCA(u,v));
}
return 0;
}