1249. 亲戚 - AcWing题库

1249. 亲戚 - AcWing题库

或许你并不知道，你的某个朋友是你的亲戚。

他可能是你的曾祖父的外公的女婿的外甥女的表姐的孙子。

如果能得到完整的家谱，判断两个人是否是亲戚应该是可行的，但如果两个人的最近公共祖先与他们相隔好几代，使得家谱十分庞大，那么检验亲戚关系实非人力所能及。

在这种情况下，最好的帮手就是计算机。

为了将问题简化，你将得到一些亲戚关系的信息，如Marry和Tom是亲戚，Tom和Ben是亲戚，等等。

从这些信息中，你可以推出Marry和Ben是亲戚。

请写一个程序，对于我们的关于亲戚关系的提问，以最快的速度给出答案。

输入格式

输入由两部分组成。

第一部分以 N,M 开始。N 为问题涉及的人的个数。这些人的编号为 1,2,3,…,N。下面有 M行，每行有两个数 ai,bi，表示已知 ai 和 bi是亲戚。

第二部分以 Q 开始。以下 Q行有 Q 个询问，每行为 ci,di，表示询问 ci 和 di 是否为亲戚。

输出格式

对于每个询问 ci,di，输出一行：若 ci 和 di 为亲戚，则输出“Yes”，否则输出“No”。

数据范围

1≤N≤20000
1≤M≤106
1≤Q≤106

输入样例：

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

思路：

需要并查集知识
1. 数据预处理
. 初始化每个人的祖宗是自己
2. 合并祖宗集合
3. 输出
. 祖宗相同输出yes 不同输出no

并查集：

并查集是指 不同集合的合并和查找
以本题为例
一个集合里面是一个族谱 他们都有共同的祖宗 那么这个族谱就可以看作是集合 祖先就作为这个族谱的代表 祖先的祖先是自己
合并操作就是将 两个不同祖先的族谱合并成一个祖先的族谱
查找操作就是 查找某个人的祖宗

集合：

#define N 20005
int p[x];								//族谱 记录每个人的祖宗是谁
for(int i = 1; i < N; i++) p[x] = i; 		//初始化每个人的祖宗是自己	

查找：

int find(int x){
	while(p[x] != x) x = find(p[x]);			//如果这个人不是祖先 继续找他的祖先 p[x]是这个人x的祖先
	return x;									//返回祖先
}

合并：

//假如有两个族谱a, b 合并两个族谱的方法就是 将两个祖宗的其中一个作为另一个的祖宗
p[find(a)] = find(b);

Code:

#include <bits/stdc++.h>
using namespace std;

#define N 20005

int n, m, q, p[N];

int find(int x){
    while(p[x] != x) x = find(p[x]);                      //找到x的祖宗 如果是祖宗 那么他的祖宗是自己 不是则继续找
    return x;                                        //返回祖宗
}

int main(){
    for(int i = 1; i < N; i++) p[i] = i;                //初始化每个人的祖宗是自己
    cin >> n >> m;
    
    while(m --){
        int a, b;
        scanf("%d%d", &a, &b);
        p[find(a)] = find(b);                          //合并操作 将a的祖宗的祖宗合并进b的祖宗集合里 这样a，b同属一个祖宗
    }
    
    cin >> q;
    while(q -- ){
        int a, b;
        scanf("%d%d", &a, &b);
        if(find(a) == find(b)) printf("Yes\n");         //如果两个人是同一个祖宗 输出yes
        else printf("No\n");
    }
    
    return 0;
}