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;}