题目
新型冠状病毒肺炎(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
思路:
利用并查集,将每个分组合并成一个等价类,而且将不同分组中一旦有共有的元素就合并成一个集合,每一个集合用一个根作为代表元素,每次查找都返回该集合的根节点。
并查集的初始化:
void init(int n)
{
for(int i=0;i<n;i++)
{
par[i]=i;//每个集合初始只有自己,自己的父节点是自己
rak[i]=1;//记录i的孩子的数目
}
}
并查集的查找
int find(int x)
{
if(par[x]==x)//满足这样条件的就是根节点
return x;
else
return par[x]=find(par[x]);//路径压缩,把每个点直接挂在根上,这样方便以后查找
}
并查集的合并
void unite(int x,int y)
{
x=find(x);
y=find(y);
if(x==y)
return ;
if(rak[x]>rak[y])
swap(x,y);//目的是将小的挂在大的上
par[x]=y;
rak[y]+=rak[x];
}
代码:
#include<iostream>
using namespace std;
int par[30005];
int rak[30005];
void init(int n)
{
for(int i=0;i<n;i++)
{
par[i]=i;
rak[i]=1;
}
}
int find(int x)
{
if(par[x]==x)
return x;
else
return par[x]=find(par[x]);
}
void unite(int x,int y)
{
x=find(x);
y=find(y);
if(x==y)
return ;
if(rak[x]>rak[y])
swap(x,y);
par[x]=y;
rak[y]+=rak[x];
}
int main()
{
int n,m;
while(true)
{
cin>>n>>m;
if(m==0&&n==0)
break;
init(n);
for(int i=0;i<m;i++)
{
int num;
cin>>num;
int pre;
cin>>pre;
for(int j=1;j<num;j++)
{
int id;
cin>>id;
unite(pre,id);
pre=id;
}
}
cout<<rak[find(0)]<<endl;
}
}