[洛谷]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个询问的答案为“具有”或“不具有”亲戚关系。

输入输出样例

输入样例#1

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

输出样例#1

Yes
Yes
No

---

思路

并查集不错的模板题。给出n个人m个关系p个询问，回答Pi和Pj是否具有亲戚关系。
把初始时候每个人看做一个无边的点，下来根据给出的亲戚关系，将各个点进行连接，而所有连接起来的点组成的图可以视为一个家族（森林）。

很容易想到并查集，因为：

在一些有N个元素的集合应用问题中，我们通常是在开始时让每个元素构成一个单元素的集合，然后按一定顺序将属于同一组的元素所在的集合合并，其间要反复查找一个元素在哪个集合中。

将两个有亲戚关系的人连起来就ok了。

 

#include <stdio.h>
#include <iostream>
using namespace std;
int f[5001],n,m,p,s;
int getfind(int v)//不断认爹的递归函数，实际上就是在一个二叉树中找到祖先 
{
	if(f[v]==v) return f[v];//如果找到祖宗，即v没有爸爸了，返回的v就是我们所求的祖先节点 
	else//v不是自己的爸爸，所以它上面还有爸爸 ，我们的目标是找祖先节点 
	{//路径压缩，这样可以提高寻找树的祖先的速度 
		f[v]=getfind(f[v]);//问爸爸的爸爸是谁，递归就好了 
		return f[v];
	}
}
inline int merge(int v,int u)//合并子集合 
{
	register int t1=getfind(v),t2=getfind(u);//t1、r2分别是v和u的祖先 
	if(t1!=t2)//判断2个点是否为同一个祖先 
	{
		f[t2]=t1;
	}//这里是靠左原则，即把右边的集合作为左边集合的子集合。
	//经过路径压缩后，将f[u]的根的值也赋值为v的祖先f[t1]
}
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	register int i,j;
	cin>>n>>m>>p;
	for(i=1;i<=n;i++)
	{
		f[i]=i;//设初始情况下每个人和其他人没有亲戚关系，所以存储自己 
	}
	for(i=1;i<=m;i++)
	{
		int x,y;
		cin>>x>>y;
		merge(x,y);
	}
	for(i=1;i<=p;i++)
	{
		int x,y;
		cin>>x>>y;
		if(getfind(x)==getfind(y))//如果x所在子集的祖先和b所在的祖先一样，则x和y在同一个集合 
		{
			cout<<"Yes"<<endl;
		}
		else
		{
			cout<<"No"<<endl;
		}
	}
	return 0;
}