父子局对决!——并查集例题“亲戚”

题目转载自“信息学奥赛一本通”!!!

【例4-7】亲戚(relation)


【题目描述】

或许你并不知道,你的某个朋友是你的亲戚。他可能是你的曾祖父的外公的女婿的外甥女的表姐的孙子。如果能得到完整的家谱,判断两个人是否是亲戚应该是可行的,但如果两个人的最近公共祖先与他们相隔好几代,使得家谱十分庞大,那么检验亲戚关系实非人力所能及。在这种情况下,最好的帮手就是计算机。为了将问题简化,你将得到一些亲戚关系的信息,如Marry和Tom是亲戚,Tom和Ben是亲戚,等等。从这些信息中,你可以推出Marry和Ben是亲戚。请写一个程序,对于我们的关于亲戚关系的提问,以最快的速度给出答案。

【输入】

输入由两部分组成。

第一部分以N,M开始。N为问题涉及的人的个数(1≤N≤20000)。这些人的编号为1,2,3,…, N。下面有M行(1≤M≤1000000),每行有两个数ai,biai,bi,表示已知aiaibibi是亲戚。

第二部分以Q开始。以下Q行有Q个询问(1≤ Q ≤1000000),每行为ci,dici,di,表示询问cicididi是否为亲戚。

【输出】

对于每个询问ci,dici,di,输出一行:若cicididi为亲戚,则输出“Yes”,否则输出“No”。

int find(int x)
{
	if(x==f[x])
		return x;
	return f[x]=find(f[x]);
}
注意是  return f[x]=find(f[x])  ;  压缩!(哈撒给)

父子局对决!——并查集例题“亲戚”_第1张图片


但是我第一次写的时候出现了个小问题。。。。

如图

for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		int j=find(x);
		int k=find(y);
		if(j!=k)
			family(x,y);
	}
不能是 family(x,y) 。不过问题不大,直接看图 (灵魂画室已上线!):

父子局对决!——并查集例题“亲戚”_第2张图片

 若连接 LDB 和 WRY,查找LC的father时只会搜到TPY,而实际情况应该如下图所示:父子局对决!——并查集例题“亲戚”_第3张图片

所以应该family(j,k)!!!

还是给出AC代码:

#include
 using namespace std;
 int f[20005];
 int find(int x)
 {
 	if(x==f[x])
		return x;
 	return f[x]=find(f[x]);
 }
 void family(int x,int y)
 {
	f[y]=x;
 }
 int main()
 {
	int n,m,x,y,t;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		f[i]=i;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		int j=find(x);
		int k=find(y);
		if(j!=k)
			family(j,k);
	}
	scanf("%d",&t);
	for(int i=1;i<=t;i++)
	{
		scanf("%d%d",&x,&y);
		if(find(x)==find(y))
			printf("Yes\n");
		else
			printf("No\n");
	}
	return 0;
 }

观众朋友们,欢迎吐槽。






你可能感兴趣的:(数据结构,图论,并查集)