主要思路:
本题显然是并查集结构的考察
需要注意的是在并查集的find()函数中,我们及时去更新我们查询时经过点的祖先,这样才能加快后续的查询(这一步也称路径压缩)
注意:本题要求输出1号元素同组元素的个数,可以简单通过将每个元素查询一遍得到结果;或者提前增加一个域,记录每组中元素个数的数组size[maxn],在find和union函数中注意修改size即可
原版本
int par[maxn];
int find(int targ){
if(targ==par[targ])
return targ;
return find(par[targ]);
}
改进版本
int par[maxn];
int find(int targ){
if(targ==par[targ])
return targ;
return par[targ]=find(par[targ]);
}
A - 氪金带东
新型冠状病毒肺炎(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 Possible Solution
#include<stdio.h>
const int maxn=3e4+10;
int stu[maxn],n,m;
void init(){
for(int i=0;i<n;i++)
stu[i]=i;
}
int find(int tar){
if(stu[tar]==tar)
return tar;
return stu[tar]=find(stu[tar]);
}
void Union(int x,int y){
int _x=find(x),_y=find(y);
if(_x!=_y)
stu[_x]=_y;
}
void disp(){
for(int i=0;i<n;i++)
printf("[%d]=%d ",i,stu[i]);
printf("\n");
}
int main(){
scanf("%d %d",&n,&m);
while(!(n==0 && m==0)){
int cnt,x,y;
init();
for(int j=0;j<m;j++){
scanf("%d %d",&cnt,&x);
for(int i=1;i<cnt;i++){
scanf("%d",&y);
Union(x,y);
x=y;
}
//disp();
}
int ans=0,virus=find(0);
for(int i=0;i<n;i++){
if(find(i)==virus)
ans++;
}
printf("%d\n",ans);
//disp();
scanf("%d %d",&n,&m);
}
return 0;
}