POJ 1611---The Suspects(并查集)

题意：0疑似有传染病，和0在一起的都疑似被传染（这些人也会传染别人），求有多少个人可能有传染病；

直接代码＋注释（16ms）

 方法1：

#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;

const int maxn=20000000;
int par[maxn],m,n;

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

void unionn(int a,int b)
{
    int fa=find(a);
    int fb=find(b);
    if(par[fb]!=fb) par[fa]=fb;     //如果fb不是头 fa并入fb即头为fb
    else par[fb]=fa;                //如果fb是头 fb并入fa
}

int main()
{
    while(scanf("%d%d",&n,&m)&&(m||n)){
        int p,a,b;
        for(int i=0;i<=n;i++){
            par[i]=i;                //每个人的头为自己
        }
        while(m--){
            scanf("%d",&p);
            p--;
            scanf("%d",&a);
            for(int i=0;i<p;i++){
                scanf("%d",&b);
                unionn(a,b);          //a后边的所有元素都并入a即a为头
            }
        }
        int sum=0;
        for(int i=0;i<n;i++){
            while (par[i]!=i&&par[i]!=par[par[i]])      //如果i不是头并且i的上一级也不是头 找i的头 par[i]=头
                par[i]=par[par[i]];
        }
        for(int i=0;i<n;i++){
            if(par[i]==par[0]) sum++;               //谁和0是同一个头 说明被传染
        }
        printf("%d
",sum);
    }
    return 0;
}

 方法二：（方法二和方法一意思差不多，就是更简洁了一些）

#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;

const int maxn=300100;
int par[maxn],m,n;

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

void unionn(int a,int b)
{
    int fa=find(a);
    int fb=find(b);
    if(fa==fb) return;
    else par[fb]=fa;
}

int main()
{
    while(scanf("%d%d",&n,&m)&&(m||n)){
        int p,a,b;
        for(int i=0;i<=n;i++){
            par[i]=i;
        }
        while(m--){
            scanf("%d",&p);
            p--;
            scanf("%d",&a);
            for(int i=0;i<p;i++){
                scanf("%d",&b);
                unionn(a,b);
            }
        }
        int sum=0;
        for(int i=0;i<n;i++){
            if(find(i)==find(0)) sum++;
        }
        printf("%d
",sum);
    }
    return 0;
}