【C++】STL —— map和set的模拟实现

已于 2022-11-29 16:07:37 修改
阅读量822 收藏 6
点赞数 4
分类专栏： C++入门到进阶 文章标签： 数据结构
于 2022-11-29 16:07:18 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/sjsjnsjnn/article/details/128096084
版权

目录

一、基础铺垫

二、基本结构分析

1. 节点结构分析 

2. 模板参数中仿函数分析

三、正向迭代器

四、封装完成的红黑树

五、map的模拟实现

六、set的模拟实现

一、基础铺垫

        在前面的博客中我们了解了map和set的基本使用，以及对二叉搜索树、AVL树和红黑树的概念和简单实现有了一定的了解后，对于map和set来说，他们的底层实现都是基于红黑树的。我们知道set是key的模型、map是key/value的模型，那么一棵红黑树是如何实现出map和set的呢？

二、基本结构分析

1. 节点结构分析 

        以上是截取的部分源码结构，我们可以看到set的模板参数是Key，map的模板参数是Key和T；set将Key这个模板参数typedef成了key_type和value_type，map将Key这个模板参数typedef成了key_type、将T这个模板参数typedef成了data_type；

        他们同时调用红黑树时，都是用的value_type，对应set的value_type是Key，对应map的value_type是pair<const key，T>；这意味着红黑树只看传过来的value_type是什么类型的，当上层容器是set的时候，结点当中存储的是键值Key；当上层容器是map的时候，结点当中存储的就是<Key, Value>键值对。

结点结构： 

enum Colour
{
	RED,
	BLACK
};

template<class T>
struct RBTreeNode
{
	RBTreeNode<T>* _left;
	RBTreeNode<T>* _right;
	RBTreeNode<T>* _parent;

	T _data;//存储的数据类型：如果是set，T对应的就是K；如果是map，T对应的就是pair
	Colour _col;

	RBTreeNode(const T& data)
		:_left(nullptr)
		, _right(nullptr)
		, _parent(nullptr)
		, _col(RED)
		, _data(data)
	{}
};

红黑树基本结构： 

template<class k, class T>
class RBTree
{
	typedef RBTreeNode<T> Node;
private:
	Node* _root;
};

map的基本结构： 

namespace mlg
{
	template<class k, class v>
	class map
	{
	private:
		RBTree<k, pair<k, v>> _t;
	};
}

set的基本结构： 

namespace mlg
{
	template<class k>
	class set
	{
	private:
		RBTree<k, k> _t;
	};
}

2. 模板参数中仿函数分析

        因为红黑树中存储的是T类型，有可能是Key，也有可能是<Key, Value>键值对；那么当我们需要进行结点的键值比较时，应该如何获取结点的键值呢？

        当上层容器是set的时候T就是键值Key，直接用T进行比较即可，但当上层容器是map的时候就不行了，此时我们需要从<Key, Value>键值对当中取出键值Key后，再用Key值进行比较。

        因此，上层容器map需要向底层红黑树提供一个仿函数，用于获取T当中的键值Key，这样一来，当底层红黑树当中需要比较两个结点的键值时，就可以通过这个仿函数来获取T当中的键值了。

map的仿函数：

namespace mlg
{
	template<class k, class v>
	class map
	{
	public:
		struct MapKeyOfT
		{
			const k& operator()(const pair<k, v>& kv)
			{
				return kv.first;
			}
		};
	private:
		RBTree<k, pair<k, v>, MapKeyOfT> _t;
	};
}

set的仿函数：

namespace mlg
{
	template<class k>
	class set
	{
	public:
		struct SetKeyOfT
		{
			const k& operator()(const k& k)
			{
				return k;
			}
		};
	private:
		RBTree<k, k, SetKeyOfT> _t;
	};
}

三、正向迭代器

        红黑树的正向迭代器实际上就是对结点指针进行了封装，因此在正向迭代器当中实际上就只有一个成员变量，那就是正向迭代器所封装结点的指针。 

template<class T, class Ref, class Ptr>
struct RBTreeIterator
{
	typedef RBTreeNode<T> Node;//结点的类型
	typedef RBTreeIterator<T, Ref, Ptr> Self;//正向迭代器的类型
	Node* _node;//正向迭代器所封装结点的指针

	RBTreeIterator(Node* node)//根据所给结点指针构造一个正向迭代器
		:_node(node)
	{}

	Ref operator*()
	{
		return _node->_data;//返回结点数据的引用
	}

	Ptr operator->()
	{
		return &_node->_data;//返回结点数据的指针
	}

	//前置++
    /*
    如果当前结点的右子树不为空，则++操作后应该找到其右子树当中的最左结点。
    如果当前结点的右子树为空，则++操作后应该在该结点的祖先结点中，找到孩子不在父亲右的祖先。
    */
	Self& operator++()
	{
		if (_node->_right)//结点的右子树不为空
		{
			Node* min = _node->_right;
			while (min->_left)
			{
				min = min->_left;
			}
			_node = min;
		}
		else  //结点的右子树为空
		{
			Node* cur = _node;
			Node* parent = cur->_parent;
			while (parent && cur == parent->_right)
			{
				cur = cur->_parent;
				parent = parent->_parent;
			}
			_node = parent;
		}
		return *this;
	}

	//前置--
    /*
    如果当前结点的左子树不为空，则--操作后应该找到其左子树当中的最右结点。
    如果当前结点的左子树为空，则--操作后应该在该结点的祖先结点中，找到孩子不在父亲左的祖先。
    */
	Self& operator--()
	{
		if (_node->_left)
		{
			Node* max = _node->_left;
			while (max->_right)
			{
				max = max->_right;
			}
			_node = max;
		}
		else
		{
			Node* cur = _node;
			Node* parent = cur->_parent;
			while (parent && cur == parent->_left)
			{
				cur = parent;
				parent = parent->_parent;
			}
			_node = parent;
		}
		return *this;
	}

	//判断两个正向迭代器是否不同
	bool operator!=(const Self& s)const
	{
		return _node != s._node;
	}

    //判断两个正向迭代器是否相同
	bool operator==(const Self& s)const
	{
		return _node == s._node;
	}
};

四、封装完成的红黑树

#pragma once
#include <iostream>
#include <algorithm>
using namespace std;

enum Colour
{
	RED,
	BLACK
};

template<class T>
struct RBTreeNode
{
	RBTreeNode<T>* _left;
	RBTreeNode<T>* _right;
	RBTreeNode<T>* _parent;

	T _data;
	Colour _col;

	RBTreeNode(const T& data)
		:_left(nullptr)
		, _right(nullptr)
		, _parent(nullptr)
		, _col(RED)
		, _data(data)
	{}
};

template<class T, class Ref, class Ptr>
struct RBTreeIterator
{
	typedef RBTreeNode<T> Node;
	typedef RBTreeIterator<T, Ref, Ptr> Self;
	Node* _node;

	RBTreeIterator(Node* node)
		:_node(node)
	{}

	Ref operator*()
	{
		return _node->_data;
	}

	Ptr operator->()
	{
		return &_node->_data;
	}
	//前置++
	Self& operator++()
	{
		if (_node->_right)
		{
			Node* min = _node->_right;
			while (min->_left)
			{
				min = min->_left;
			}
			_node = min;
		}
		else
		{
			Node* cur = _node;
			Node* parent = cur->_parent;
			while (parent && cur == parent->_right)
			{
				cur = cur->_parent;
				parent = parent->_parent;
			}
			_node = parent;
		}
		return *this;
	}
	//前置--
	Self& operator--()
	{
		if (_node->_left)
		{
			Node* max = _node->_left;
			while (max->_right)
			{
				max = max->_right;
			}
			_node = max;
		}
		else
		{
			Node* cur = _node;
			Node* parent = cur->_parent;
			while (parent && cur == parent->_left)
			{
				cur = parent;
				parent = parent->_parent;
			}
			_node = parent;
		}
		return *this;
	}
	
	bool operator!=(const Self& s)const
	{
		return _node != s._node;
	}
	bool operator==(const Self& s)const
	{
		return _node == s._node;
	}
};


//set RBTree<k, k>
//map RBTree<k, pair<k, v>>
template<class k, class T, class keyofT>
class RBTree
{
	typedef RBTreeNode<T> Node;
public:
	typedef RBTreeIterator<T, T&, T*> iterator;
	typedef RBTreeIterator<T, const T&, const T*> const_iterator;
	
	iterator begin()
	{
		Node* min = _root;
		while (min && min->_left)
		{
			min = min->_left;
		}
		return iterator(min);
	}

	iterator end()
	{
		return iterator(nullptr);
	}

	RBTree()
		:_root(nullptr)
	{}

	RBTree(const RBTree<k, T, keyofT>& t)
	{
		_root = Copy(t._root);
	}
	Node* Copy(Node* root)
	{
		if (root == nullptr)
		{
			return nullptr;
		}
		Node* newRoot = new Node(root->_data);
		newRoot->_col = root->_col;

		newRoot->_left = Copy(root->_left);
		newRoot->_right = Copy(root->_right);
		if (newRoot->_left)
		{
			newRoot->_left->_parent = newRoot;
		}
		if (newRoot->_right)
		{
			newRoot->_right->_parent = newRoot;
		}
		return newRoot;
	}

	RBTree<k, T, keyofT>& operator=(RBTree<k, T, keyofT> t)
	{
		swap(_root, t._root);
		return *this;
	}

	~RBTree()
	{
		Destroy(_root);
		_root = nullptr;
	}

	void Destroy(Node* root)
	{
		if (root == nullptr)
		{
			return;
		}
		Destroy(root->_left);
		Destroy(root->_right);
		delete root;
	}
	iterator Find(const k& key)
	{
		Node* cur = _root;
		keyofT kot;
		while (cur)
		{
			if (kot(cur->_data) < key)
			{
				cur = cur->_right;
			}
			else if (kot(cur->_data) > key)
			{
				cur = cur->_left;
			}
			else
			{
				return iterator(cur);
			}
		}
		return end();
	}

	pair<iterator, bool> Insert(const T& data)
	{
		if (_root == nullptr)
		{
			_root = new Node(data);
			_root->_col = BLACK;
			return make_pair(iterator(_root), true);
		}

		keyofT kot;

		Node* parent = nullptr;
		Node* cur = _root;
		while (cur)
		{
			if (kot(cur->_data) < kot(data))
			{
				parent = cur;
				cur = cur->_right;
			}
			else if (kot(cur->_data) > kot(data))
			{
				parent = cur;
				cur = cur->_left;
			}
			else
			{
				return make_pair(iterator(cur), false);
			}
		}

		cur = new Node(data);
		Node* newnode = cur;
		cur->_col = RED; // 新增节点
		if (kot(parent->_data) < kot(data))
		{
			parent->_right = cur;
			cur->_parent = parent;
		}
		else
		{
			parent->_left = cur;
			cur->_parent = parent;
		}

		// 控制平衡
		while (parent && parent->_col == RED)
		{
			Node* grandfather = parent->_parent;
			if (parent == grandfather->_left)
			{
				Node* uncle = grandfather->_right;
				// 1、uncle存在且为红
				if (uncle && uncle->_col == RED)
				{
					// 变色+继续向上处理
					parent->_col = uncle->_col = BLACK;
					grandfather->_col = RED;

					cur = grandfather;
					parent = cur->_parent;
				}
				else // 2 + 3、uncle不存在/ 存在且为黑
				{
					if (cur == parent->_left)
					{
						// 单旋
						RotateR(grandfather);
						parent->_col = BLACK;
						grandfather->_col = RED;
					}
					else
					{
						// 双旋
						RotateL(parent);
						RotateR(grandfather);
						cur->_col = BLACK;
						grandfather->_col = RED;
					}
					break;
				}
			}
			else // parent == grandfather->_right
			{
				Node* uncle = grandfather->_left;
				if (uncle && uncle->_col == RED)
				{
					// 变色+继续向上处理
					parent->_col = uncle->_col = BLACK;
					grandfather->_col = RED;

					cur = grandfather;
					parent = cur->_parent;
				}
				else // 2 + 3、uncle不存在/ 存在且为黑
				{
					if (cur == parent->_right)
					{
						RotateL(grandfather);
						parent->_col = BLACK;
						grandfather->_col = RED;
					}
					else
					{
						RotateR(parent);
						RotateL(grandfather);
						cur->_col = BLACK;
						grandfather->_col = RED;
					}

					break;
				}
			}
		}
		_root->_col = BLACK;
		return make_pair(iterator(newnode), true);
	}

	//左单选
	void RotateL(Node* parent)
	{
		Node* subR = parent->_right;
		Node* subRL = subR->_left;
		parent->_right = subRL;
		if (subRL)
		{
			subRL->_parent = parent;
		}
		Node* parentParent = parent->_parent;
		subR->_left = parent;
		parent->_parent = subR;

		if (parent == _root)//这里表明原来是根
		{
			_root = subR;
			subR->_parent = nullptr;
		}
		else
		{
			if (parentParent->_left == parent)
			{
				parentParent->_left = subR;
			}
			else
			{
				parentParent->_right = subR;
			}
			subR->_parent = parentParent;
		}
	}

	//由单旋
	void RotateR(Node* parent)
	{
		Node* subL = parent->_left;
		Node* subLR = subL->_right;
		parent->_left = subLR;
		if (subLR)
		{
			subLR->_parent = parent;
		}
		Node* parentParent = parent->_parent;
		subL->_right = parent;
		parent->_parent = subL;
		if (parent == _root)//这里表明原来是根
		{
			_root = subL;
			_root->_parent = nullptr;
		}
		else
		{
			if (parentParent->_left == parent)
			{
				parentParent->_left = subL;
			}
			else
			{
				parentParent->_right = subL;
			}
			subL->_parent = parentParent;
		}
	}

	void InOrder()
	{
		_InOrder(_root);
	}
	void _InOrder(Node* root)
	{
		if (root == nullptr)
		{
			return;
		}
		_InOrder(root->_left);
		cout << root->_kv.first << ":" << root->_kv.second << endl;
		_InOrder(root->_right);
	}
private:
	Node* _root;
};

五、map的模拟实现

#include "RBTree.h"
namespace mlg
{
	template<class k, class v>
	class map
	{
	public:
		struct MapKeyOfT
		{
			const k& operator()(const pair<k, v>& kv)
			{
				return kv.first;
			}
		};
		typedef typename RBTree<k, pair<k, v>, MapKeyOfT>::iterator iterator;

		iterator begin()
		{
			return _t.begin();
		}

		iterator end()
		{
			return _t.end();
		}

		pair<iterator, bool> insert(const pair<k, v>& kv)
		{
			return _t.Insert(kv);
		}

		iterator find(const k& key)
		{
			return _t.Find(key);
		}
		v& operator[](const k& key)
		{
			auto ret = _t.Insert(make_pair(key, v()));
			return ret.first->second;
		}
	private:
		RBTree<k, pair<k, v>, MapKeyOfT> _t;
	};
}

六、set的模拟实现

#include "RBTree.h"
namespace mlg
{
	template<class k>
	class set
	{
	public:
		struct SetKeyOfT
		{
			const k& operator()(const k& k)
			{
				return k;
			}
		};

		typedef typename RBTree<k, k, SetKeyOfT>::iterator iterator;
		
		iterator begin()
		{
			return _t.begin();
		}
		
		iterator end()
		{
			return _t.end();
		}
		
		pair<iterator, bool> insert(const k& key)
		{
			return _t.Insert(key);
		}

		iterator find(const k& key)
		{
			return _t.Find(key);
		}
	private:
		RBTree<k, k, SetKeyOfT> _t;
	};
}

霄沫凡
关注
  • 4
    点赞
  • 踩
  • 6
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
专栏目录
STL标准模板库2
weixin_65012900的博客
10-19 433
queue队列容器、stack栈容器、priority_queue 优先队列容器、deque双向队列容器、set集合容器、multiset、map容器、multimaps、bitset容器
C++ 【set、map模拟实现】
我心如火
12-11 690
set和map使用、模拟实现
C++STL——map与set的模拟实现
qq_63580639的博客
04-23 544
改造红黑树，模拟实现set与map
C++STL详解(九)--使用红黑树封装实现set和map
清欢Allen的技术博客
04-19 619
如果我们用一棵KV模型的红黑树同时实现map和set,我们就需要控制map和set中的所传入红黑树中的模板参数,其中,封装过程中,为了与红黑树的模板参数区分,我们将红黑树中模板参数由V改成T.在插入中,结点中存储的类型T可能为Set容器中的key,也有可能是Map中的pair键值对,底层红黑树如何针对不同类型的容器进行比较呢?所以,当容器为set时,红黑树中的仿函数就实例化为set容器中的仿函数,当容器为map时,红黑树中的仿函数就实例化为map中的仿函数.正向迭代器还包括了==,!
STL之set
ancientear的博客
03-18 276
1.set介绍 set容器是一种实现了平衡二叉树的数据结构，容器中的数据不能重复，即每个数据都是唯一的，并且会对存进去的数据进行自动排序。 构造set集合的主要目的是为了快速检索，去重与排序，使用set前，需要在程序头文件中包含声明#include&amp;lt; set&amp;gt; 2.set常用函数 insert(): 插入元素 erase(): 删除元素 find(): 返...
【STL】容器 - set和map的使用与实现
Hello_World_213的博客
11-05 1305
STL - set与map的使用与实现(常用及重点接口详解)
「C++」STL —— Map
太初有泪的博客
08-13 271
文章目录1. 简介2.3. 基本操作函数4.5.6.7.8.9.10. 这个库需要学习一下。 1. 简介 Map是STL的一个关联容器，它提供 < Key - Value > 形式的数据处理能力，其内部自建一颗红黑树(一 种非严格意义上的平衡二叉树)，这颗树具有对数据自动排序的功能，所以在map内部所有的数据都是有序的。 2. 3. 基本操作函数 begin() 返回指向map头部的迭代器 clear(） 删除所有元素 coun
map和set的模拟实现
06-06
这将帮助我们更好地理解和实现红黑树，从而有效地模拟出Map和Set。 总之，通过封装红黑树并结合C++的面向对象特性，我们可以构建出高性能的Map和Set模拟实现。这不仅可以加深对数据结构和算法的理解，也有助于在...
C++ STL--数据结构与算法实现（余文溪）示例程序代码.rar
11-10
C++ STL，全称为Standard Template Library（标准模板库），是C++编程语言中的一部分，它提供了丰富的容器、迭代器、算法和函数对象等组件，极大地简化了数据结构和算法的实现。余文溪的《C++ STL--数据结构与算法...
set_STL_C++_set_
10-04
`set`可以与其他STL容器（如`vector`、`map`）结合使用，例如通过迭代器将`set`的元素复制到`vector`，或者将`set`的键作为`map`的键。 ### 6. 示例代码 在提供的`set.cpp`文件中，可能会包含一个展示`set`容器...
以STL::map实现引用表
01-15
以STL::map实现引用表，如有错误，欢迎指教！
C++STL vector list map set dqueue 等应用举例及PPT讲解示例,代码演示
06-28
在这个主题中，我们将深入探讨vector、list、map、set和deque这五个主要的STL容器，并通过具体的例子和PPT讲解来理解它们的应用。 1. **vector**：vector是动态数组，它可以方便地在任何位置插入和删除元素，但主要...
windbg导出stl map和set的插件
07-28
在处理复杂的系统问题时，尤其是对于C++程序中的STL（Standard Template Library，标准模板库）容器，如map和set，理解其内部结构和内容至关重要。这个插件“stlkit.dll”就是为了帮助开发者更便捷地在Windbg中查看...
c++ STL中的map
$好记性还是要多记录$
05-06 2412
1、map容器 map是C++ STL的一个关联容器，它提供一对一的数据处理能力。其中，各个键值对的键和值可以是任意数据类型，包括 C++ 基本数据类型（int、double 等）、使用结构体或类自定义的类型。 第一个可以称为关键字(key)； 第二个可能称为该关键字的值(value)； 该容器存储的都是 pair<const K, T> 类型（其中 K 和 T 分别表示键和值的数据类型）的键值对元素。 使用 map 容器存储的各个键值对，键的值既不能重复也不能被修改。换句话说，map 容
C++---STL标准库之map函数全解析，图文并茂，实例分析，迅速上手掌握！
livercy
09-17 8415
map()的定义 map 翻译为映射，也是常用的STL容器。众所周知，在定义数组时（如 int array[100]），其实是定义了一个从 int 型到 int型的映射，比如 array[0]=25、array[4]=36就分别是将0映射到 25、将4映射到36。一个double 型数组则是将 int 型映射到 double 型。 例如 db[0]=3.14，db[1]=0.01。但是，无论是什么类型，它总是将 int 型映射到其他类型。这似乎表现出一个弊端∶当需要以其他类型作为关键字来做映射时，会显得不太
面向初学者的 STL set 详解，一看就懂！
Linux猿
09-10 7308
???? 作者：Linux猿 ???? 简介：CSDN博客专家????，C/C++、面试、刷题、算法尽管咨询我，关注我，有问题私聊！ 目录 一、什么是 set ? 二、set 的定义 2.1 头文件 2.2 定义 2.3 常用方法 三、set 方法实例演示 3.1 size()、clear()、empty() 方法 3.2 begin()、end() 方法 3.3 rbegin()、rend() 方法 3.4 lower_bound()、upper_bound() 方法 3.5 e
【C++进阶】【STL】set和map的讲解及模拟实现
2301_79551553的博客
04-05 1093
1.了解了关联性容器和键值对的概念2.了解了以红黑树为基础的四大容器的作用3.重写了红黑树4.模拟了map和set，并分析了种种设置的原因赠人一赞，日行一善。
C++STL详解（五）map&set的使用及其模拟实现
Tau_Ceti的博客
09-03 710
map&set的使用及其模拟实现
数据结构 Java DS——链表部分经典题目 (1)
最新发布
cjejwe的博客
07-09 671
笔者计划在暑假啃完JavaDS,Mysql的内容当然也会继续更这次给读者们分享的是链表的几个比较典型的题目,关于如何手搓一个链表,笔者还在筹划中,毕竟链表的种类也有那么多,但是在下面的题目中,只有单向链表。
学习STL+MAP%2C+STL+SET之数据结构基础.
08-05
STL MAP, STL SET 是C++中非常重要的数据结构之一，它们提供了非常方便的操作接口，使得在C++中可以非常方便地实现对于有序和无序关联数据的存储和操作。本文通过列举一些基本的问题，以及对这些问题的解答，来讲解...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

霄沫凡

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

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

余额充值