并查集详解 —— c++实现

并查集原理

将不同元素划分到不同的集合,开始时, 每个元素就是一个集合,然后按照一定的规律进行集合的合并,这样就可以查找集合的数目,元素的归属,将复杂问题简化。

并查集应用

1. 查找元素属于哪一个集合
2. 查看两个元素是否在同一个集合
3. 合并两个集合
4. 集合的个数

并查集实现

class UnionFindSet {
public:
	UnionFindSet(size_t number) {
		_ufs.resize(number, -1);
	}
	int FindRoot(int index) {
		while (_ufs[index] >= 0) {
			index = _ufs[index];
		}
		return index;
	}
	bool Unoin(int x1, int x2) {
		int u1 = FindRoot(x1);
		int u2 = FindRoot(x2);
		if (u1 == u2) {
			return false;
		}
		_ufs[u1] += _ufs[u2];
		_ufs[u2] = u1;
		return true;
	}
	int Count() {
		int ret = 0;
		for (auto e : _ufs) {
			if (e < 0) {
				++ret;
			}
		}
		return ret;
	}
private:
	vector<int> _ufs;
};

你可能感兴趣的:(算法与数据结构)