week 6 作业 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
输出要隔离的人数，每组数据的答案输出占一行。

样例

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
Output
4
1
1

思路分析

首先，参与同一团体的学生们的接触状况肯定是一样的，若有同学同时参与两个团体，那么这两个团体都该属于同一集合。

所以这里用到了并查集的思想：

数组student存放节点的“父母”，刚开始都是孤立节点，“父母”都是自身，为根节点。当两个顶点根相同时，在同一个集合中。

find(int x)：当student[x]!=x说明当前顶点处在一个集合中，继续找其父母的父母。直到相等时，找到整个集合的根。

Union(int root1, int root2)：合并两个集合。找到两个集合根后，将较小者作为较大者的父母。

void init()
{
 count=0;
 for(int i=0;i<n;i++)
 {
  student[i]=i;//当开始的根都是自己 
 }
 } 
int find(int x) 
{//函数搜索并返回包含元素x的树的根元素 
 while(student[x]!=x)
 {
  x = student[x];
 }
 return x;
}
void Union(int root1, int root2) 
{
 if (student[root1]<student[root2]) //把小的作为根 
 {    
  student[root2]=root1;    
 }
 else
 {
  student[root1]=root2;           
 } 
}

同一团体中，若前后两个人的根不一样，则把两个集合并到根较小的那个集合中。

最后计算根为0的人数，即是需要被隔离的人。

AC代码

#include<iostream>
using namespace std;
int student[30005];
int n,m,count;
void init()
{
 count=0;
 for(int i=0;i<n;i++)
 {
  student[i]=i;//当开始的根都是自己 
 }
 } 
int find(int x) 
{//函数搜索并返回包含元素x的树的根元素 
 while(student[x]!=x)
 {
  x = student[x];
 }
 return x;
}
void Union(int root1, int root2) 
{
 if (student[root1]<student[root2]) //把小的作为根 
 {    
  student[root2]=root1;    
 }
 else
 {
  student[root1]=root2;           
 } 
}
int main()
{
 while(cin>>n>>m)
 {
  if(n==0&&m==0) return 0;
  init();
  for(int j=0;j<m;j++)
  {
   int num;
      cin>>num;//团体人数数量
      int s[num];
      for(int i=0;i<num;i++)
      {
        cin>>s[i];
       } 
      for(int i=1;i<num;i++)
      {
       int root1=find(s[i-1]);
       int root2=find(s[i]);
       Union(root1,root2);
   } 
  }
  for(int i=0;i<n;i++)
  {
   if(find(i)==0) 
   {
    //cout<<"这个 "<<i<<endl;
    count++;
   } 
  }
  cout<<count<<endl;  
  } 
 return 0;
 }