欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-05-24 19:05:50
...

经过长时间的学习终于可以开始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_

相关标签: tinystl stl c++