POJ 1611 The Suspects

 

链接：http://poj.org/problem?id=1611

题意：N个人中如果某个人和另一个人接触了那么这两个人都有可能感染SARS，给出M组有接触的人群，问感染SARS的嫌疑者人数。（0号人一定是SARS感染嫌疑者）

分析：并查集。每个集合有一个节点记录该集合的人数。

 

Source Code：

#include<cstdio>
#define maxn 30010

int father[maxn],num[maxn];
int n,m;

void make_set(){
    for(int i=0;i<=n;i++){
        father[i]=i;
        num[i]=1;
    }
}

int find(int x){
    if(x!=father[x]) father[x]=find(father[x]);
    return father[x];
}

void Union(int a,int b){
    int x=find(a);
    int y=find(b);
    if(x==y) return ;
    father[x]=y;
    num[y]+=num[x];
}

int main()
{
    while(scanf("%d %d",&n,&m),n+m){
        make_set();
        for(int i=0;i<m;i++){
            int a,b,k;
            scanf("%d",&k);
            if(k<=0) continue;
            scanf("%d",&a);
            for(int j=2;j<=k;j++){
                scanf("%d",&b);
                Union(a,b);      //这里处理集合合并的方法很好
            }
        }
        printf("%d\n",num[find(0)]);
    }
    return 0;
}