程序设计思维与实践 Week6 作业B 并查集实现

题意：

给定多组数据，每组数据中有学生数量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;
}