戴好口罩!
一、题目
新型冠状病毒肺炎(Corona Virus Disease 2019,COVID-19),简称“新冠肺炎”,是指2019新型冠状病毒感染导致的肺炎。
如果一个感染者走入一个群体,那么这个群体需要被隔离!
小A同学被确诊为新冠感染,并且没有戴口罩!!!!!!
危!!!
时间紧迫!!!!
需要尽快找到所有和小A同学直接或者间接接触过的同学,将他们隔离,防止更大范围的扩散。
众所周知,学生的交际可能是分小团体的,一位学生可能同时参与多个小团体内。
请你编写程序解决!戴口罩!!
二、输入
多组数据,对于每组测试数据:
第一行为两个整数n和m(n = m = 0表示输入结束,不需要处理),n是学生的数量,m是学生群体的数量。0 < n <= 3e4 , 0 <= m <= 5e2
学生编号为0~n-1
小A编号为0
随后,m行,每行有一个整数num即小团体人员数量。随后有num个整数代表这个小团体的学生。
三、输出
输出要隔离的人数,每组数据的答案输出占一行
四、样例输入输出
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
Output
4
1
1
五、解题思路
并查集思想。
定义a数组存放元素,b数组存放数目。将a数组的每个元素看成一个集合,每个集合有一个根。当读入小团体时,对a数组进行处理,将这个小团体中的元素合并为一个集合,两个元素合并的具体操作为:分别查找两个元素对应的集合的根,若相同,则属于同一个集合,不需要合并;若不同,则让一个集合的根指向另一个集合的根,即可完成合并。
注意:可以使用路径压缩的方式来简化查询的过程。在一个集合中,虽然通过查找最终都能查找到集合的根,但这些节点并不一定都直接指向根,所以经过的查找次数较多。可以在查找时将集合中所有节点均指向根来达到路径压缩的目的。
六、代码样例
#include<iostream>
using namespace std;
int a[30005],b[30005];//a存放元素,b存放数目
int find(int x){
if(a[x] == x)
return x;
a[x] = find(a[x]);
return a[x];
}
bool unite(int x,int y)
{
x=find(x);
y=find(y);
if(x==y) return false;
if(b[x]<b[y])
{
b[y]+=b[x];
a[x]=y;
}
else
{
b[x]+=b[y];
a[y]=x;
}
return true;
}
int main()
{
while(1)
{
for(int i=0;i<30005;i++) a[i]=b[i]=-1; //初始化
int n,m; cin>>n>>m;
if(m==0&&n==0) return 0;
for(int i=0;i<n;i++)
{
a[i]=i;//节点初始化
b[i]=1;
}
int mi,x,y;
for(int i=0;i<m;i++)
{
cin>>mi;
cin>>x;
for(int j=1;j<mi;j++)
{
cin>>y;
unite(x,y);
}
}
int ans=find(0);
cout<<b[ans]<<endl;
}
}