并查集原理
将不同元素划分到不同的集合,开始时, 每个元素就是一个集合,然后按照一定的规律进行集合的合并,这样就可以查找集合的数目,元素的归属,将复杂问题简化。
并查集应用
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;
};