【C++ 真题】P1551 亲戚

P1551 亲戚

题目背景

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

题目描述

规定：$x$ 和 $y$ 是亲戚，$y$ 和 $z$ 是亲戚，那么 $x$ 和 $z$ 也是亲戚。如果 $x$，$y$ 是亲戚，那么 $x$ 的亲戚都是 $y$ 的亲戚，$y$ 的亲戚也都是 $x$ 的亲戚。

输入格式

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

以下 $m$ 行：每行两个数 $M_i$，$M_j$，$1\le M_i,M_j\le n$，表示 $M_i$ 和 $M_j$ 具有亲戚关系。

接下来 $p$ 行：每行两个数 $P_i,P_j$，询问 $P_i$ 和 $P_j$ 是否具有亲戚关系。

输出格式

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

输入输出样例 #1

输入 #1

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

输出 #1

Yes
Yes
No

题解

#include "bits/stdc++.h"
using namespace std;
const int N = 1e5+7;
int n, m, a, b, p[N] = {0};
int find(int x){//找它的根节点
	if(p[x] == x) return p[x];
	p[x] = find(p[x]);//继续找根节点
	return p[x];
}
void add(int x, int y){
	int fx = find(x);
	int fy = find(y);
	if(fx != fy){//两个根节点不一样，任取一个作为另一个父节点
		p[fy] = fx;
	}
	return ;
}
int main() {
    int q, c, d;
	cin>>n>>m>>q;
	for(int i=1;i<=n;++i) p[i] = i;//初始默认自己是自己的父节点
	for(int i=1;i<=m;++i){
		scanf("%d %d", &a, &b);
		add(a, b);
	}
	
	while(q--){//问题
		scanf("%d %d", &c, &d);
		if(find(c) == find(d)) printf("%s\n", "Yes");//两个根是一样的
		else printf("%s\n", "No");
	}
	return 0;
}