经过长时间的学习终于可以开始tinystl的仿(chao)写工作了,本文参考了这位大神的github,坦白讲我只是补充了注释,因为tinystl的代码真的非常经典而我又没什么这种大型项目的经验,所以只能这样做,不过相信能够有助于大家的学习
#强烈建议按顺序阅读本专栏
#pragma once
#ifndef _UF_SET_H_
#define _UF_SET_H_
#include <cstring>
namespace mySTL {
template<size_t N>
class uf_set {//并查集
uf_set();
int Find(int index);
void Union(int index1, int index2);
void Clear();
private:
int parent[N];//parent[i] = -n 表示节点i是根节点且以i为根的树中共有n个节点
};
//构造就是清空
template<size_t N>
uf_set<N>::uf_set() {
Clear();
}
template<size_t N>
int uf_set<N>::Find(int index) {
auto root = index;
for (; parent[root] >= 0; root = parent[root]) {}//一直查,查到为止
while (root != index) {//路径压缩
auto t = parent[index];
parent[index] = root;//全部子点都直接链接到根
index = t;
}
return root;
}
//
template<size_t N>
void uf_set<N>::Union(int index1, int index2) {
auto root1 = Find(index1), root2 = Find(index2);
auto total_nodes = parent[root1] + parent[root2];//total nodes
if (parent[root1] > parent[root2]) {//加权合并(由于已经find所以都是负数)
parent[root1] = root2;
parent[root2] = total_nodes;
}
else {
parent[root2] = root1;
parent[root1] = total_nodes;
}
}
//全部赋-1就是清空
template<size_t N>
void uf_set<N>::Clear() {
memset(parent, -1, sizeof(int) * N);
}
}
#endif // !_UF_SET_H_