戴好口罩——并查集

问题描述

新型冠状病毒肺炎（Corona Virus Disease 2019，COVID-19），简称“新冠肺炎”，是指2019新型冠状病毒感染导致的肺炎。
如果一个感染者走入一个群体，那么这个群体需要被隔离！
小A同学被确诊为新冠感染，并且没有戴口罩！！！！！！
危！！！
时间紧迫！！！！
需要尽快找到所有和小A同学直接或者间接接触过的同学，将他们隔离，防止更大范围的扩散。
众所周知，学生的交际可能是分小团体的，一位学生可能同时参与多个小团体内。
请你编写程序解决！戴口罩！！
【输入】
多组数据，对于每组测试数据：
第一行为两个整数n和m（n = m = 0表示输入结束，不需要处理），n是学生的数量，m是学生群体的数量。0 < n <= 3e4 ， 0 <= m <= 5e2
学生编号为0~n-1
小A编号为0
随后，m行，每行有一个整数num即小团体人员数量。随后有num个整数代表这个小团体的学生。
【输出】
输出要隔离的人数，每组数据的答案输出占一行
【样例输入】
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
【样例输出】
4
1
1

关于并查集

并查集是一种用来管理元素分组情况的数据结构
1、查询元素A和元素B是否属于同一组
2、合并元素A和元素B所在的组
3、将所有元素的父亲初始化为自己本身
实现：
合并：把一个分组的根挂到另一个分组的根
查询：不断查询一个节点的祖先，此时进行路径压缩，为提高查询的复杂度，丢弃节点间的关系，把自己直接挂在根节点下。

问题分析

要找出与小A同学直接或间接接触的人群，在存储团队信息时，利用并查集将有关系的团队成员挂在同一集合下。则查询与小A同一个父亲的同学将其隔离

#include<iostream>
#include<cstdio>
using namespace std;

int n,m;
int par[50010];

int find(int x)
{
	if(par[x]==x)//父亲 
		return x;
	return par[x]=find(par[x]);
}

void add(int x,int y)
{
	int px=find(x);
	int py=find(y);
	par[px]=py;
}

int main()
{
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		if(n==0&&m==0) break;
		for(int i=0;i<n;i++)
			par[i]=i;
		for(int i=1;i<=m;i++)
		{
			int num,stu1,stu2;
			cin>>num;
			cin>>stu1;
			stu1=find(stu1);
			for(int j=1;j<num;j++)
			{
				cin>>stu2;
				stu2=find(stu2);
				if(stu1!=stu2)
				add(stu1,stu2);//疑惑点：合并 
			}
		}
		int fa=find(0);
		int cont=1;
		for(int j=1;j<n;j++)
		{
			if(find(j)==fa) cont++;
		}
		cout<<cont<<endl;
	}
}