【C++进阶】 用红黑树封装map和set

已于 2024-02-27 18:44:57 修改
阅读量1k 收藏 17
点赞数 28
分类专栏： C++ 文章标签： c++ java 开发语言 小程序 运维 服务器 linux
于 2024-01-20 11:22:49 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/Weraphael/article/details/135656432
版权
本文详细解析了C++STL中set和map如何共用同一红黑树模板，展示了底层结构、插入操作改造、迭代器实现以及[]操作符的处理，揭示了模板参数对存储结构的影响。
摘要由CSDN通过智能技术生成

在这里插入图片描述

👦个人主页：@Weraphael
✍🏻作者简介：目前学习C++和算法
✈️专栏：C++航路
🐋 希望大家多多支持，咱一起进步！😁
如果文章对你有帮助的话
欢迎 评论💬 点赞👍🏻 收藏 📂 加关注✨

目录

一、set和map源码剖析

源码地址：点击跳转

我们首先可以观察到，在set和map中包含有如下的头文件

在这里插入图片描述

先剧透一下：我们知道，set是key模型，map是key-value模型，按道理来说应该用两颗不同的红黑树分别来封装key和key-value。但其实在源码实现中，它们使用的是同一个模板。

那么它是如何做到的呢？

我们首先可以打开stl_set.h头文件以及stl_map.h来剖析。

  • stl_set.h头文件

在以往的博客中讲过，看源码首先需要看它的成员变量

在这里插入图片描述

我们发现：set的底层好像是key-value结构（和我们一开始说的key模型不太一样），不同的是它们都是Key。接下来我们再来看看stl_map.h头文件

  • stl_map.h头文件

在这里插入图片描述

我们发现：map是key_value结构，但它的value竟然是个pair结构

也就是说：

  • set是<K,K>模型的红黑树
  • map是<K，pair>模型的红黑树

为什么STL要这样设计呢？由于 set和map的底层是红黑树，接下来我们再看看红黑树stl_tree.h的实现

  • stl_tree.h头文件

在这里插入图片描述

以上源代码的核心是link_type header，其类型是rb_tree_node*，也就是红黑树结点的指针。

接下来我们需要关心这颗树的Value。

在这里插入图片描述

大家发现没有 结点存储的是Value。

但是这个Value不是我们前面所说的key-val模型中的val。这里的Value对于map而言是pair，对于set而言是key。所以说，真正决定红黑树里面存储的是什么，是由第二个模板参数决定的，这也就为什么map和set可以共用一颗树。其实也不是，更严谨一点来说是共用同一个类模板（红黑树）。

我们现在再来观测一下这个红黑树结点里面有什么

在这里插入图片描述

我们发现：这里是通过一个继承关系来搞定的。派生类存储的是value，而基类存储的是三叉链的指针和颜色。

那么在这里我们似乎看到了，在这棵树里面，我们好像并不是很需要key-value结构中的key类型的模板参数，那么事实上是如此的吗？其实不是的，这个key还必须得传入，因为会有一些接口需要key。比如对于map而言，find需要通过key来查找。

并且还可以发现：map和set的接口本质都是调用红黑树的接口

在这里插入图片描述

二、改造红黑树

2.1 红黑树基本结构的改造

在前头说过：真正决定红黑树里面存储的是什么，是由第二个模板参数决定的。

【RBTree.h】

在这里插入图片描述

同时也可以写出map和set的基本结构

【set.h】

在这里插入图片描述

【map.h】

在这里插入图片描述

2.2 插入操作的改造

首先我们来分析插入的模板参数应该是什么？对于set就是key；对于map则是pair。那么模板参数应该是V。

那现在就会遇到一个非常棘手的问题：插入的位置代码不怎么好写。

在这里插入图片描述

原因是：假设是set，V在实例化后，data的类型此时是没有问题的；如果是map，V在实例化后，data是pair，此处应该比较pair中的first，而pair的重载和我们期望的不一样，它默认先比first，再比second

在这里插入图片描述

解决方法：写一个仿函数。首先在map和set里面写两个内部类，这个内部类里面是一个运算符重载，充当仿函数。对于map返回pair的first，对于set返回它本身，也就是key

  • 【set.h】

在这里插入图片描述

  • 【map.h】
    在这里插入图片描述

  • 【RBTree.h】

在这里插入图片描述

2.3 添加查找操作

【RBTree.h】

在这里插入图片描述

【map.h】

【set.h】

三、迭代器

首先来看看库里的set的迭代器是如何实现的

在这里插入图片描述

如上图所示，迭代器同样是依靠着红黑树里面的迭代器去实现的。那么我们可以去stl_tree.h文件看看

在这里插入图片描述

我们可以看到，这个迭代器与list的迭代器是比较相似的。

在这里插入图片描述

接下来可以简单看看迭代器里的操作

在这里插入图片描述

大家注意到没，这里的++和--操作和我们在模拟实现list的时候一样，封装了一个结点的指针，然后重载运算符。但是这是一颗红黑树啊，++和--往哪走？

在这里插入图片描述

首先可以想到：迭代器遍历的时候默认是升序，也就是中序遍历。那么 a.begin()返回的一定是树中的最小值。比如访问结点7。所以，如果右树为空，那么下一个访问的就是孩子是父亲的左孩子的祖先；假设记下来访问10，那么如果右树不为空，访问的是右树的最左结点。

在这里插入图片描述

当然了，--操作就是++反向操作，因为++走的是中序遍历（左子树，根，右子树），那么--就是右子树，根，左子树。

  • 左树不为空，则找左树的最右结点
  • 左树为空，则找孩子是父亲右的祖父结点

在这里插入图片描述

有了迭代器类，我们可以在红黑树层次去调用这个迭代器类了

在这里插入图片描述

然后我们依次到map和set层次去封装

  • 【set.h】

在这里插入图片描述

  • 【map.h】

在这里插入图片描述

但是以上代码还是不完美，不管是set和map，都可以修改容器里的数据，因为一旦修改了树里面的数据，那么就意味着这棵树已经乱了。可能不再是一个搜索二叉树了。

在这里插入图片描述

而在库里面是这样实现的：

在这里插入图片描述

  • 对于set，普通迭代器和const迭代器本质都是const迭代器。
  • 对于map，它的key应该不可以被修改，而value是可以被修改的（参考统计次数），这里是通过对pair的第一个参数first进行const限定的，从而锁死了第一个参数。

【RBTree.h】

在这里插入图片描述

当然了，咱们自定义类型的迭代器也得修改

在这里插入图片描述

然后我们来修改set中的迭代器函数。

在这里插入图片描述

【测试】

在这里插入图片描述

当测试时，就连普通迭代器都运行不起来了。我们可以参考库中是如何实现的

在这里插入图片描述

【修改set.h后】

在这里插入图片描述

然后发现可以打印出结果，这是为什么呢？

如果不在后面加const，那么_set就是一个普通对象，那么begin()返回的就是普通begin()，而普通的iterator已经被重命名为const_iterator，因此就会导致返回类型不匹配。

或者直接提供const版本的begin()和end()也可以，因为set的普通对象和const对象都是调用const迭代器（权限可以缩小和平移）

在这里插入图片描述

然后我们来处理map的迭代器问题

处理方式很简单：只需要让pair中的第一个参数给带上const即可。保证key不可以被修改，而value可以修改。

【map.h】

在这里插入图片描述

四、[] 操作符

在这里插入图片描述

map的[]操作主要是依靠insert操作实现的。所以还得将insert在进一步完善

首先分别修改map.h和set.h中insert返回值

在这里插入图片描述

在这里插入图片描述

【对RBTree.h修改】

除了修改函数的返回值，还需要返回指向新插入节点的那个迭代器

在这里插入图片描述

但是当我们运行的时候，报错了，我们发现是set中的pair的迭代器出问题了

在这里插入图片描述
在这里插入图片描述

这是因为：_set是一个普通对象，去红黑树调用返回的是普通的迭代器，而在set中，迭代器都是const迭代器，因此就会导致类型不匹配

我们可以参考库里面是怎么处理的

【stl_set.h】

在这里插入图片描述

我们可以尝试仿照它写

在这里插入图片描述

但是还是编译不通过

在这里插入图片描述
这时因为：

  1. 普通对象_set调用insert返回的是<iterator, bool>，这里迭代器是普通迭代器。
  2. 但是set中的迭代器为了防止key被修改，iterator和const_iterator都封装的是红黑树const_iterator。

因此想要从pair<iterator, bool>转换pair<const_iterator, bool>，会调用pair的构造函数：

在这里插入图片描述

正常情况下普通迭代器不能转化为const迭代器。

为了解决这种情况，我们在迭代器类内添加一个拷贝构造即可。

在这里插入图片描述

  • 当传进来的是普通迭代器的时候，Iterator是普通迭代器，这个函数相当于拷贝构造
  • 当传进来的是const迭代器的时候，Iterator依然是普通迭代器，此时该函数就相当于构造函数（普通迭代构造const迭代器）

那么最后这个[]操作就很容易实现了

在这里插入图片描述

五、本篇源代码以及stl相关源码

Gitte仓库：点击跳转

没有难学的知识~
关注
专栏目录
【C++】红黑树详解并封装实现map和set
love_guanghui的博客
03-18 3231
红黑树详解
普通平衡树的2种stl实现方法
AndrewLi的博客
08-11 1014
以luogu上普通平衡树一题为例 vector #include<bits/stdc++.h> using namespace std; #define loop(i,start,end) for(register int i=start;i<=end;++i) #define anti_loop(i,start,end) for(register int i=start;i&g...
STL之tree的实现详解
热门推荐
一个菜鸟的博客
03-17 1万+
1、红黑树介绍 关联容器都有一个key（键）和一个value（值）。当元素被插入到关联式容器中时，内部结构依照其键值的大小，以特定的规则将元素放到合适的位置（实现查找算法吧）。 一般关联时容器内部结构是一个平衡二叉树，用于在恶劣的环境下获得良好的搜寻效率。平衡二叉树的实现有：AVL-tree、RB-tree、AA-tree。用的最广的就是RB-tree，不论是在Nginx还是Linux内核任务...
【C++进阶】红黑树封装map,set（模拟实现）
weixin_64906519的博客
04-07 478
我们讲解完了map和set的主要的一些接口，如果大家想看完整代码，可以查看我的gitee仓库：链接:map和set的封装下一节我们讲解哈希结构，在此之前也希望大家可以好好理解红黑树的原理，这一部分也是C++上难度的部分，经过了这一部分，后面的C++内容就会想对容易一些。
【C++进阶】红黑树封装实现map和set
m0_67077469的博客
03-20 485
红黑树封装实现map和set
C++进阶：封装红黑树实现map和set
2301_81699364的博客
11-02 860
RBTree实现了泛型不知道T参数导致是K，还是pair，那么insert内部进行插入逻辑比较时，就没办法进行比较，因为pair的默认支持的是key和value一起参与比较，我们需要时的任何时候只比较key，所以我们在map和set层分别实现一个MapKeyOfT和SetKeyOfT的仿函数传给 RBTree的KeyOfT，然后RBTree中通过KeyOfT仿函数取出T类型对象中的key，再进行比较，具体细节参考如下代码实现。如果右子树为空，就需要往上找了，找的是孩子是父亲左的那个祖先。
【C++】红黑树封装map和set
qq_67582098的博客
07-09 954
本文介绍了使用红黑树封装map和set
C++进阶 红黑树封装map和set
小萌新的博客
01-16 921
C++进阶 红黑树封装map和set
【C++】用红黑树封装set和map
最新发布
春风不语。
11-13 292
本篇内容到这里就结束啦，主要讲了如何用红黑树来封装set和map容器，希望对大家有帮助，祝生活愉快！
C++ 改造红黑树,封装map和set
Wzs040810的博客
03-15 1591
C++ 改造红黑树,封装map和set
C++红黑树封装实现map和set模拟
本资源将通过封装红黑树来模拟实现map和set。红黑树是一种自平衡的二叉搜索树，通过对插入和删除操作的特定处理，它可以保证最坏情况下的时间复杂度为O(log n)，这对于需要高效搜索和插入操作的场景非常有用。 在...
【C++】map和set的封装（红黑树）
qq_66314292的博客
02-27 1392
map和set封装
【C++ STL】模拟实现 map 和 set（对一颗红黑树进行封装）
codewinter的博客
06-25 758
阅读STL源码就可以发现，其实 set 和 map 自己没有实现啥东西，整体就是对红黑树进行了一个封装，set 和 map 的 insert / erase 接口底层就是调用红黑树的 insert / erase 接口，并且 set 和 map 的迭代器也都是取的红黑树里面的迭代器那 set 和 map 的核心区别在哪里呢？set 底层封装了一颗红黑树，是 KV 结构，值得注意的是 key_type 和 value_type 都是 key，说明传了两个 key 给红黑树。 map 底层封装了一颗红黑树，是
C++红黑树
zhao19971014的博客
11-09 647
红黑树，是一种二叉搜索树，但在每个结点上增加一个存储位表示结点的颜色，可以是Red或Black。 通过对任何一条从根到叶子的路径上各个结点着色方式的限制，红黑树确保没有一条路径会比其他路径长出俩倍，因而是接近平衡的
编译器、IDE对C/C++新标准的支持
编程语言技巧经验分享
11-08 1151
很遗憾的是，微软VC++对C99的支持直到VS2015才基本完全支持，在VS2012之前，仅仅支持C89. 也正是2015年之前，这么多人骂微软并纷纷转移到其它C语言编译器的原因。VS2019开始才正式宣布支持C11/C17.对于C++标准，VS2013开始支持C++11，VS2017支持C++17, VS2019支持C++20, 显然，微软更偏爱C++.
STL set
hls0611的博客
11-09 246
练习题：P1059 [NOIP2006 普及组] 明明的随机数。
C 学习（4）
search7的博客
11-05 884
C语言学习，每个模块来自工作中的？？？
C++_用 using 语句继承基类的构造函数
学海无涯
11-09 255
**默认情况下**，派生类会继承基类的所有成员，包括数据成员和函数成员，但不会继承基类的**构造函数和析构函数**，因为构造函数和析构函数的作用是创建和销毁对象，派生类通常需要有自己的构造和析构的逻辑。C++11 引入了using语句，使得派生类可以通过 using 关键字使用基类的构造函数，更方便地复用基类的构造逻辑。
GitLab 提交 C++ 技巧
nbspzs的专栏
11-08 377
在 GitLab 中提交 C++ 代码时，通常应只提交 源代码文件（例如 .cpp、.h）和 CMake 配置文件（例如 CMakeLists.txt），而不要提交通过 CMake 编译后的中间文件或构建产物。这样，当其他开发者克隆你的项目后，他们可以在自己的机器上使用 cmake 构建项目，而不需要下载编译产物。2. 跨平台兼容性：不同的开发者和平台可能会生成不同的编译产物，提交这些文件会造成不必要的冲突和问题。1. 在 .gitignore 文件中忽略编译生成的文件和目录。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值