hrbustoj 1073:病毒（并查集，入门题）
病毒
Time Limit: 1000 MS Memory Limit: 65536 K
Total Submit: 719(185 users) Total Accepted: 247(163 users) Rating: Special Judge: No
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
并查集，入门题。
思路是先根据关系建立关系树，然后记录0号小朋友的根节点，依次与剩下所有的小朋友的根节点比较，相等则计数+1，表示这是和0号小朋友有关系的，即被感染的人。最后输出总计数（即被0号小朋友感染的人数）。
套模板很容易过。
代码：
1 #include <stdio.h>
2
3 /* 并查集模板
4 */
5 int UFS_NUM; //并查集中元素总数
6 typedef struct node{
7 int data; //节点对应的编号
8 int rank; //节点对应秩
9 int parent; //节点对应的双亲下标
10 }UFSTree; //并查集树的节点类型
11 void MAKE_SET(UFSTree t[]) //初始化并查集树
12 {
13 int i;
14 for(i=0;i<UFS_NUM;i++){
15 t[i].data = i; //数据为该点编号
16 t[i].rank = 0; //秩初始化为0
17 t[i].parent = i; //双亲初始化为指向自己
18 }
19 }
20 int FIND_SET(UFSTree t[],int x) //在x所在的子树中查找集合编号
21 {
22 if(t[x].parent == x) //双亲是自己
23 return x; //双亲是自己，返回 x
24 else //双亲不是自己
25 return FIND_SET(t,t[x].parent); //递归在双亲中查找x
26 }
27 void UNION(UFSTree t[],int x,int y) //将x和y所在的子树合并
28 {
29 x = FIND_SET(t,x); //查找 x 所在分离集合树的编号
30 y = FIND_SET(t,y); //查找 y 所在分离集合树的编号
31 if(t[x].rank > t[y].rank) //y 节点的秩小于 x节点的秩
32 t[y].parent = x; //将 y 连接到 x 节点上，x 作为 y 的双亲节点
33 else{ //y 节点的秩大于等于 x 节点的秩
34 t[x].parent = y; //将 x 连接到 y 节点上，y 作为 x 的双亲节点
35 if(t[x].rank==t[y].rank) //x 和 y的双亲节点秩相同
36 t[y].rank++; //y 节点的秩增 1
37 }
38 }
39 int main()
40 {
41 int i,n,m,x,y;
42 while(scanf("%d%d",&n,&m)!=EOF){
43 UFSTree t[50005];
44 UFS_NUM = n;
45 MAKE_SET(t);
46 for(i=1;i<=m;i++){
47 scanf("%d%d",&x,&y);
48 UNION(t,x,y);
49 }
50 int f0 = FIND_SET(t,0);
51 int sum=0;
52 for(i=0;i<n;i++)
53 if(FIND_SET(t,i)==f0)
54 sum++;
55 printf("%d\n",sum);
56 }
57 return 0;
58 }
Freecode : www.cnblogs.com/yym2013
