POJ 1611 The Suspects 并查集

大学里总共有n个学生，m个社团。学生的编号从0到(n-1)。每个学生可以参加多个社团。现在0号学生疑似感染了SARS。与疑似感染者同一个社团的人都是疑似感染者。求疑似感染者总数。

这是一个简单的并查集题目。初始时，每个学生都各是一个集合。然后把同一社团的学生都合并到同一集合去。完成合并后，0号学生所在的集合的元素个数即是疑似感染者总数。

#include <stdio.h>
#define MAX 30005
//并查集
//还有一种设置并查集的办法：将father[]都初始化为-1，然后利用祖宗的father[]记录负的集合元素数。
int father[MAX];

int findset(int x)
{
int fx = x,tmp;
while(fx != father[fx])//找到x的最老的祖宗
fx = father[fx];
while(x != father[x])//路径压缩：把x的各级father都置为最老的祖宗
{
tmp = father[x];
father[x] = fx;
x = tmp;
}
return fx;
}

void unionset(int x,int y)
{
x = findset(x);
y = findset(y);
if(x < y)//让编号最小的成为最老的祖宗，这是为了让0号学生成为代表元素
father[y] = x;
else
father[x] = y;
return;
}

int main()
{
int m,n,k,i,j,suspects,tmp1,tmp2;
//FILE *fp = NULL;
// if(NULL == (fp = fopen("t.txt","r")) )
// printf("cannot open file.\n");
while(1)
{
suspects = 0;
//fscanf(fp,"%d %d",&n,&m);
scanf("%d %d",&n,&m);
if(0 == m && 0 == n)
break;
if(0 == m)
{
suspects = 1;
printf("%d\n",suspects);
continue;
}
for(i=0;i<n;i++)//make set
father[i] = i;
for(i=0;i<m;i++)
{
//fscanf(fp,"%d",&k);
scanf("%d",&k);
//fscanf(fp,"%d",&tmp1);
scanf("%d",&tmp1);
for(j=1;j<k;j++)
{
//fscanf(fp,"%d",&group[j]);
scanf("%d",&tmp2);
unionset(tmp1,tmp2);
}
}
for(i=0;i<n;i++)
//if(0 == father[i])
if(0 == findset(i))
suspects++;
printf("%d\n",suspects);
}

return 0;
}