简单并查集

最新推荐文章于 2024-05-31 21:15:25 发布

烟蓑雨笠

最新推荐文章于 2024-05-31 21:15:25 发布

阅读量937

点赞数

CC 4.0 BY-SA版权

分类专栏： Algorithm 文章标签： C++ 并查集

本文链接：https://blog.youkuaiyun.com/cstopcoder/article/details/21552029

Algorithm 专栏收录该内容

34 篇文章

订阅专栏

本文介绍了一种使用并查集算法解决朋友圈分组问题的方法。通过实现并查集的基本操作如查找与合并，解决了给定若干好友关系下朋友圈数量的问题。文章提供了完整的AC代码，并附带了样例输入输出。

简单并查集

题目描述：

假如已知有n个人和m对好友关系（存于数字r）。如果两个人是直接或间接的好友（好友的好友的好友...），则认为他们属于同一个朋友圈，请写程序求出这n个人里一共有多少个朋友圈。
假如：n = 5 ， m = 3 ， r = {{1 , 2} , {2 , 3} , {4 , 5}}，表示有5个人，1和2是好友，2和3是好友，4和5是好友，则1、2、3属于一个朋友圈，4、5属于另一个朋友圈，结果为2个朋友圈。

输入：

输入包含多个测试用例，每个测试用例的第一行包含两个正整数 n、m，1=<n，m<=100000。接下来有m行，每行分别输入两个人的编号f，t（1=<f，t<=n），表示f和t是好友。当n为0时，输入结束，该用例不被处理。

输出：

对应每个测试用例，输出在这n个人里一共有多少个朋友圈。

样例输入：

样例输出：

2
1

AC代码：

#include<cstdio>
#define MAX 100005

using namespace std;

int pre[MAX];

int find(int x)
{
	int p=x;
	while(pre[p]!=p)
		p=pre[p];
	int i,j;
	i=x;
	while(i!=p)//compress path algorithm
	{
		j=pre[i];
		pre[i]=p;
		i=j;
	}

	return p;
}

void Union(int x,int y)
{
	int fx=find(x);
	int fy=find(y);
	if(fx==fy)
		return;
	pre[fx]=fy;
}

int main(int argc,char *argv[])
{
	int n,m;
	while(scanf("%d",&n)&&n)
	{
		int x,y;
		int ans=0;
		scanf("%d",&m);
		for(int i=1;i<=n;i++)
			pre[i]=i;
		for(int j=0;j<m;j++)
		{
			scanf("%d %d",&x,&y);
			Union(x,y);
		}
		for(int i=1;i<=n;i++)
		{
			if(pre[i]==i)
				ans++;
		}
		printf("%d\n",ans);
	}

	return 0;
}