tinystl实现(第十五步:并查集实现)

经过长时间的学习终于可以开始tinystl的仿(chao)写工作了,本文参考了这位大神的github,坦白讲我只是补充了注释,因为tinystl的代码真的非常经典而我又没什么这种大型项目的经验,所以只能这样做,不过相信能够有助于大家的学习
#强烈建议按顺序阅读本专栏

#pragma once
#ifndef _UF_SET_H_
#define _UF_SET_H_

#include 
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_

你可能感兴趣的:(tinystl,c++再学习,学生)