Hrbust-1073病毒(并查集的简单应用)

Description

某种病毒袭击了某地区，该地区有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

 代码

#include <stdio.h>
const int MAXN = 50005;
int f[MAXN];

int find(int x)
{//查找x的根节点
    if (x != f[x])
    {
        f[x] = find(f[x]);//路径压缩，直接赋值为找到的根节点
    }
    return f[x];
}

void union_set(int x, int y)
{//合并
    int px = find(x);
    int py = find(y);
    if(px != py)
    {
        f[px] = py;
    }
}
 int main()
 {
     int cnt, f0, m, n, x, y;
     while(scanf("%d %d", &n, &m) != EOF)
     {
         for(int i = 0; i < MAXN; i++)
           f[i] = i;
         for(int i = 0; i < m; i++)
         {
             scanf("%d %d", &x, &y);
             union_set(x, y);
         }
         cnt = 1;
         f0 = find(0);
         for(int i = 1; i < n; i++)
         {
             if(f0 == find(i))
               cnt++;
         }
         printf("%d\n", cnt);
     }
     return 0;
 }