洛谷 P1551 亲戚 并查集

P1551 亲戚

题目背景

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

题目描述

规定：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个询问的答案为“具有”或“不具有”亲戚关系。

输入输出样例

输入

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>
#define MAXN 10005
using namespace std;

int pre[MAXN];
int n, m, p;


int _find(int x){
	if(pre[x] == x) return x;
	else return pre[x] = _find(pre[x]);
} 

void _link(int x, int y){
	pre[_find(y)] = _find(x); //使得a的祖先成为b的祖先的父节点 
} 

int main(){
	scanf("%d%d%d", &n, &m, &p);
	
	for(int i = 1; i <= n; i++)
		pre[i] = i; //初始化每一个子集合 
		
	for(int i = 0; i < m; i++){
		
		int a, b;
		scanf("%d%d", &a, &b);
		_link(a, b); //连接a,b两个节点 
	}
	
	for(int i = 0; i < p; i++){
		
		int a, b;
		scanf("%d%d", &a, &b);
		
		if (_find(a) == _find(b)) 
			printf("Yes\n");
		else 
			printf("No\n");
	}
	
	return 0;		
}