并查集的简单应用

题目描述

新型冠状病毒肺炎（Corona Virus Disease 2019，COVID-19），简称“新冠肺炎”，是指2019新型冠状病毒感染导致的肺炎。
如果一个感染者走入一个群体，那么这个群体需要被隔离！
小A同学被确诊为新冠感染，并且没有戴口罩！！！！！！
危！！！
时间紧迫！！！！
需要尽快找到所有和小A同学直接或者间接接触过的同学，将他们隔离，防止更大范围的扩散。
众所周知，学生的交际可能是分小团体的，一位学生可能同时参与多个小团体内。
请你编写程序解决！戴口罩！！

Input

多组数据，对于每组测试数据：
第一行为两个整数n和m（n = m = 0表示输入结束，不需要处理），n是学生的数量，m是学生群体的数量。0 < n <= 3e4 ， 0 <= m <= 5e2
学生编号为0~n-1
小A编号为0
随后，m行，每行有一个整数num即小团体人员数量。随后有num个整数代表这个小团体的学生。

Output

输出要隔离的人数，每组数据的答案输出占一行

Sample Input

100 4
2 1 2
5 10 13 11 12 14
2 0 1
2 99 2
200 2
1 5
5 1 2 3 4 5
1 0
0 0

Sample Output

4
1
1

解题思路

本题题目特别简单，小白我也一看就知道是一道并查集的题目，我觉得题目更多想考察的应该是代码的适当优化吧。虽然并查集维护的是一棵树，但也却不是必须要保持树的结构，因为数的作用仅仅是为了找到根节点。所以可以在此处进行适当的优化，直接将每一个点都指向根节点即可。
还有就是，在进行合并操作时，为了减少find的次数，将“小树”挂到“大树”上。这样也可以降低复杂度。

代码

#include<iostream>
#include<cstring>
using namespace std;
const int maxn=30000;//最大的学生数量
 
int par[maxn],rnk[maxn];
void init(int n) { for(int i=0;i<n;i++) par[i]=i, rnk[i]=1; }//初始化操作 
int find(int x) { return par[x]==x?x:par[x]=find(par[x]); }//合并操作，同时进行优化 
bool unite(int x,int y)//合并操作 
{
	x=find(x),y=find(y);
	if(x==y) return false;//如果已经属于一个集合则不用合并 
	
	if(rnk[x]>rnk[y]) swap(x,y);//小的挂到大的上，减少后面的查找操作 
	par[x]=y;
	rnk[x]+=rnk[y];//集合的大小 
	rnk[y]=rnk[x];
}
int main()
{
	int n,m,num,x,y;
	while(cin>>n>>m,n!=0||m!=0)
	{
		init(n);
		for(int i=0;i<m;i++)
		{
			cin>>num;
			if(num!=0) cin>>x;
			for(int j=1;j<num;j++)
			{
				cin>>y;
				unite(x,y);
			}
		}
		cout<<rnk[find(0)]<<endl;//输出对应集合的大小 
	}
	return 0;
	
}