POJ 1611 The Suspects (并查集)

Description

　　严重急性呼吸系统综合症( SARS), 一种原因不明的非典型性肺炎,从2003年3月中旬开始被认为是全球威胁。为了减少传播给别人的机会, 最好的策略是隔离可能的患者。在Not-Spreading-Your-Sickness大学( NSYSU), 有许多学生团体。同一组的学生经常彼此相通,一个学生可以同时加入几个小组。为了防止非典的传播,NSYSU收集了所有学生团体的成员名单。他们的标准操作程序(SOP)如下：一旦一组中有一个可能的患者, 组内的所有成员就都是可能的患者。然而,他们发现当一个学生被确认为可能的患者后不容易识别所有可能的患者。你的工作是编写一个程序, 发现所有可能的患者。

Input

输入文件包含多组数据。

对于每组测试数据：

第一行为两个整数n和m, 其中n是学生的数量, m是团体的数量。0 < n <= 30000，0 <= m <= 500。每个学生编号是一个0到n-1之间的整数，一开始只有0号学生被视为可能的患者。紧随其后的是团体的成员列表，每组一行。每一行有一个整数k，代表成员数量。之后,有k个整数代表这个群体的学生。一行中的所有整数由至少一个空格隔开。n = m = 0表示输入结束，不需要处理。

Output

对于每组测试数据, 输出一行可能的患者。

解题思路：

　　利用并查集，优先把小的作为父节点，所以只要 0 出现那么就会成为根节点，最后利用压缩路径把所有以 0 为根节点的节点都变为 0 的儿子节点。。最后统计下有多少个以 0 为根节点的就OK啦~~~~~

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <cctype>
#include <algorithm>
using namespace std;
const int MAXN = 1e5 + 3;
int pre[MAXN];
int a[MAXN]={0};
int Find(int x)
{
    int r = x;
    while(pre[r] != r)
    {
        r = pre[r];
    }
    int i = x,j;
    while(pre[i] != r)
    {
        j = i;
        i = pre[i];
        pre[j] = r;
    }
    return r;
}

void Mix(int a,int b)
{
    int x = Find(a);
    int y = Find(b);
    if(x > y) //小的作为父节点~~
    {
        pre[x] = y;
    }
    if(x < y)
    {
        pre[y] = x;
    }
}


void Mst()
{
    for(int i = 0; i <MAXN; i++)
    {
        pre[i] = i;
    }
}

int main()
{
   //freopen("in.txt","r",stdin);
   //freopen("out.txt","w",stdout);
    int m,n;
    while(~scanf("%d%d",&n,&m)&&(n||m))
    {
        Mst();
        for(int i = 0; i < m; i++)
        {
            int num;
            scanf("%d",&num);
            memset(a,0,sizeof(a));
            for(int i = 0;i < num;i++)
            {
                scanf("%d",&a[i]);
                if(i>0)
                    Mix(a[i],a[i-1]);
            }
        }
        for(int i = 0; i < n; i++)
            Find(i);
        int ans =0;
        for(int i = 0; i < n; i++)
            if(pre[i] == 0)
                ans++;
        printf("%d
",ans);
    }
    return 0;
}