基础的并查集,题目大意:一共有N个人,给你M组人,每组有K人,所有人都被从0到N-1唯一标号。如果组内有一个人得病则全组都被传感。现假设0号同学得病,问最后所有得病的人数。
基本思路:
将每组的人都进行合并操作,可以得到一些树(集合),0号同学所在的集合全部得病。其他集合则没有。
代码如下了:
#include<cstdio>
#include<string.h>
#define N 30007
using namespace std;
int father[N],rank[N];
int n,m,i,j,k;
void init()
{
for( i =0; i<=n; i++)
{
father[i]=i;
rank[i]=1;
}
}
int find(int x)
{
if(x!=father[x])//压缩
father[x]=find(father[x]);
return father[x];
}
void unon(int x,int y)
{
int fx=find(x);
int fy=find(y);
if(fx==fy) return ;
if(rank[fx]>=rank[fy])
{
father[fy]=fx;
rank[fx]+=rank[fy];
}
else
{
father[fx]=fy;
rank[fy] += rank[fx];
}
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF&&n!=0||m!=0)
{
init();
for(i=0; i<m; i++)
{
int temp1,temp2;
scanf("%d%d",&k,&temp1);
for(j=1; j<k; j++)
{
scanf("%d",&temp2);
unon(temp1,temp2);
}
}
printf("%d\n",rank[father[0]]);
}
return 0;
}
刚真正开始ACM之旅,加油自己!