并查集的启发式合并

直接实现的话，时间复杂度最坏可以到O(n)。
两个常见优化，启发式合并，路径压缩。
启发式合并：把大小较小的集合挂在较大的集合上。（有的写法是考虑深度而不是大小）
路径压缩：询问过的点到根节点的路径，都直接挂在根节点上。
实现其中任意一个时间复杂度变为O(logn)。
实现其中两个，时间复杂度变为O(α(n))，其中α(n)是阿克曼函数的反函数，可以认为非常小。

多数情况下为了简单，都实现路径压缩（只需要一句赋值）而不实现启发式合并（需要记录深度）。

在某些题目中由于会爆栈，需要使用非递归的find函数。

启发式合并的时间复杂度证明，请参考文献：

doi:10.1145/73007.73040
S2CID 13470414

const int N = 1e4 + 10;
int p[N], rk[N];  //rk是以i为根节点的集合的深度
int n, m;

void init()
{
	for(int i = 1; i <= n; i ++)
		p[i] = i, rk[i] = 0;
}

int find(int x)          //路径压缩，非递归式
{
	int son = x, tem;
	while(p[x] != x) x = p[x];
	while(son != x)
	{
		tem = p[son];
		p[son] = x;
		son = tem;
	}
	return x;
}

void unit(int r1, int r2)   
{
	if(rk[r1] > rk[r2])    //启发式合并
		p[r2] = r1;
	else if(rk[r1] < rk[r2])
		p[r1] = r2;
	else              //如果深度相等
	{
		p[r1] = r2;
		rk[r2] ++;
	}
}