病毒
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
",sum);
56 }
57 return 0;
58 }
Freecode : www.cnblogs.com/yym2013