并查集（2）病毒

                                                   病 毒

某种病毒袭击了某地区，该地区有N(1≤N≤50000)人，分别编号为0,1,...,N-1，现在0号已被确诊，所有0的直接朋友和间接朋友都要被隔离。例如：0与1是直接朋友，1与2是直接朋友，则0、2就是间接朋友，那么0、1、2都须被隔离。现在，已查明有M(1≤M≤10000)个直接朋友关系。如：0,2就表示0,2是直接朋友关系。
请你编程计算，有多少人要被隔离。

Input

第一行包含两个正整数N(1≤N≤50000),M(1≤M≤100000)，分别表示人数和接触关系数量；
在接下来的M行中，每行表示一次接触，；
每行包括两个整数U, V(0 <= U, V < N)表示一个直接朋友关系。

Output

输出数据仅包含一个整数，为共需隔离的人数（包含0号在内）。

Sample Input

100 4
0 1
1 2
3 4
4 5

Sample Output

3

代码（1）

#include<stdio.h>
int f[500000]={0},n,m,k,sum=0;
void init()
{
    int i;
    for(i=0;i<=n;i++)
        f[i]=i;
}
int getf(int v)
{
    if(f[v]==v)
        return v;
    else{
        f[v]=getf(f[v]);
        return f[v];
    }
}
void merge(int v,int u)
{
    int t1,t2;
    t1=getf(v);
    t2=getf(u);
    if(t1!=t2)
    {
        f[t2]=t1;
    }
}
int main()
{
    int i,x,y,q;
    sum=0;
    scanf("%d %d",&n,&m);
    init();
    for(i=1;i<=m;i++)
    {
        scanf("%d %d",&x,&y);
        merge(x,y);
    }
    for(i=0;i<=n;i++)
 {
     if(getf(0)==getf(i))
        sum++;
 }
   printf("%d\n",sum);
    return 0;
}

代码（2）

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
using namespace std;
int fa[50002],a,b,m,n;
void build(int qwq)
{
    for(int i=1;i<=qwq;i++)
        fa[i]=i;
        return ;
}
int find(const int &x)
{
    return fa[x]==x?x:fa[x]=find(fa[x]);
}
bool che(const int &x,const int &y)
{
    return find(x)==find(y);
}
void mer(const int &x,const int &y)
{
    if(!che(x,y))
        fa[fa[x]]=fa[y];
    return ;
}

int main()
{
    int i,ans=0;
    scanf("%d%d",&n,&m);
    build(n);
    for(i=1;i<=m;i++)
        scanf("%d%d",&a,&b),
        mer(a,b);
    for(i=0;i<=n;i++)
    {
        if(che(0,i))
            ans++;
    }
    printf("%d\n",ans);
    return 0;
}