week6 B-带好口罩 并查集

题目：
新型冠状病毒肺炎（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
本题很明显是用并查集解决。
因为一个人可能属于多个团队，所以在依据输入的值将每个人归类时，要分成两种情况，一种是在此之前不属于任何团队，一种是在此之前已经是一个团队的了，则把两个团队合并。
以下是完整代码：

#include<iostream>
using namespace std;
int n,m;
int pre[400000];
int find(int x)//查找 
{
 if(pre[x]==x)
 {
  return x;
 }
 return find(pre[x]);
}
void unite(int x,int y)//合并 
{
 x=find(x);
 y=find(y);
 if(x!=y)
 {
  pre[y]=x;
 }
}
int main()
{
 while(1)
 {
  cin>>n>>m;
  if(n==0&&m==0)//0 0表示结束 
  {
   break;
  }
  for(int i=0;i<=n;i++)//现将pre数组进行初始化 
  {
   pre[i]=i;
  }
  int p;
  for(int i=0;i<m;i++)
  {
   cin>>p;
   int ele;//ele是该团队的第一个输入的人，将此人最为该团队的标志 
   cin>>ele;
   int x;
   for(int j=1;j<p;j++)
   {
    cin>>x;
    if(find(x)!=find(ele)&&find(x)!=x)//如果x和ele不同，说明x之前已经属于一个团队了，那么将两个团队合并 
    {
     unite(find(x),find(ele));
    }
    else if(find(x)==x)
    {
     unite(x,ele);
    }
   }
  }
  int temp=find(0);
  int ans=1;
  for(int i=1;i<=n;i++)
  {
   if(find(i)==temp)//与0号同一团队，则说明需要隔离
   {
    ans++;
   }
  }
  cout<<ans<<'\n';
 }
}