洛谷亲戚C++

亲戚

题目背景

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

题目描述

规定：$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

思路

将所有亲戚全都指向一个共同的祖先，最后构造一个find函数寻找 是否为亲戚的2人的祖先。若返回的祖先为同一个，即为亲戚关系。

实现find函数之前 需要将各个人的祖先设置为自己

for(int i=0;i<=n;i++)
   	{
   		fa[i]=i;	
	}

find函数

int find(int x)
{
	if(x==fa[x])
	{
		return x;
	}else{
		find(fa[x]);
	}
}

亲戚关系的绑定

while(m--)
	{
		cin>>a>>b;
		u = find(a);
		v = find(b);
		if(u!=v)
		{
			fa[u]=v;
		} 
	}

是否为亲戚的判定

while(p--)
	{
		cin>>a>>b;
		u = find(a);
		v = find(b);
		if(u!=v)
		{
			cout<<"No"<<endl;
		}else{
			cout<<"Yes"<<endl;
		}
	}

AC代码

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e4;
int fa[maxn];
int find(int x)
{
	if(x==fa[x])
	{
		return x;
	}else{
		find(fa[x]);
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
   	int n,m,p;
   	cin>>n>>m>>p;
   	for(int i=0;i<=n;i++)
   	{
   		fa[i]=i;	
	}
	int a,b,u,v;
	while(m--)
	{
		cin>>a>>b;
		u = find(a);
		v = find(b);
		if(u!=v)
		{
			fa[u]=v;
		} 
	}
	while(p--)
	{
		cin>>a>>b;
		u = find(a);
		v = find(b);
		if(u!=v)
		{
			cout<<"No"<<endl;
		}else{
			cout<<"Yes"<<endl;
		}
	}
    return 0;
}