第六周 B - 戴好口罩！

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

解题思路

使用并查集，以感染源为一个源集合，每一个小团体里有源集合成员则将该小团体并入源集合，最后输出源集合内成员个数。复杂度为O(n).

代码

#include<iostream>
using namespace std;

long long int n,m,ans;
long long int par[100010];

long long int find(long long int x){
    return par[x]=(par[x]==x? x:find(par[x]));
}

int main(){
    while(~scanf("%lld%lld",&n,&m)&&!(n==0&&m==0)){
        ans=1;
        for(long long int i=0;i<n;i++) par[i]=i;
		for(long long int i=1;i<=m;i++) {
            long long int cnt,first;
            scanf("%lld%lld",&cnt,&first);
			for(long long int i=1;i<=cnt-1;i++) {
                long long int temp;
                scanf("%lld",&temp);
                if(par[temp]!=temp) {
                	par[find(temp)]=find(first);
				}
                else par[temp]=find(first);
            }
        }
        for(long long int i=1;i<n;i++)
            if(find(i)==find(0)) ans++;
        printf("%lld\n",ans);
    }
    return 0;
}