程序设计思维 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

思路

一、并查集
并查集由数组par和数组rank构成，其中par[i] = k表示第i个点属于以第k个点为根的集合、rank[k]表示以第k个点为根的集合的大小。初始时，par[i] = i、rank[i] = 1，如图：

假设有两个集合分别以V和W为根，如图有两个集合分别以第1个点和第5个点为根：

若要将这两个集合合并，则需要进行如下两步操作：
① par[W] = V
② rank[V] += rank[W]
合并后的结果如图：

查询时，更新该点的par：par[i] = find(par[i])，例如，查询第6个点后的结果如图：

二、解题

这题用并查集就能解决。
将0号所在的所有集合合并成一个大集合，再输出这个大集合的大小，即为需要隔离的人数。

代码

#include <iostream>
#define MAX 100000

using namespace std;

int par[MAX];
int Rank[MAX];

int n, m;

void init()
{
	for (int i = 0; i < n; i++) {
		par[i] = i;
		Rank[i] = 1;
	}
}

int find(int x)
{
	return par[x] == x ? x : par[x] = find(par[x]);
}

void unite(int x, int y)
{
	int xPar = find(x);
	int yPar = find(y);

	if (xPar == yPar) return;

	if (Rank[xPar] > Rank[yPar]) {
		par[yPar] = xPar;
		Rank[xPar] = (Rank[yPar] += Rank[xPar]);
	}
	else {
		par[xPar] = yPar;
		Rank[yPar] = (Rank[xPar] += Rank[yPar]);
	}
}

int main()
{
	while (cin >> n >> m) {
		if (n == 0 && m == 0) break;
		init();
		for (int i = 0; i < m; i++) {
			int num, first;
			cin >> num;
			cin >> first;
			for (int j = 0; j < num - 1; j++) {
				int other;
				cin >> other;
				unite(first, other);
			}
		}
		cout << Rank[find(0)] << endl;
	}

	return 0;
}