【Week6 作业B】戴好口罩！

题目描述

新型冠状病毒肺炎（Corona Virus Disease 2019，COVID-19），简称“新冠肺炎”，是指2019新型冠状病毒感染导致的肺炎。

如果一个感染者走入一个群体，那么这个群体需要被隔离！

小A同学被确诊为新冠感染，并且没有戴口罩！！！！！！

危！！！

时间紧迫！！！！

需要尽快找到所有和小A同学直接或者间接接触过的同学，将他们隔离，防止更大范围的扩散。

众所周知，学生的交际可能是分小团体的，一位学生可能同时参与多个小团体内。

请你编写程序解决！戴口罩！！

输入说明

多组数据，对于每组测试数据：

第一行为两个整数n和m（n = m = 0表示输入结束，不需要处理），n是学生的数量，m是学生群体的数量。0 < n <= 3e4 ， 0 <= m <= 5e2

学生编号为0~n-1

扫描二维码关注公众号，回复： 10252254 查看本文章

小A编号为0

随后，m行，每行有一个整数num即小团体人员数量。随后有num个整数代表这个小团体的学生。

输出说明

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

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号学生所在集合的学生数目即可。
对于多组输入数据，需在每次处理数据之前，对并查集进行初始化处理。

代码

#include <iostream>
#include<string.h>
using namespace std; 
const int size=3e4+10;
int v[size];
int number[size];
void ini(int n)
{
	for(int i=0;i<n;i++)
		v[i]=i,number[i]=1;
}
int find(int n)
{
	return v[n]==n?n:v[n]=find(v[n]);
}
bool unite(int a,int b)
{
	int fa=find(a),fb=find(b);
	if(fa==fb)
	{
		return false;
	}
	else
	{
		if(number[fa]<number[fb])swap(fa,fb);
		number[fa]+=number[fb];
		v[fb]=fa;
		return true;
	}
}

int main(int argc, char** argv) {
	int m,n;
	while(~scanf("%d%d",&n,&m)&&!(m==0&&n==0))
	{
		ini(n); 
		int sum=0;
		for(int i=0;i<m;i++)
		{
			int num,temp=-1;
			scanf("%d",&num);
			for(int j=0;j<num;j++)
			{
				int a;scanf("%d",&a);
				if(temp!=-1)unite(temp,a);
				temp=a;
			}	
		}
		printf("%d\n",number[find(0)]);
	}
	
	return 0;
}