题目背景
若某个家族人员过于庞大,要判断两个是否是亲戚,确实还很不容易,现在给出某个亲戚关系图,求任意给出的两个人是否具有亲戚关系。
题目描述
规定:x和y是亲戚,y和z是亲戚,那么x和z也是亲戚。如果x,y是亲戚,那么x的亲戚都是y的亲戚,y的亲戚也都是x的亲戚。
输入输出格式
输入格式:
第一行:三个整数n,m,p,(n<=5000,m<=5000,p<=5000),分别表示有n个人,m个亲戚关系,询问p对亲戚关系。
以下m行:每行两个数Mi,Mj,1<=Mi,Mj<=N,表示Mi和Mj具有亲戚关系。
接下来p行:每行两个数Pi,Pj,询问Pi和Pj是否具有亲戚关系。
输出格式:
P行,每行一个’Yes’或’No’。表示第i个询问的答案为“具有”或“不具有”亲戚关系。
思路:并查集
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 5;
int fa[maxn];
int find(int x) {
return x == fa[x] ? x : fa[x] = find(fa[x]);
}
int main() {
int n, m, q, u, v;
cin >> n >> m >> q;
for (int i = 1; i <= n; i++) { //初始化每个集合
fa[i] = i;
}
for (int i = 1; i <= m; i++) {
cin >> u >> v;
fa[find(u)] = find(v); //合并集合
}
for(int i = 1; i <= q; i++) {
cin >> u >> v;
if(find(u) == find(v)) {//判断是否是同一个集合
cout << "Yes" << endl;
} else {
cout << "No" << endl;
}
}
return 0;
}