【题目描述】
或许你并不知道,你的某个朋友是你的亲戚。他可能是你的曾祖父的外公的女婿的外甥女的表姐的孙子。如果能得到完整的家谱,判断两个人是否是亲戚应该是可行的,但如果两个人的最近公共祖先与他们相隔好几代,使得家谱十分庞大,那么检验亲戚关系实非人力所能及。在这种情况下,最好的帮手就是计算机。为了将问题简化,你将得到一些亲戚关系的信息,如Marry和Tom是亲戚,Tom和Ben是亲戚,等等。从这些信息中,你可以推出Marry和Ben是亲戚。请写一个程序,对于我们的关于亲戚关系的提问,以最快的速度给出答案。
【输入】
输入由两部分组成。
第一部分以N,M开始。N为问题涉及的人的个数(1≤N≤20000)。这些人的编号为1,2,3,…, N。下面有M行(1≤M≤1000000),每行有两个数ai,bi,表示已知ai和bi是亲戚。
第二部分以Q开始。以下Q行有Q个询问(1≤ Q ≤1000000),每行为ci,di,表示询问c和di是否为亲戚。
【输出】
对于每个询问ci,di,输出一行:若ci和di为亲戚,则输出“Yes”,否则输出“No”。
【输入样例】
10 7
2 4
5 7
1 3
8 9
1 2
5 6
2 3
3
3 4
7 10
8 9
【输出样例】
Yes
No
Yes
解题思路:
使用并查集(Disjoint Set)数据结构。并查集是一种用于处理一些不交集的合并及查询问题的数据结构,非常适合处理这种亲戚关系的问题。
程序中,首先使用init
函数初始化并查集,将每个人的父节点初始化为自己。然后,使用unite
函数将已知的亲戚关系合并到同一个集合中。最后,使用same
函数判断两个人是否在同一个集合中,即是否为亲戚。如果是,则输出“Yes”,否则输出“No”。
#include <stdio.h>
#define MAXN 20005 // 定义最大人数常量
int parent[MAXN]; // 存储每个人的父节点(代表元)
// 初始化并查集,将每个人的父节点设为自己
void init(int n) {
for (int i = 1; i <= n; i++) {
parent[i] = i;
}
}
// 查找x所在的集合的代表元(即根节点),并进行路径压缩
int find(int x) {
if (parent[x] != x) {
// 路径压缩:直接将x的父节点设置为根节点,减少树的高度
parent[x] = find(parent[x]);
}
return parent[x]; // 返回x所在集合的代表元
}
// 合并x和y所在的两个集合,将x所在集合的代表元合并到y所在集合的代表元上
void unite(int x, int y) {
int px = find(x); // 查找x所在集合的代表元
int py = find(y); // 查找y所在集合的代表元
if (px != py) {
// 如果x和y不在同一个集合中,则合并两个集合
parent[px] = py; // 将x所在集合的代表元合并到y所在集合的代表元上
}
}
int main() {
int N, M, Q, a, b;
// 读入人数N和已知亲戚关系对数M
scanf("%d %d", &N, &M);
// 初始化并查集
init(N);
// 读入已知的亲戚关系对,并进行合并操作
for (int i = 0; i < M; i++) {
scanf("%d %d", &a, &b); // 读入一对亲戚关系
unite(a, b); // 合并a和b所在的集合
}
// 读入询问个数Q
scanf("%d", &Q);
// 对每个询问判断两个人是否为亲戚
for (int i = 0; i < Q; i++) {
scanf("%d %d", &a, &b); // 读入一对询问
// 如果a和b的根节点相同,说明他们在同一个集合中,即互为亲戚
if (find(a) == find(b)) {
printf("Yes\n"); // 输出"Yes"
} else {
printf("No\n"); // 否则输出"No"
}
}
return 0; // 程序结束
}
代码步骤:
1.初始化了并查集
2.读入并合并了所有已知的亲戚关系对。对于每个查询,我们通过查找两个人所在集合的代表元来判断他们是否属于同一个集合,即是否互为亲戚。通过路径压缩技术,我们进一步优化了并查集的查找效率。
3.代码以C语言风格的输入输出进行了实现,以适应处理大量数据的需求。