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个整数代表这个小团体的学生。

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

Output：

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

sample output：

	4
	1
	1

---

个人思路：

只是用了一个并查集，就可以完美解决这个问题。题目中的一个个小团体就可以作为一个个集合，并且集合以第一个元素作为代表元。

在输入数据的时候，先将代表元sign输入，然后依次输入之后的num-1个元素，并且unite（sign，x）。

输入结束后，将元素0所在集合的rank输出，就是最后的答案。

---

代码块：

#include<iostream>
using namespace std;
int n, m;
const int maxn = 30100;
int par[maxn], rnk[maxn];
void init() {
	for (int i = 0; i < n; ++i) {
		par[i] = i;
		rnk[i] = 1;
	}
}
int find(int x) {
	return par[x] == x ? x : par[x] = find(par[x]);
}
void unite(int x, int y) {
	x = find(x), y = find(y);
	if (x == y)return;

	if (rnk[x] > rnk[y])par[y] = x, rnk[x] = (rnk[y] += rnk[x]);
	else par[x] = y, rnk[y] = (rnk[x] += rnk[y]);
	
	return;
}
int main() {
	int num, sign, st;
	while (~scanf_s("%d%d", &n, &m)) {
		if (m == 0 && n == 0)break;
		init();
		for (int i = 0; i < m; ++i) {
			scanf_s("%d", &num);
			scanf_s("%d", &sign);
			for (int j = 0; j < num - 1; ++j) {
				scanf_s("%d", &st);
				unite(sign, st);
			}
		}
		printf( "%d\n",rnk[find(0)]);
	}
	return 0;
}

---