程序设计思维 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

思路

这是一道考察并查集的题目。
首先将每个学生初始化为一个独立的分组，即father[i]=i。接着将每一个团体内的学生所在分组合并，在合并的同时更新分组内元素个数。

代码

#include <cstdio>
#include <algorithm>
using namespace std;
int father[30005],rnk[30005];

int find(int x){
    if(father[x]==x)
        return x;
    return father[x]=find(father[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);
    father[x]=y;
    rnk[x]=(rnk[y]+=rnk[x]);//rnk[y]+=rnk[x];rnk[x]=rnk[y];
}

int main() {
    int n,m;
    while (scanf("%d%d",&n,&m)!=EOF&&!(n==0&&m==0)) {
        for(int i=0;i<n;i++)
            father[i]=i;
        for(int i=0;i<n;i++)//不可memset(rnk,1,sizeof(rnk)),赋为0x0101而不是1
            rnk[i]=1;
        for(int i=0;i<m;i++){
            int num;
            scanf("%d",&num);
            int stu1,stu2;
            scanf("%d",&stu1);
            for(int j=0;j<num-1;j++){
                scanf("%d",&stu2);
                unit(stu1,stu2);
                stu1=stu2;
            }
        }
        printf("%d\n",rnk[find(0)]);
    }
    return 0;
}

题目链接