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

39 篇文章 4 订阅
28 篇文章 37 订阅

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

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值