POJ 1611 The Suspects 并查集

表示刚接触并查集做这道题对并查集有些感觉了，刚开始的时候想的一种方式有bug，最终调试了半个小时终于知道为什么错了 T_T

题目大意：

对于n个学生编号为0至n - 1， 其中编号为0的学生感染了病毒， 接下来给出m组人，每行开头表示该组有多少人，接下来是该组的组员编号，与0在同一组或有同组的人与0在一组的人视为感染，求感染的总人数。

很明显的一个并查集问题，就是分组的问题，问0所在的集合的元素数量。

大致思路：

对于每次给的一个人，都与该组第一个人在同一集合，合并他们所在的集合。

完成所有合并后，扫描一遍0所在集合的元素数量（与0所在集合相同，则所在集合代表元素相同）

代码如下：

Result : Accepted   Memory : 388 KB   Time : 16 ms

/*
 * Author : Gatevin
 * Time : 2014.4.10 17:20
 * Problem : POJ 1611 The Suspects
 * Tag : Disjoint Set
 */

#include<cstdio>
int f[30010];
int n,m;

int find(int i)//压缩路径
{
    if(i != f[i])
    {
        f[i] = find(f[i]);
    }
    return f[i];
}

void Union(int x, int y)//合并集合
{
    int rx = find(x);
    int ry = find(y);
    if(rx != ry)
    {
        f[rx] = ry;
    }
    return;
}

int main()
{
    while(scanf("%d %d",&n,&m) == 2)
    {
        if(!(n || m)) return 0;
        for(int i = 0; i <= n - 1; i++)//初始化各个集合
            f[i] = i;
        while(m--)
        {
            int t,first,tmp;
            scanf("%d",&t);
            scanf("%d",&first);//每组的第一个人
            for(int i = 1; i <= t - 1; i++)
            {
                scanf("%d",&tmp);
                Union(first,tmp);//连接合并集合
            }
        }
        int tmp = find(0);
        int cnt = 1;
        for(int i = 1; i <= n - 1; i++)
        {
            if(find(i) == tmp) cnt++;//统计与0在同一个集合的元素个数
        }
        printf("%d\n",cnt);
    }
    return 0;
}