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

输出要隔离的人数，每组数据的答案输出占一行

Smple 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

我的思路：
这道题是要求和0接触过的人，这里可以使用并查集的方法，来计算两个数是不是在一棵树上。先对于输入的点初始化，利用并查集的规则不断合并，然后找到所有和0在一棵树上的数（包括0），计算个数，就是最后的结果。

我的总结：
在判断两个点是不是在同一棵树上的时候，并查集是一个很好的方法。

我的代码：

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

int par[30000],rnk[30000];

int m,n,num,a[30000],ans;



void init(){for(int i=0;i<n;i++) par[i]=i,rnk[i]=1;}
int find(int x) {return par[x]==x?x:par[x]=find(par[x]);}
void unit(int x,int y)
{
    x=find(x),y=find(y);
    if(x==y) return;
    if(rnk[x]>rnk[y]) swap(x,y);
    par[x]=y,rnk[y]=(rnk[x]+=rnk[y]);
}

int main()
{
    while(1)
    {
        scanf("%d%d",&n,&m);
        if(!m&&!n) break;
        init();
        ans=1;
        for (int i = 0;i < m;i++)
        {
            scanf("%d",&num);
            for (int k=0;k<num;k++) scanf("%d",&a[k]);
            for(int k=1;k<num;k++) unit(a[k-1],a[k]);
             
        }
        int tar=find(0);
        for(int k=1;k<n;k++) if(find(k)==tar) ans++;
        printf("%d\n",ans);
    }
    return 0;
}