并查集问题

并查集

并查集是一种树型的数据结构，用于处理一些不交集的合并及查询问题。
这里面主要就两种操作：
1、查询find()函数：就是一种递归查找，需要找x的根，就不断寻找x的父亲，直到找到那个根。

int find(int x)
{
	if(pa[x]==x)
		return x;
	return find(pa[x]);
}

对于查找操作，这里有一个优化的算法，使得查找的复杂度降低，那就是路径压缩，在查找的时候，顺便将所有的结点都挂在根上：

2、合并操作unite()函数：合并操作建立在查找的基础上，就是把x所在的根挂在y所在的根下。

bool unite(int x,int y)
{
	x=find(x);y=find(y);
	if(x==y) return false;
	pa[x]=y;
	return true;
}

合并操作也有优化的地方，我们应该让“小”数挂在“大”数上面。这里有两个点来衡量，一是这个节点的子树高度；二是这个节点子树节点个数。如果用h[i]记录i节点的高度，w[i]记录节点个数；那么程序可以改为：

bool unite(int x,int y)
{
	x=find(x);y=find(y);
	if(x==y) return false;
	if（h[x]>h[y]) pa[y]=x;//x是大树，也可以比较w[x]、w[y]
	else pa[x]=y;
	return true;
}

具体题目：

新型冠状病毒肺炎（Corona Virus Disease 2019，COVID-19），简称“新冠肺炎”，是指2019新型冠状病毒感染导致的肺炎。
如果一个感染者走入一个群体，那么这个群体需要被隔离！
小A同学被确诊为新冠感染，并且没有戴口罩！！！！！！
危！！！
时间紧迫！！！！
需要尽快找到所有和小A同学直接或者间接接触过的同学，将他们隔离，防止更大范围的扩散。
众所周知，学生的交际可能是分小团体的，一位学生可能同时参与多个小团体内。
请你编写程序解决！戴口罩！！

Input
多组数据，对于每组测试数据：
第一行为两个整数n和m（n = m = 0表示输入结束，不需要处理），n是学生的数量，m是学生群体的数量。0 < n <= 3e4 ， 0 <= m <= 5e2
学生编号为0~n-1
小A编号为0
随后，m行，每行有一个整数num即小团体人员数量。随后有num个整数代表这个小团体的学生。

所有代码：

#include<stdio.h>
//using namespace std;
int pa[30002],rnk[30002];
void init(int n)
{
	for(int i=0;i<n;i++)
	{
		pa[i]=i;rnk[i]=1;
	}
}
int find(int x)
{
	return pa[x]==x ? x: pa[x]=find(pa[x]);//路径压缩
}
bool unite(int x,int y)
{
	x=find(x);y=find(y);
	if(x==y) return false;
	pa[x]=y;
	rnk[y]+=rnk[x];
	return true;
}
int main()
{
	int n,m;
	while(1)
	{
		scanf("%d %d",&n,&m);
		if(n==0&&m==0) break;
		init(n);
		for(int i=0;i<m;i++)
		{
			//printf("  999  \n");
			int num,last=-1;
			scanf("%d",&num);
			for(int j=0;j<num;j++)
			{
				int p;
				scanf("%d",&p);//团体里面的人p
				if(last!=-1)
					unite(p,last);
				last=p;
			}
		}
		printf("%d\n",rnk[find(0)]);
	}
	return 0;
}