题目
新型冠状病毒肺炎(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;
}