week-6-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:
输出要隔离的人数，每组数据的答案输出占一行

样例：
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
OUTPUT:
4
1
1

思路

• 分析此题，我们可以使用并查集进行求解，首先就是并查集的初始化，将所有人都变成一个集合，然后每一小组的人进行并查集的合并；
• 如果一个并查集中出现了0 ，而且该集合中的元素如果也出现在其他集合中，那么在进行并查集的合并的时候，就已经更新了，合并了；
• 最后去统计和0号元素一样的值的元素个数即可；
• 此处使用的数组描述的并查集，能够在合并时将所有的值进行更新；

总结

此题使用数组描述的并查集的时候，能够在每次进行合并的时候，将所有属于此并查集的所有元素更新为同一个值；
此处应用的数组并查集在数据范围过大的时候，合并更新的时候复杂度会变得较高，可能会超时；

代码

#include<iostream>
#include<vector>
using namespace std;
int N;
int a[30000];
void init()//并查集的初始化
{
	for (int i = 0; i < N; i++)
		a[i] = i;
}
int find(int x)//并查集的查询
{
	return a[x];
}
void unite(int x, int y)
{
	y = find(y);
	x = find(x);
	for (int i = 0; i < N; i++)
	{
		if (a[i] == y)
			a[i] = x;
	}

}
int main()
{

	while (cin >> N)
	{
		if (N == 0)
			break;
		//并查集的初始化
		init();
		int m;
		cin >> m;
		int sum = 0;
		for (int i = 0; i < m; i++)
		{
			int sum;
			cin >> sum;
			int* p = new int[sum];
			for (int i = 0; i < sum; i++)
			{
				cin >> p[i];
			}
			if (sum >= 2)
			{//进行并查集的合并
				for (int i = 0; i < sum; i++)
					if (i < sum - 1)
						unite(p[i], p[i + 1]);
			}
		}
		if (m == 0) {
			cout << 1 << endl;
		}
		else {
			//去寻找感染人数
			for (int i = 0; i < N; i++)
			{
				if (a[0] == a[i])
					sum++;
			}
			cout << sum << endl;
		}
	}
	return 0;
}