并查集笔记

并查集（Union Find）

1、查找（Find）

判断两个值是否在同一个集合中。

复杂度为：O（n）↓

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

优化代码↓
复杂度：O（1）

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

优化原因：
优化前：当查找某一个值所在的集合时，只是单纯的查找某一值所在的集合。那么当集合嵌套很深的情况，即 A 属于 B 集合，B 集合又属于 C 集合， ……，D 集合又属于 E 集合。那么每次查找的时候都需要遍历所有的节点。
优化后：在查找到某一值所在的集合后，直接把该值指向集合属于的集合，（还用上述的嵌套关系）即在知道 D 属于 E 集合后，把 D 指向 E；在得知 C 集合属于 D 集合后，直接把 C 指向 D 所属的集合即 E 集合……，这样以后每个值只会指向最大的那个集合，若再次执行 find 只需 O（1）。

2、合并（Union）

可以把两个值对应的集合合并。

复杂度为：O（n）↓

void union(int a, int b){
	int root_a = find(a);
	int root_b = find(b);
	if(root_a != root_b){
		father[root_a] == root_b;
	}
}

经过上述的优化后，合并阶段的复杂度也为O（1）.

3、结合后的完整代码

public class UnionFind{
	private int[] father = null;
	
	int find(int x){
		if(father[x] == x){
				return x;
		}
		father[x] = find(father[x]);
		return father[x];
	}

	void union(int a, int b){
		int root_a = find(a);
		int root_b = find(b);
		if(root_a != root_b){
			father[root_a] == root_b;
		}
	}
}