程序设计思维与实践 Week6 作业 B-戴好口罩

题目链接：B-戴好口罩

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

思路：
本题用到了并查集的知识，需要查询和0号病人（小A同学）在同一个集合中的人数，即统计每个节点的子节点数。如果两个节点的父亲节点相同，则不需要合并，用sum统计0号节点对应的数目，最后输出sum

总结：
并查集在上学期数据结构实验中用到过，但是是为了统计节点子节点的数目，刚好适用于本题。并查集不是严格的树形结构，同时通过设置代表元素，从而查询某个团体集合中所具有的元素个数。初始将每一个元素看作一个单独的团体，然后通过递归查询，直到找到该元素所在团体的代表元素。

代码：

#include<iostream>
#include<string.h>
#include<stdio.h>
using namespace std;
int n,m,num,sum,father[30010],student[30010];
int find(int x){
	return x==father[x]?x:father[x]=find(father[x]);
}
void unite(int a,int b){
	int x=find(a);
	int y=find(b);
	if(x!=y){
		father[x]=y;
	}
}
int main(){
	while(scanf("%d%d",&n,&m)){
		if(n==0&&m==0)
			break;
		for(int i=0;i<n;i++)
		    father[i]=i;
		while(m--){
			scanf("%d",&num);
			for(int i=0;i<num;i++)
				scanf("%d",&student[i]);
			for(int i=1;i<num;i++)
				unite(student[i-1],student[i]);
		}
		sum=1;
		for(int i=1;i<n;i++)
			if(find(0)==find(i))
				sum++;
		printf("%d\n",sum);
	}
	return 0;
}