题目描述:
新型冠状病毒肺炎(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个整数代表这个小团体的学生。
输出:
输出要隔离的人数,每组数据的答案输出占一行
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
output:
4
1
1
题目分析:
本题是一个等价类问题,说白了就是要找几个数组里面所有共有的都合并到一个集合里面,那么就需要一个问题,一个集合要用什么来标识自己呢?这就需要一种方法——孩子双亲表示法,就是用一个数组,数组下标是当前节点,数组内容就是结点的双亲,那么初始化都是独立的,所以每个节点的双亲就是自己
for(int i=0;i<n;i++)
{
par[i]=i;
ran[i]=1;
}
然后既然能确定每个集合,如何找到集合的元素呢?这就用到dfs遍历了,很明显,找到了就返回,找不到就找双亲,直到找到为止
int f(int x)
{
if(par[x]==x)
return x;
else
return par[x]=f(par[x]);
}
对于数组的合并,实际上并不需要动所有的数组,只需要对两个根节点进行合并即可,首先对于两个元素得找双亲,然后双亲就可以结合。不过最方便的合并是小的合并到大的里面,那么就需要判断谁大,小的双亲设定为大的就可
bool unite(int x,int y)
{
x=f(x);
y=f(y);
if(x==y)
return false;
else
{
par[x]=y;
ran[y]=ran[y]+ran[x];
return true;
}
}
好了,这样我们只需要每输入一个数就和上一个数合并,这样我们查找的时候只需要找到0号对应的集合就好了,每次合并的时候我们有一个数组记录代表元有几个元素
int p;
cin>>p;
if(last!=-1)
unite(p,last);
last=p
代码如下:
#include<iostream>
using namespace std;
#define maxn 30005
int par[maxn],ran[maxn];//记录父母和代表元有几个数
void ini(int n)//初始化
{
for(int i=0;i<n;i++)
{
par[i]=i;
ran[i]=1;
}
}
int f(int x)//查找
{
if(par[x]==x)
return x;
else
return par[x]=f(par[x]);
}
bool unite(int x,int y)//合并
{
x=f(x);
y=f(y);
if(x==y)
return false;
else
{
par[x]=y;
ran[y]=ran[y]+ran[x];
return true;
}
}
int main()
{
int n,m;
while(1)
{
cin>>n>>m;
if(m==0&&n==0)
break;
ini(n);
for(int i=0;i<m;i++)
{
int num;
int last=-1;//初始为-1,这样如果没有合并的元素就重新建立一个数组
cin>>num;
for(int j=0;j<num;j++)
{
int p;
cin>>p;
if(last!=-1)
unite(p,last);
last=p;
}
}
cout<<ran[f(0)]<<endl;
}
}