戴好口罩（并查集）

问题描述

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

问题分析

该题需要利用一种数据结构——并查集。
 并查集是一种用来管理元素分组情况的数据结构。并查集可以高效地进行如下操作：
 查询元素A和元素B是否属于同一组
 合并元素A和元素B所在的组
注意：需要进行路径压缩，否则会超时。
参考代码如下：

#include<stdio.h>
int fa[30010];//记录所属集合
int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}//查找属于哪个集合
void unite(int x,int y){//合并
	x=find(x);y=find(y);
	fa[x]=y;
}
int main(){
	int n,m,num,ans,x,y,A;
	scanf("%d%d",&n,&m);
	while(n!=0 || m!=0){
		ans=0;
		for(int i=0;i<n;i++)fa[i]=i;
		for(int i=0;i<m;i++){
			scanf("%d",&num);
			if(num!=0)scanf("%d",&x);
			for(int j=1;j<num;j++){
				scanf("%d",&y);
				unite(x,y);
			}	
		}
		A=find(0);
		for(int i=0;i<n;i++)
			if(A==find(i))ans++;
		printf("%d\n",ans);
		scanf("%d%d",&n,&m);
	}
	return 0;
}