A1107 Social Clusters (30) 并查集（本题有一个易错点）

最新推荐文章于 2023-08-07 00:20:23 发布

原创最新推荐文章于 2023-08-07 00:20:23 发布 · 207 阅读

0 ·

CC 4.0 BY-SA版权

pat刷题专栏收录该内容

32 篇文章

订阅专栏

本文深入探讨了并查集算法的实现细节，包括如何通过findfather函数查找根节点，以及如何使用unio函数合并不同集合。文章通过具体的代码示例展示了如何在一系列课程注册过程中应用并查集算法来统计每门课程的根节点数量，最后按数量从大到小排序输出。

合并完成之后，也有可能出现当前节点的父节点没有及时变化的情况，所以isroot里面的是findfather，而不是father数组的形式

#include<cstdio>
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn=1010;
int course[maxn]={0};
int father[maxn];
int isroot[maxn]={0};
vector<int> ans;
bool cmp(int a,int b)
{
	return a>b;
}
int findfather(int x)
{
	int a=x;
	while(x!=father[x])
	{
		x=father[x];
	}
	int z;
	while(a!=father[a])
	{
		z=a;
		a=father[a];
		father[z]=x;
	}
	return x;
}
void unio(int a,int b)
{
	int faa=findfather(a);
	int fab=findfather(b);
	if(faa!=fab)
	{
	   father[faa]=fab;
	}
}
int main()
{
	int n,k,num;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		father[i]=i;	
	}
	for(int i=1;i<=n;i++)
	{
		scanf("%d:",&k);
		for(int j=0;j<k;j++)
		{
			scanf("%d",&num);
			if(course[num]==0)
			{
				course[num]=i;
			}
			else
			{
				unio(course[num],i);
			}
		}
	}
	for(int i=1;i<=n;i++)
	{
		isroot[findfather(i)]++;
	}
	for(int i=1;i<=n;i++)
	{
		if(isroot[i]!=0)
		ans.push_back(isroot[i]);
	}
	sort(ans.begin(),ans.end(),cmp);
	printf("%d\n",ans.size());
	for(int i=0;i<ans.size();i++)
	{
		printf("%d",ans[i]);
		if(i!=ans.size()-1)
	    printf(" ");
	}
}