【戴好口罩】并查集

题目

题意
新型冠状病毒肺炎（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

题目大意

本题给出了若干个学生集合，每个学生集合给出其中的人数与学生编号，要求找到所有与 0 号同学有交集的学生人数。

解题思路

本题是一道典型的并查集问题。首先将每位同学的所在并查集代表元值初始化为自己的编号，在输入过程中将每组的第一位同学的编号作为代表元。之后遍历同组的其他同学，对于某位同学，如果他还未被归到某个并查集内，就将它放到该组代表元同学所在的并查集；如果该同学已经属于某个并查集，就将他所在的并查集和该组代表元同学的并查集合并，并查集的总体操作思路就是这样。
在所有组操作结束后只要找到和 0 号同学在一组内的人数即可。此时遍历所有同学，通过并查集中的 find() 函数可以找到所在组的代表元。

具体代码

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<math.h>
#include<queue>
#include<string.h>
#define MAXN 1000005
using namespace std;

int n, m, ans;
int fa[MAXN];
int find(int x)
{
	if(fa[x] == x)
	{
		return x;
	}
	else
	{
		return find(fa[x]);
	}
}

int main()
{
    while(cin >> n >> m)
	{
    	ans = 1;
    	if(n == 0 && m == 0)
    	{
    		break;
		}
    	for(int i = 0; i < n; i++)
		{
			fa[i] = i;
		}
    	for(int i = 0; i < m; i++)
		{
    		int tot, one;
    		cin >> tot >> one;
    		for(int i = 1; i < tot; i++)
			{
    			int num;
    			cin >> num;
    			if(fa[num] == num)
    			{
    				fa[num] = find(one);
				}
    			else
    			{
    				fa[find(num)] = find(one);
				}
    			one = num;
    		}
    	}
    	for(int i = 1; i < n; i++)
		{
    		if(find(0) == find(i))
    		{
    			ans++;
			}
    	}
    	cout << ans << endl;
    }
    
    return 0;
}```