并查集UnionFind Java实现

并查集java实现

class UnionFind{
	private int count; // 集合计数
	private int[] parents;	// 记录根节点
	private int[] size;	// 记录集合中的节点数

	/* 初始化，每个节点的根节点都是自身，共 n 个集合 */
	public UnionFind(int n) {
		count = n;
		parents = new int[n];
		size = new int[n];
		for (int i = 0; i < n; ++i) {
			parents[i] = i;
			size[i] = 1;
		}
	}

	/* 寻找根节点 */
	public int find(int x) {
		while (x != parent[x]) {
			parent[x] = parent[parent[x]]; // 优化，压缩路径
			x = parent[x];
		}
		return x;
	}

	/* 合并集合 */
	public boolean union(int x, int y) {
		x = find(x);
		y = find(y);
		if (x == y) return false;
		/* 优化，将元素少的集合指向元素多的集合的根结点，从而减少层数 */
		if (size[x] < size[y]) {
			int tmp = x;
			x = y;
			y = tmp;
		}
		parents[y] = x;
		size[x] += size[y]
		-- count;
		return true;
	}

	/* 获取集合数 */
	public int getCount() {
		return count;
	}
}