蓝桥备赛之暴力解决最近的共同祖先

最近的共同祖先

时限：1000毫秒		内存限制：10000K
提交总数： 47290		接受： 22609

描述

有根树是计算机科学和工程中众所周知的数据结构。示例如下所示：
 

在图中，每个节点都标有 {1， 2,...,16} 中的整数。节点 8 是树的根。如果节点 x 位于根和节点 y 之间的路径中，则节点 x 是节点 y 的祖先。例如，节点 4 是节点 16 的祖先。节点 10 也是节点 16 的祖先。事实上，节点 8、4、10 和 16 是节点 16 的祖先。请记住，节点是自身的祖先。节点 8、4、6 和 7 是节点 7 的祖先。如果节点 x 是节点 y 的祖先和节点 z 的祖先，则节点 x 称为两个不同节点 y 和 z 的共同祖先。因此，节点 8 和 4 是节点 16 和 7 的共同祖先。如果 x 是 y 和 z 的共同祖先，并且最接近其共同祖先中的 y 和 z，则节点 x 称为节点 y 和 z 的最近共同祖先。因此，节点 16 和 7 最近的共同祖先是节点 4。节点 4 比节点 16 更接近节点 7 和 8。

对于其他示例，节点 2 和 3 的最近公共祖先是节点 10，节点 6 和 13 的最近公共祖先是节点 8，节点 4 和 12 的最近公共祖先是节点 4。在最后一个示例中，如果 y 是 z 的祖先，则 y 和 z 最近的共同祖先是 y。 编写一个程序来查找树中两个不同节点的最近共同祖先。


 

输入

输入由 T 个测试用例组成。测试用例 （T） 的数量在输入文件的第一行给出。每个测试用例都以一行开头，其中包含整数 N，树中的节点数，2<=N<=10，000。节点用整数 1、2,..., N 标记。接下来的 N -1 行中的每一行都包含一对表示边的整数 - 第一个整数是第二个整数的父节点。请注意，具有 N 个节点的树正好有 N - 1 条边。每个测试用例的最后一行包含两个不同的整数，要计算其最近的共同祖先。

输出

为每个测试用例精确打印一行。该行应包含最接近的共同祖先的整数。

示例输入

<span style="color:#000000">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
</span>

示例输出

<span style="color:#000000">4
3
</span>

1. 首先，创建一个并查集数据结构，其中每个节点的初始根节点为自身。
2. 根据输入的边信息，将每个节点的父节点设置为指定的值，即将节点 l 的父节点设置为 r。
3. 使用 rmq 函数查找节点 r 和节点 l 的最近公共祖先。
   • 如果 r 和 l 相等，则返回 r。
   • 如果 r 和 l 不相等，则将 r 和 l 进行路径压缩操作，即将它们的父节点设置为根节点，然后再次查找它们的父节点。
   • 当 r 和 l 的父节点相等时，说明找到了最近公共祖先，返回该节点。

这种方法的时间复杂度为 O(n)，其中 n 是树中的节点数。在大多数情况下，这种方法的效率与基于深度优先搜索的方法相当。但是，在某些情况下，例如树的深度很大且查询次数较多时，基于深度优先搜索的方法可能更有效。

#include<iostream>
#include<cstring>
using namespace std;
const int maxn = 10002;
int n,r,l;
int f[maxn],vis[maxn];
void create_st(){
	for(int i=1;i<=n;i++){
		f[i]=i;
	}
}
int rmq(int r,int l){
	if(r==l)
	return r;
	vis[r]=1;
	while(f[r]!=r){
		r=f[r];
		vis[r]=1;
	}
	if(vis[l]==1)return l;
	while(f[l]!=l){
		l=f[l];
		if(vis[l])return l;
	}
} 
int main()
{
	int k;
	scanf("%d",&k);
	while(k--){
		memset(vis,0,sizeof(vis));
		scanf("%d",&n);
		create_st();
		for(int i=1;i<n;i++)
		{
			scanf("%d%d",&r,&l);
			f[l]=r;
		}
		scanf("%d%d",&r,&l);
		printf("%d\n",rmq(r,l));
	}
	return 0;
}