poj_1611 The Suspects

题意

有n个学生，m个组，每个组有k个学生，一开始学生0感染了病毒，如果一个组里有一个学生感染了病毒那么这个组里的人都会被传染上病毒，有多组数据，n和m为0时表示输入结束。

思路

这道题是一个比较简单的并查集题目，我们每次读入一个组学生的编号时，就把2~k学生的编号和编号1学生的编号连在一起，表示他们是同一个组的，我们在操作“集”的时候要判断一下，如果这两个要连起来的学生不是同一个组的，我们就让rank[第一个学生]+=rank[第二个学生]，rank[i]表示学生i所在的组里面有几个学生，最后输出rank[find(0)],表示和学生0在同一组的有几个学生。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int t,n,m,father[30001],s,fi,rank[30001];
int find(int x)
{
int t1=x,t2;
while(x!=father[x]) x=father[x];
while(father[t1]!=t1)//并查集的压缩路径
{
t2=father[t1];
father[t1]=x;
t1=t2;
}
return x;
}
void unio(int x,int y)
{
x=find(x);
y=find(y);
if (x==y) return;//判断是否在同一组
else//如果不在就合起来，并且统计组员个数
{
father[y]=x;
rank[x]+=rank[y];
}
}
int main()
{
while(scanf("%d%d",&n,&m),n||m)
{
for (int i=0;i<=n;i++)
father[i]=i,rank[i]=1;//初始化
for (int i=1;i<=m;i++)
{
scanf("%d%d",&t,&fi);
for (int j=1;j<t;j++)//直接让后面的同学都和
{                //第一个同学连起来
scanf("%d",&s);
unio(fi,s);
}
}
printf("%d\n",rank[find(0)]);//这里不输出
}                    //rank[0]是因为0的祖先可能会变
}