B - 戴好口罩！（Week6作业）

题目

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

题目思路

思想：并查集 路径压缩
实现：
·对于每个输入的学生集合进行合并，并利用rnk[i]记录集合的总人数
·最后查找有0号学生的集合的根，输出根记录的人数
细节：find()函数一定要路径压缩，否则很可能会超时

代码实现

#include<cstdio>
#include<cstring>
using namespace std;
int par[30005],rnk[100009],n,m,ans,sign,cnt;
int find(int x)
{
	if(par[x]==x)
		return x;
	else
		return 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]) 
	{
		par[y]=x;
		rnk[x]=(rnk[y]+=rnk[x]);
	}
	else
	{
		par[x]=y;
		rnk[y]=(rnk[x]+=rnk[y]);
	}
	return true;
}
int main()
{
	//freopen("a.txt","r",stdin);
	while(scanf("%d",&n)&&scanf("%d",&m)&&(m!=0||n!=0))
	{
		ans=0;
		for(int i=0;i<=n;i++)
			par[i]=i,rnk[i]=1;
		for(int i=0;i<m;i++)
		{
			scanf("%d",&cnt);
			int temp;
			for(int j=0;j<cnt;j++)
			{
				int z;
				scanf("%d",&z);
				if(j==0) temp=z;
				else{
					unite(temp,z);
				}	
			}
		}
		int a=find(0);
		ans=rnk[a];
		printf("%d\n",ans);
	}
	return 0;
}