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

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

思路分析

该题由于是一个群体共进退的问题，即群体中一人感染，全部隔离，因此用并查集来解决。

定义find函数，用于寻找根元素即代表元，和unite函数，用于合并具有相同特征的集合。在输入每个团体中成员编号的过程中合并成员，最后，find(0)就是找到编号为0的同学所在的集合的代表元，rankl[find(0)]就是需要被隔离的人数啦。

注意

用并查集查找根元素时要进行路径压缩，因为我们只需要找到同一个集合的代表元素即可，不必要维护树形结构。否则查找的复杂度可能会降为O(n)。

代码

#include<iostream>
#include<stdio.h>
using namespace std;
int n,m,num,index;
const int maxn=3e4+5;
//rank用于记录树中元素个数，用于合并时树大小的比较，以及输出隔离总人数
int par[maxn],rank[maxn];

int find(int x)
{
 if(par[x]==x) return x;
 return par[x]=find(par[x]);
}
bool unite(int x,int y)
{//合并
 x=find(x);
 y=find(y);
 if(x==y) return false;//合并失败，两个集合是同一个
 if(rank[x]>rank[y]) swap(x,y);//小树挂大树
 par[x]=y,rank[y]=(rank[x]+=rank[y]);
 return true;
}

int main()
{
 while(cin>>n>>m)
 {
  if(n==0&&m==0) break;
  for(int i=0;i<n;i++) 
  {//数组初始化
   rank[i]=1;
   par[i]=i;
  }
  for(int i=0;i<m;i++)
  {
   cin>>num;
   int flag=-1;//
   for(int j=0;j<num;j++)
   {
    cin>>index;
    //合并一个团体
    if(flag!=-1) unite(index,flag);
    flag=index;
   }
  }
  int ans=rank[find(0)];//妙哉
  cout<<ans<<endl;
 }
 return 0;
}