题目大意:
给你一棵树,找出两点的最近公共祖先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
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;iscanf("%d%d",&u,&v);fa[v]=u;}scanf("%d%d",&u,&v);printf("%d\n",LCA(u,v));}return 0;
}
上一篇:React插槽