题目背景
若某个家族人员过于庞大,要判断两个是否是亲戚,确实还很不容易,
现在给出某个亲戚关系图,求任意给出的两个人是否具有亲戚关系。
题目描述
规定:x 和 y 是亲戚,y 和 z 是亲戚,那么 x 和 z 也是亲戚。
如果 x, y 是亲戚,那么 x 的亲戚都是 y 的亲戚,y 的亲戚也都是 x 的亲戚。
输入格式
第一行:三个整数 n, m, p,分别表示有 n 个人,m 个亲戚关系,询问 p 对亲戚关系。
以下 m 行:每行两个数 Mi,Mj,表示 Mi 和 Mj 具有亲戚关系。
接下来 p 行:每行两个数 Pi,Pj,询问 Pi 和 Pj 是否具有亲戚关系。
输出格式
P 行,每行一个 ’Yes’ 或 ’No’,表示第i个询问的答案为“具有”或“不具有”亲戚关系。
输入样例
6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6
输出样例
Yes
Yes
No
数据范围
1 ≤ n, m, p ≤ 5000
1 ≤ Mi,Mj ≤ N
题解
并查集:
#include <iostream>
using namespace std;
const int N = 5010;
int f[N];
int find(int x)
{
if(f[x] != x) f[x] = find(f[x]);
return f[x];
}
int main()
{
int n, m, p, a, b;
cin >> n >> m >> p;
for (int i = 1; i <= n; i ++) f[i] = i;
while(m --)
{
cin >> a >> b;
f[find(b)] = find(a); // a 是 b 的父节点
}
while(p --)
{
cin >> a >> b;
if(find(a) == find(b)) cout << "Yes" << endl;
else cout << "No" << endl;
}
return 0;
}