算法-程序设计课week6-作业-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

思路

这是个并查集的问题，不算难。

1. 获取学生总数和组数
2. 对每个组的学生都进行一次并集运算
3. 最后看看与0号同组的学生有多少，就知道多少人需要隔离了。

心得

并查集写起来太简洁了，功能又很强大，牛！

int find(int x) {
	if (father[x] != x)father[x] = find(father[x]);
	return father[x];
}

inline void a_union_b(int a, int b) {
	father[find(a)] = find(b);
}

#include <iostream>
using namespace std;
int father[50010];
int n;
int find(int x) {
	//如果当前非根节点，则将本节点的父节点设置为最终根节点
	if (father[x] != x)
		father[x] = find(father[x]);
	return father[x];  //返回根节点
}

//union是C++关键词，不能用
//将a，b的组合并，本次合并所需的路径刷新在下次find过程进行
inline void a_union_b(int a, int b) {
	father[find(a)] = find(b);
}

inline void initunion() {
	for (int i = 0; i < n; i++) father[i] = i;
}

int main() {
	int group_num, ans;

	int a, b;
	while (~scanf("%d %d", &n, &group_num) && n || group_num) {
		ans = 0;
		initunion();
		for (int i = 1; i <= group_num; i++) {
			int current_num;  //每组个体数目
			scanf("%d %d", &current_num, &a);
			a = find(a);

			//从第二个人开始进行遍历
			for (int j = 2; j <= current_num; j++) {
				scanf("%d", &b);
				b = find(b);
				if (a != b)
					a_union_b(a, b);
			}
		}

		//0看看有多人跟0号病人有过直接或间接接触
		int flag = find(0);
		for (int i = 0; i < n; i++)
			if (find(i) == flag)
				ans++;
		printf("%d\n", ans);
	}
	return 0;
}