P1551 亲戚（并查集模板题）

 展开

题目背景

若某个家族人员过于庞大，要判断两个是否是亲戚，确实还很不容易，现在给出某个亲戚关系图，求任意给出的两个人是否具有亲戚关系。

题目描述

规定：xx 和 yy 是亲戚，yy 和 zz 是亲戚，那么 xx 和 zz 也是亲戚。如果 xx，yy 是亲戚，那么 xx 的亲戚都是 yy 的亲戚，yy 的亲戚也都是 xx 的亲戚。

输入格式

第一行：三个整数 n,m,pn,m,p，（n,m,p \le 5000n,m,p≤5000），分别表示有 nn 个人，mm 个亲戚关系，询问 pp 对亲戚关系。

以下 mm 行：每行两个数 M_iMi​，M_jMj​，1 \le M_i,~M_j\le N1≤Mi​, Mj​≤N，表示 M_iMi​ 和 M_jMj​ 具有亲戚关系。

接下来 pp 行：每行两个数 P_i,P_jPi​,Pj​，询问 P_iPi​ 和 P_jPj​ 是否具有亲戚关系。

输出格式

pp 行，每行一个 Yes 或 No。表示第 ii 个询问的答案为“具有”或“不具有”亲戚关系。

输入输出样例

输入 

6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6

输出

Yes
Yes
No

代码咯

#include <bits/stdc++.h>
using namespace std;
int fa[5000];

void init (int n) {//初始化
	for (int i = 1; i <= n; i++ )
		fa[i] = i;
}

int find(int x) {//找祖宗
	if (x == fa[x])
		return x;
	else {//路径压缩
		fa[x] = find(fa[x]);
		return fa[x];
	}
}

void merge(int i, int j) {//让i的祖宗认j的祖宗为祖宗
	fa[find(i)] = find(j);
}

int main() {
	int n, m, p;
	int x, y;
	cin >> n >> m >> p;
	init(n);
	while (m--) {
		cin >> x >> y;
		merge(x, y);
	}
	while (p--) {
		cin >> x >> y;
		if (find(x) == find(y))
			cout << "Yes\n";
		else
			cout << "No\n";
	}
	return 0;
}