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

思路

本题使用并查集进行解题，首先将并查集数组进行初始化(每个人都是一个小集合)，即a[i]=i，然后对于每个小群体进行更新并查集，如果群体中的人不在一个并查集中，进行合并，在一个并查集中则检查下一个。更新完成后，即可使用for循环对于每个人都检查是否和0号同学在一个并查集中，若在则需要隔离的人数加一。

代码

#include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;
int a[30002];
void ini(int n){for(int i=0;i<n;i++) a[i]=i;}
int find(int x)
{
    if(x==a[x]) return x;
    else return find(a[x]);
}
void unite(int x,int y)
{ 
    int temp=find(y);
    a[temp]=find(x);
}
int main()
{
    int n,m;
    while(~scanf("%d %d",&n,&m)&&(m!=0||n!=0))
    {
        ini(n);
        for(int k=0;k<m;k++)
        {
            int num;
            scanf("%d",&num);
            int last=-1;
            while(num--)
            {
                int x;
                scanf("%d",&x);
                if(x!=a[x]&&last==-1)
                    last=find(x);
                else if(x==a[x]&&last==-1)
                    last=x;
                else if(x!=a[x]&&last!=-1)
                    unite(last,x);
                else
                unite(last,x);
            }
        }
        int temp=find(0),count=0;
        for(int i=0;i<n;i++)
        {
            if(find(i)==temp)
                count++;
        }
        cout<<count<<endl;
    }
    return 0;
}