带权并查集 路径压缩

并查集

并查集是一种树形结构，又叫“不相交集合”，保持了一组不相交的动态集合，每个集合通过一个代表来识别，代表即集合中的某个成员，通常选择根做这个代表。
其实并查集很常用，比如果刚进入大学还没有分班，这个时候你并不知道你的同学，此时可以看做每个人都是独立的。而分班后，则你与同班的同学有了关系，同班同学的关系。在并查集中，若同学a与同学b是同班同学，则他们都会有一个共同点——有着相同的root(根)，这个根也是同班的某一个同学，只是借助它来区分a与b的关系。比如说a与root同学c是同班同学，b与root同学c也是同班同学，则a与b也是同班同学。

以下是带路径压缩的一个并查集模板：
find(int p)表示找这个元素的根
connect(int p,int q)将所处不同根的两个合并 比如将一个班与另一个班合并组成一个更大的班
图例：（union与connect相同）
此时每个元素所在“根系”中都只有自己，每元素都是root (i=id[i])

结果：

此时 如果find(3)结果为4 代表3的root为4

//Union Find class
class UF {
public:
	int* id;	//父链接数组
	int* sz;	//各个节点所对应的分量的大小
	int count;	//连通分量的个数
	UF(int n) {
		count = n;		//初始化 任意两个元素不连通因此由n个连通分量
		id = new int[n];
		sz = new int[n];
		fill(sz, sz + n, 1);	// 初始每个点都只有一个分量就是自己
		for (int i = 0; i < n; i++) id[i] = i;
	}
	int connect(int p, int q) {
		int i = find(p);
		int j = find(q);
		if (i == j) return 0;
		if (sz[i] < sz[j]) { id[i] = j; sz[j] += sz[i]; }
		else { id[j] = i; sz[i] += sz[j]; }
		count--;
	}

	int find(int p) {
		while (p != id[p]) {
			id[p] = id[id[p]];	//路径压缩 让父亲指向其祖父
			p = id[p];
			
		}
		return p;
	}
};

路径压缩：
树的深度越低，查找根的速度就越慢。
因此我们要做的是将深度降低

int find(int p) {
		while (p != id[p]) {
			id[p] = id[id[p]];	//路径压缩 让父亲指向其祖父
			p = id[p];
		}
		return p;
	}

每次查找的时候 都让其父节点指向祖父节点
但[算法第四版]提到在实际运用中 两种展开差异不大
还一种方式是每次查根都让其指向root 如下图所示


首先让p指向root 再让p的父节点6指向root 再让6的父节点指向root 依次类推

注：本文配图来自算法第四版的公开课