并查集-The Suspects POJ - 1611

并查集-The Suspects POJ - 1611

题目：
2003年4月16日，世界卫生组织根据包括中国内地和香港地区，加拿大、美国在内的11个国家和地区的13个实验室通力合作研究的结果，宣布重症急性呼吸综合征的病因是一种新型的冠状病毒，称为SARS冠状病毒。
很不幸，小明同学昨天晚上被确诊以经感染非典病毒，为此，校医院需要对他隔离治疗并隔离观察与他直接或间接接触者。
可以认为，同一个社团内如果有人感染病毒或可能感染病毒，那么这个社团内所有人都被认为是可能已经感染了病毒，由于时间紧迫，需要尽快找到所有可能携带病毒的同学并隔离，以防止更大范围的病毒扩散，院长请你帮忙编写程序计算需要隔离多少人

Input
输入文件包含多组数据。
对于每组测试数据：
第一行为两个整数n和m
其中n是学生的数量，m是学生社团的数量。0 < n <= 30000 ， 0 <= m <= 500
每个学生编号是一个0到n-1之间的整数，已知小明同学编号是0
紧随其后的是团体的成员列表，每组一行。
每一行有一个整数k，代表成员数量。之后有k个整数代表这个群体的学生。一行中的所有整数由至少一个空格隔开。
n = m = 0表示输入结束，不需要处理。

Output
对于每组测试数据
输出需要隔离的人数，每组输出占一行

Sample Input
100 4
2 1 2
5 10 13 11 12 14
2 0 1
2 99 2
200 2
1 5
5 1 2 3 4 5
1 0
0 0

Sample Output
4
1
1

题意：
输入点的数量n，点集的数量m
输入m个点集中点的数量k，以及k个点的编号
输出与0号点所在点集的剩余点的数量

思路：
采用并查集。
每读入一个新点都将前一个点的祖先跟在新点的祖先下面。
最后求和0号点相同祖先的点的数量。
注意路径压缩。

代码：

#include
#include
#include
using namespace std;
const int maxn=3e4+5;
int n,m,pre[maxn];

int Find(int root)
{
    if(root==pre[root]) return root;
    return pre[root]=Find(pre[root]);//路径压缩的递归写法
}

int Find1(int root)//路径压缩的迭代写法
{
    int son,tmp;
    son=root;
    while(root!=pre[root])
        root=pre[root];//root先指向祖先
    while(son!=root)//把root原先所在的位置以及其上级直接指向祖先
    {
        tmp=pre[son];
        pre[son]=root;
        son=tmp;
    }
    return root;
}

int Find2(int root)
{
    while(root!=pre[root])
        root=pre[root];
    return root;
}

int Find3(int root)
{
    if(root==pre[root]) return root;;
    root=Find3(pre[root]);
}

int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        if(n==0&&m==0) return 0;
        //memset(pre,-1,sizeof(pre));
        for(int i=0;i<n;i++) pre[i]=i;//不能用memset成同一值，否则死循环
        for(int i=0;i<m;i++)
        {
            int num,root;//root即每个社团的主席（祖先）
            scanf("%d%d",&num,&root);//先输入个root
            for(int j=1;j<num;j++)
            {
                int cnt;
                scanf("%d",&cnt);
                pre[Find1(root)]=Find1(cnt);//是Find（root）而不是root，因为多个社团可能共有同一个人，root的祖先跟在cnt的祖先之下
                root=cnt;//更新一下root
            }
        }

        int ans=0;
        for(int i=0;i<n;i++)
            if(Find1(pre[i])==Find1(0))
                ans++;
        printf("%d\n",ans);
    }

    return 0;
}

总结：

1. pre数组初始化为各点本身，不能为同一值
2. return pre[root]=Find(pre[root]);//路径压缩的递归写法