题意:
给定多组数据,每组数据中有学生数量n和学生团体的数量m,接下来的m组数据即为该组学生团体的人数和其中的人员编号,已知0号人员感染,只要和0号人员接触过,即在一个团体中过,即需要隔离,求最后一共要隔离的人的数量。
input:
多组数据,对于每组测试数据:
第一行为两个整数n和m(n = m = 0表示输入结束,不需要处理),n是学生的数量,m是学生群体的数量。0 < n <= 3e4 , 0 <= m <= 5e2
学生编号为0~n-1
小A编号为0
随后,m行,每行有一个整数num即小团体人员数量。随后有num个整数代表这个小团体的学生。
output:
输出要隔离的人数,每组数据的答案输出占一行。
思路:
根据并查集来进行求解。对于每组样例。有一个sum数组存取每个点所在集合中的人的数量(一般为根节点的才是当前集合的人的数量)。一个N数组记录每个点的父节点,最终的父节点即为标记该集合的特殊点。在每个团体中,先读取第一个数,然后依次将后面的数所在的集合与第一个数所在的集合进行合并。即根据while循环不断找父节点,直到找到根节点,即可判断是否为同一类。如果所在的集合相同,则返回。否则将后面的集合中的数的父节点全部标记为第一类的根节点,并将第一类的根节点所标记的sum值,加上第二类的sum值。全部处理完成后,找到0的根节点,即可得到所在的集合的数量。
代码:
#include <iostream>
#include <cstdio>
using namespace std;
int n,m;
int N[100000];
int sum[100000];
void unit(int x, int y){
while(x!=N[x]){ //根
x=N[x];
}
int b=y;
while(b!=N[b]){ //根
b=N[b];
}
if(b==x) return; //已经是一组
sum[x]+=sum[b];
int t=N[y]; //t是y的头
while(y!=t){
N[y]=x;
y=t;
t=N[y];
}
N[t]=x;
}
#include <iostream>
#include <cstdio>
using namespace std;
int n,m;
int N[100000];
int sum[100000];
void unit(int x, int y){
while(x!=N[x]){ //根
x=N[x];
}
int b=y;
while(b!=N[b]){ //根
b=N[b];
}
if(b==x) return; //已经是一组
sum[x]+=sum[b];
int t=N[y]; //t是y的头
while(y!=t){
N[y]=x;
y=t;
t=N[y];
}
N[t]=x;
}