戴好口罩(并查集） c++

题目
新型冠状病毒肺炎（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

解题思路
1.很明显的思路是把有0的组都认定为感染，然后感染的人又有组队的，又感染。

2.那么可以使用并查集来实现这一点.

代码实现

#include<iostream>
using namespace std;
const int Max = 30000;
int n,m;
int par[Max];  //记录学生i的老大 
int r[Max];    //i 所在的组有多少人 

void ini()     //初始化 
{
	for(int i=0;i<Max;i++)
		{par[i] = i;r[i] = 1;}    //每个人一开始老大就是他自己 
}

int find(int a)   //找a 的老大
{
	if( par[a] == a )
		return a;
	else 
		par[a] = find( par[a] );
	return par[a];
 } 
 
bool unite(int x,int y)  //合并
{
	x = find(x) , y = find(y);
	if( x == y )  return 0;    //同一组 
	if( r[x]> r[y] )  par[y] = x , r[x] += r[y] ; 
	else              par[x] = y , r[y] += r[x] ;
	//维护这组的个数
	return 1;
}

int main()
{
	
	int num;
	int a1,a2;
	ios::sync_with_stdio(0);
	while( 1 )
	{
		cin>>n >>m;
		if(n==0&&m==0) return 0;
		ini();
		for( int I =0;I<m;I++ )
		{
			cin>>num;  
			cin>>a1;
			for( int i=1;i<num;i++ )
			{
				cin>>a2;
				unite( a1,a2 );
			}
		}
		//找到 0 所在的组
		//返回该组的人数
		int r0 = find(0);
		cout<<r[r0]<<endl;
	}
	return 0;
}

小结
每次使用find查找老大的时候，我们都将所有该组的par指向老大，而不是一次次的向上查找，压缩路径可以降低查找的时间。
在unite函数里，我们还利用了每组的人数来判断到底谁加入谁，这样可以保证形成的树结构是比较平衡的。