Description
学校里面有n (0 < n <= 30000 )个同学,编号为(0 ~ n-1),其中0号同学为SARS的疑似患者。给出m (0 <= m <= 500)组同学的集合,每个集合里的同学都经常在一起,问这n名同学中有几个是疑似患者。(与0号同学有直接或间接接触的都算为疑似患者,包括0号同学)
Input
输入有多组,第一行为两个整数n,m。后续m行第一个整数表示集合里面人的个数,然后是每个人的编号。输入以0,0结束。
Output
每组测试输出一行,即一个整数表示疑似患者的人数。
Sample Input
200 2
1 5
5 1 2 3 4 5
0 0
Sample Output
1
思路:用并查集将所有数连成树,然后查找0所在的树的根是几,再从其他的点寻找与0的根相等即为患者。
代码:
#include <stdio.h>
#include <stdlib.h>
#define MAX_N 30000
int f[MAX_N+5];
int n,m;
void init(int n)//初始化,起初为n个树
{
int i;
for(i=0;i<=n;i++) f[i] = i;
}
int find(int v)//寻找根
{
if(v != f[v])
f[v] = find(f[v]);
return f[v];
}
void merge(int x,int y)//合并
{
int fx = find(x);
int fy = find(y);
if(fx != fy)
f[fy] = fx;
}
int main()
{
int i,j,num,x,y;
while(scanf("%d%d",&n,&m)!=EOF && (n || m))
{
init(n);
for(i=0;i<m;i++)
{
scanf("%d",&num);
scanf("%d",&x);
for(j=1;j<num;j++)
{
scanf("%d",&y);
merge(x,y);
}
}
int point = find(0),cot = 1;
for(i=1;i<=n;i++)
{
if(find(i) == point)//与0的根相等即为患者
cot++;
}
printf("%d\n",cot);
}
return 0;
}