【STL】模拟实现map和set {带头结点的红黑树;红黑树的核心结构;红黑树的迭代器;红黑树的插入和查找;map和set的封装}

模拟实现map和set

  • map和set是红黑树的两种不同封装形式,底层使用同一颗泛型结构的红黑树。
  • set是红黑树的K模型;map是红黑树的KV模型。

下面的代码和讲解着重体现红黑树的底层实现和map\set上层封装的衔接。关于二叉搜索树性质,map和set的介绍,搜索树的旋转,红黑树的性质等内容请依次阅读下面几篇文章:

  1. 【高阶数据结构】二叉搜索树 {概念;实现:核心结构,增删查,默认成员函数;应用:K模型和KV模型;性能分析;相关练习}
  2. 【STL】map和set的介绍和使用 {关联式容器;键值对;map和set;multimap和multiset;OJ练习}
  3. 【高阶数据结构】AVL树 {概念及实现;节点的定义;插入并调整平衡因子;旋转操作:左单旋,右单旋,左右双旋,右左双旋;AVL树的验证及性能分析}
  4. 【高阶数据结构】红黑树 {概念及性质;红黑树节点的定义;红黑树插入操作详细解释;红黑树的验证}

一、红黑树的核心结构

  • 问题一:map和set底层使用同一颗泛型结构的红黑树,如何处理map(pair<key,value>)和set(key)存储值不同的问题?

    解决方案:泛型底层红黑树的存储类型,通过不同的实例化参数,实现出map和set。

1.1 节点的定义

// RBTree.hpp
#pragma once
#include <utility>
#include <assert.h>
using std::pair;
using std::make_pair;enum Color{RED,BLACK
};template <class T>
struct RBTreeNode{typedef RBTreeNode<T> Node;Node *_left;Node *_right;Node *_parent;T _data; //泛型底层红黑树的存储类型,通过不同的实例化参数,实现出map和set。Color _color;RBTreeNode(const T &data = T(), Color color = RED):_left(nullptr),_right(nullptr),_parent(nullptr),_data(data),_color(color){}
};

1.2 红黑树的结构定义

STL中的红黑树结构

  • 为了后续实现关联式容器map/set,STL红黑树的实现中增加一个头结点;
  • 因为根节点必须为黑色,为了与根节点进行区分,将头结点给成红色;
  • 并且让头结点的_parent域指向红黑树的根节点,_left域指向红黑树中最小的节点,_right域指向红黑树中最大的节点。

在这里插入图片描述

头结点的作用(begin, end):

  • STL明确规定,begin()与end()代表的是一段前闭后开的区间,而对红黑树进行中序遍历后,可以得到一个有序的序列,因此:begin()可以放在红黑树中最小节点(即最左侧节点)的位置,end()放在最大节点(最右侧节点)的下一个位置,关键是最大节点的下一个位置在哪块?
  • 能否给成nullptr呢?答案是行不通的,因为对end()位置的迭代器进行–操作,必须要能找最后一个元素,此处就不行,因此最好的方式是将end()放在头结点的位置
// RBTree.hpp
// K: key的类型,
// T: 如果是map,则为pair<K, V>; 如果是set,则为K
// KofT: 通过T类型的data来获取key的一个仿函数类
template <class K, class T, class KofT>
class RBTree{typedef RBTreeNode<T> Node; //第二个模版参数T,决定红黑树的存储类型Node *_phead; //指向头结点的指针public:typedef RBT_iterator<T, T&, T*> iterator; //普通迭代器typedef RBT_iterator<T, const T&, const T*> const_iterator; //const迭代器//构造,创建头结点RBTree(){_phead = new Node(T(), RED); //为了和根节点区分,头节点设为红色_phead->_left = _phead->_right = _phead;}//普通对象返回普通迭代器iterator begin(){return iterator(_phead->_left); //begin放在红黑树最左节点的位置}iterator end(){return iterator(_phead); //end放在头结点的位置} //const对象返回const迭代器const_iterator begin() const{ return const_iterator(_phead->_left); }const_iterator end() const{return const_iterator(_phead);}//插入和查找pair<iterator, bool> Insert(const T& data);iterator Find(const K &k);private://获取根节点,注意返回指针的引用便于修改Node*& GetRoot(){return _phead->_parent;}//获取最左节点Node* LeftMost(){Node *proot = GetRoot();if(proot == nullptr){return _phead;}else{Node *left = proot;while(left->_left != nullptr){left = left->_left;}return left;}}//获取最右节点Node* RightMost(){Node *proot = GetRoot();if(proot == nullptr){return _phead;}else{Node *right = proot;while(right->_right != nullptr){right = right->_right;}return right;}}//旋转void RotateL(Node *parent);void RotateR(Node *parent);

二、红黑树的迭代器

2.1 迭代器的基本操作

红黑树的迭代器底层封装一个指向节点的指针,基本操作的详细解释请参照list迭代器的实现。

// RBTree.hpp
template<class T, class Ref, class Ptr>
class RBT_iterator{typedef RBT_iterator<T, Ref, Ptr> iterator;typedef RBTreeNode<T> Node;Node *_pnode; //红黑树的迭代器底层封装一个指向节点的指针public://基本操作请参照list迭代器的实现,不做过多解释RBT_iterator(Node *pnode):_pnode(pnode){}Ref operator*() const{ return _pnode->_data;}Ptr operator->() const{return &_pnode->_data;}bool operator==(const iterator &it) const{return _pnode == it._pnode;}bool operator!=(const iterator &it) const{return _pnode != it._pnode;}//......
};

2.2 operator++

红黑树迭代器的实现难点在于++和–操作:

在这里插入图片描述

operator++:(中序:左子树,根,右子树)

  1. 如果当前节点的右子树不为空,++就是找右子树中序的第一个节点(最左节点)。11->12。

  2. 如果当前节点的右子树为空,++就是找孩子不是右节点的那个祖先。7->8。

  3. 特殊情况:根节点没有右孩子,迭代器在根位置(如下图),此时进行++迭代器因该指向头结点end遍历结束。但如果按照上面的逻辑由于根节点恰好是头结点的右孩子(根节点没有右孩子),最终迭代器又会指向根节点永远无法到达头结点end,导致程序陷入死循环。因此需要特殊处理。

在这里插入图片描述

提示:右子树为空或孩子是右节点,说明这棵子树已经遍历访问完了。

  //前置和后置++,后置复用前置iterator& operator++(){if(_pnode->_right != nullptr) //如果右子树不为空,++就是找右子树的最左节点{Node *left = _pnode->_right; while(left->_left != nullptr){left = left->_left;}_pnode = left;}else //如果右子树为空,++就是找孩子不是右节点的那个祖先{Node *parent = _pnode->_parent;while(_pnode == parent->_right){_pnode = parent;parent = parent->_parent;}//特殊情况:根节点没有右孩子,迭代器在根位置。//经过循环此时_pnode指向头节点,parent指向根节点,做特殊处理使_pnode指向头结点if(_pnode->_right != parent)_pnode = parent;}return *this;}iterator operator++(int){iterator it(_pnode);++*this;return it;}

2.3 operator–

在这里插入图片描述

operator–:和++相反(右子树,根,左子树)

  1. 如果当前节点的左子树不为空,–就是找左子树的最右节点。8->7
  2. 如果当前节点的左子树为空,–就是找孩子不是左节点的那个祖先。12->11
  3. 特殊情况:如果迭代器指向end,即头结点。此时进行–操作因该使迭代器指向最右节点。
  4. –操作也存在++操作中的特殊情况——根节点没有左孩子。但不需要做特殊处理,–后仍指向根节点即可。

提示:左子树为空或孩子是左节点,说明这棵子树已经遍历访问完了。

  //前置和后置--,后置复用前置iterator operator--(){//特殊情况:如果迭代器指向end,进行--操作因该使迭代器指向最右节点。if(_pnode->_parent->_parent == _pnode && _pnode->_color == RED){_pnode = _pnode->_right;}else if(_pnode->_left != nullptr) //如果左子树不为空,--就是找左子树的最右节点{Node *right = _pnode->_left;while(right->_right != nullptr){right = right->_right;}_pnode = right;}else //如果左子树为空,--就是找孩子不是左节点的那个祖先{Node *parent = _pnode->_parent;while(_pnode == parent->_left){_pnode = parent;parent = parent->_parent;}_pnode = parent; //不需要特殊处理}return *this;}iterator operator--(int){iterator it(_pnode);--*this;return it;}

三、红黑树的插入和查找

  • 问题二:在进行查找、插入、删除等操作时,要对key值进行比较。在同一模版中,如何区别比较map和set中的key值?

    解决方案:通过传入仿函数KofT(KeyofTree)解决。如果是set,传入SetKofT返回data的值;如果是map,传入MapKofT返回data.first的值;

    注意:pair中重载了关系运算符,但first和second都参与运算,不符合要求。要求只比较pair.first (key)。

3.1 插入

pair<iterator, bool> Insert(const T& data){Node* &rproot = GetRoot(); //这里注意要用引用接收返回值if(rproot == nullptr){rproot = new Node(data, BLACK); //因为GetRoot返回指针的引用,所以改的实际是_phead->_parentrproot->_parent = _phead;_phead->_left = _phead->_right = rproot;//返回pair<iterator, bool>,方便实现operator[]。return make_pair(iterator(rproot), true); }KofT kot; //创建KofT对象,用于取出data中的keyNode *cur = rproot;Node *parent = nullptr;while(cur != nullptr){parent = cur;if(kot(data) > kot(cur->_data)) //不管是map还是set都能正确的取出key进行比较。cur = cur->_right;else if(kot(data) < kot(cur->_data))cur = cur->_left;elsereturn make_pair(iterator(cur), false);}cur = new Node(data, RED);if(kot(data) > kot(parent->_data)){parent->_right = cur;}else{parent->_left = cur;}cur->_parent = parent;Node* newnode = cur; //在调整红黑树的过程中cur的指向会改变,所以要提前记录新节点的指针。//上一次循环中grandparent 为根节点,此次循环parent == _phead//关于红黑树性质的检查和平衡调整请阅读文章:【高阶数据结构】红黑树while(parent != _phead && parent->_color == RED){Node* grandparent = parent->_parent;assert(grandparent!=nullptr);assert(grandparent->_color == BLACK);Node* uncle = grandparent->_left;if(grandparent->_left == parent)uncle = grandparent->_right;if(uncle != nullptr && uncle->_color == RED){parent->_color = uncle->_color = BLACK;grandparent->_color = RED;cur = grandparent;parent = grandparent->_parent;}else{if(parent == grandparent->_left){if(cur == parent->_left){RotateR(grandparent);parent->_color = BLACK;grandparent->_color = RED;}else{RotateL(parent);RotateR(grandparent);cur->_color = BLACK;grandparent->_color = RED;}}else{if(cur == parent->_right){RotateL(grandparent);parent->_color = BLACK;grandparent->_color = RED;}else{RotateR(parent);RotateL(grandparent);cur->_color = BLACK;grandparent->_color = RED;}}break;}} //end of while//如果在调整过程中将根节点变为红色,记得重新变回黑色。if(parent == _phead)cur->_color = BLACK;//令头节点的左指针指向红黑树的最左节点_phead->_left = LeftMost();//令头节点的右指针指向红黑树的最右节点_phead->_right = RightMost();//返回pair<iterator, bool>,方便实现operator[]。return make_pair(iterator(newnode), true);}

3.2 旋转

  void RotateL(Node *parent){Node* subR = parent->_right;Node* subRL = subR->_left;Node* ppNode = parent->_parent;subR->_left = parent;parent->_parent = subR;parent->_right = subRL;if(subRL != nullptr)subRL->_parent = parent;if(ppNode == _phead) //如果parent是根节点,要修改头结点的_parent指针。{_phead->_parent = subR;}else{if(parent == ppNode->_left)ppNode->_left = subR; elseppNode->_right = subR;}subR->_parent = ppNode;}void RotateR(Node *parent){Node *subL = parent->_left;Node *subLR = subL->_right;Node *ppNode = parent->_parent;subL->_right = parent;parent->_parent = subL;parent->_left = subLR;if(subLR != nullptr)subLR->_parent = parent;if(ppNode == _phead) //如果parent是根节点,要修改头结点的_parent指针。{_phead->_parent = subL;}else{if(parent == ppNode->_left)ppNode->_left = subL;elseppNode->_right = subL;}subL->_parent = ppNode;}

3.3 查找

  iterator Find(const K &k){KofT kot; //创建KofT对象,用于取出data中的keyNode *cur = GetRoot();if(cur == nullptr) return end(); //如果是空树,返回end。while(cur != nullptr){if(k > kot(cur->_data)) //不管是map还是set都能正确的取出key进行比较。cur = cur->_right;else if(k < kot(cur->_data))cur = cur->_left;elsereturn iterator(cur); //找到返回指向节点的迭代器}return end(); //如果找不到,返回end。}

四、map和set的封装

4.1 map

//Map.hpp
#pragma once
#include "RBTree.hpp"namespace zty{template <class K, class V>class map{//MapKofT返回kv.firststruct MapKofT{const K& operator()(const pair<K,V> &kv){return kv.first;}};//map是KV模型的红黑树,存储pair<K,V>键值对typedef RBTree<K, pair<K,V>, MapKofT> RBT;RBT _rbt;public://取类模版中的内嵌类型时,需要在类型前加typename;告诉编译器,后面这一串是类型不是静态成员。typedef typename RBT::iterator iterator; //map的迭代器类型typedef typename RBT::const_iterator const_iterator; //const迭代器//C++11的用法:使用了default关键字,表示使用编译器自动生成的默认构造函数map() = default;//迭代器区间构造    template <class InputIterator>map(InputIterator first, InputIterator last){while(first!=last){_rbt.Insert(*first);++first;}}iterator begin(){return _rbt.begin();}iterator end(){return _rbt.end();}const_iterator begin() const{return _rbt.begin();}const_iterator end() const{return _rbt.end();}std::pair<iterator, bool> insert(const pair<K,V> &kv){return _rbt.Insert(kv);}//Insert返回pair<iterator, bool>,方便实现operator[]。//具体内容阅读文章:【STL】map和set的介绍和使用 V& operator[](const K& k){pair<iterator, bool> ret = _rbt.Insert(make_pair(k, V()));return ret.first->second;}iterator find(const K& k){return _rbt.Find(k);}};
}

4.2 set

//Set.hpp
#pragma once
#include "RBTree.hpp"namespace zty{
template <class K>class set{//SetKofT返回kstruct SetKofT{const K& operator()(const K &k){return k;}};//set是K模型的红黑树,只存储key值typedef RBTree<K, K, SetKofT> RBT;RBT _rbt;public://取类模版中的内嵌类型时,需要在类型前加typename;告诉编译器,后面这一串是类型不是静态成员。typedef typename RBT::iterator iterator; //set的迭代器类型typedef typename RBT::const_iterator const_iterator; //const迭代器//C++11的用法:使用了default关键字,表示使用编译器自动生成的默认构造函数set() = default;//迭代器区间构造template <class InputIterator>set(InputIterator first, InputIterator last){while(first!=last){_rbt.Insert(*first);++first;}}iterator begin(){return _rbt.begin();} iterator end(){return _rbt.end();}const_iterator begin() const{return _rbt.begin();}const_iterator end() const{return _rbt.end();}std::pair<iterator, bool> insert(const K& k){return _rbt.insert(k);}iterator find(const K& k){return _rbt.Find(k);}};
}

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/77011.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

UMA 2 - Unity Multipurpose Avatar☀️四.UMA人物部位的默认颜色和自定义(共享)颜色

文章目录 🟥 人物颜色介绍1️⃣ 使用默认颜色2️⃣ 使用自定义颜色🟧 UMA自定义颜色的作用🟨 自定义颜色还可作为共享颜色🟥 人物颜色介绍 UMA不同部位的颜色分为默认的内置颜色和我们新定义的颜色. 1️⃣ 使用默认颜色 比如不勾选UseSharedColor时,使用的眼睛的默认…

品牌策划经理工作内容|工作职责|品牌策划经理做什么?

一位美国作家曾说过“品牌是一系列期望、记忆、故事和关系&#xff0c;他们共同构成了消费者最终原则一个产品或者服务的原因。” 所以&#xff0c;品牌经理这个岗位主要是创造感知价值主张&#xff0c;激发消费者购买这个品牌后带来的感知价值&#xff0c;这种回报的本质相对…

Git 基本操作【本地仓库与远程仓库的推送、克隆和拉取】

文章目录 一、Git简介二、Git的下载安装三、Git常规命令四、新建本地仓库五、本地分支操作六、Git远程仓库七、远程仓库克隆、抓取和拉取八、总结九、学习交流 一、Git简介 Git是分布式版本控制系统&#xff08;Distributed Version Control System&#xff0c;简称 DVCS&…

使用rpm重新安装包

#查询 rpm -qa | grep cloudstack #卸载 rpm -e cloudstack-agent-4.18.0.0-1.x86_64 #安装 rpm -ivh cloudstack-agent-4.18.0.0-1.x86_64.rpm

【PowerQuery】PowerQuery学习路径

PowerQuery这么好,怎么去学习呢?相信很多初读本书的朋友迫切的希望了解整个PowerQuery全景知识和它提供的相应的功能。但是对于PowerQuery来说,一开始就会进行自定义函数的构建当然也是不可能的,这里有相应的学习路径来进行由浅入深的学习,帮助读者更好的理解PowerQuery的…

leetcode 649. Dota2 参议院

2023.9.11 先简化一下题意&#xff1a;本题的意思就是每次投票中&#xff0c;前面的议员可以干掉后面的议员(当然是干掉敌对方的)&#xff0c;然后他将参与下一次的投票&#xff0c;而被干掉的议员则不能参与投票了。 如&#xff1a;[R D D] -> [R D] ->[R D]->[D] 。…

【算法基础】时间复杂度和空间复杂度

目录 1 算法的评价 2 算法复杂度 2.1 时间复杂度&#xff08;Time Complexity&#xff09; 2.1.1 如何计算时间复杂度&#xff1a; 2.1.2 常见的时间复杂度类别与示例 2.2 空间复杂度 2.2.1 如何计算空间复杂度 2.2.2 常见的空间复杂度与示例 3 时间复杂度和空间复杂度…

【附安装包】2023最新版Python安装详细教程!一键安装,永久使用

一、python官网 Python官网主要有python的About (简介)、Downloads (下载)、Documentation(文档)、Community (团体)、Success Stories (成功案例)、News (新闻)、Events (事件动态)等栏目。 Python官网地址&#xff1a;https://www.python.org/ 【领取方式见文末】 二、在…

免费的低代码助力售后工单管理:快速搭建,高效定制

编者按&#xff1a;本文旨在阐述免费且高效的低代码平台在实现售后工单管理系统方面的优势、功能及其作用。这些优势和功能对于提高企业的服务质量和效率具有重要的意义。 关键词&#xff1a;低代码平台、售后工单系统、私有化部署 1.售后工单系统有什么作用&#xff1f; 售后工…

day55:C++ day5,运算符重载剩余部分、静态成员、继承

#include <iostream> #include <cstring> #define pi 3.14 using namespace std;class Shape { protected:double round;double area; public://无参构造Shape():round(40),area(100){cout<<"Shape::无参构造函数&#xff0c;默认周长为40&#xff0c;面…

Python的命令行参数

Python的命令行参数&#xff0c;提供了很多有用的功能&#xff0c;可以方便调试和运行&#xff0c;通过man python就能查看&#xff0c;以下是一些常用参数使用实例和场景: 1. -B参数 在import时候&#xff0c;不产生pyc或者pyo文件: 比如有程序main.py如下: from Hello im…

软件测试/测试开发丨使用ChatGPT自动进行需求分析

简介 在实际工作过程中&#xff0c;常常需要拿到产品的PRD文档或者原型图进行需求分析&#xff0c;为产品的功能设计和优化提供建议。 而使用ChatGPT可以很好地帮助分析和整理用户需求。 实践演练 接下来&#xff0c;需要使用ChatGPT 辅助我们完成需求分析的任务 注意&…

数据链路层相关知识

数据链路层的作用 两个设备(同一种数据链路节点)之间进行传递数据 认识以太网 "以太网"是一种技术标准;既包含了数据链路层的内容,也包含了一些物理层的内容.例如:规定了网络拓扑结构,访问控制方式,传输速率等;以太网中的网线必须使用双绞线;传输速率有10M,100M,1…

Linux设备驱动——自动创建设备节点udev机制的实现过程

创建设备文件的机制有以下下列几种&#xff1a; mknod命令&#xff1a;手动创建设备节点的命令devfs:可以用于创建设备节点&#xff0c;创建设备节点的逻辑在内核空间&#xff08;内核2.4版本之前使用&#xff09;udev:自动创建设备节点的机制&#xff0c;创建设备节点的逻辑在…

python机器人编程——用python实现一个写字机器人

目录 一、前言二、整体框架2.1 系统构成2.2 硬件介绍2.2.1主要组成部分2.2.2机械结构2.2.3驱动及控制主板PS电机驱动原理简介: 2.2.4其余部分 2.3 机器人python程序框架2.3.1通信服务模块2.3.2消息处理模块2.3.3轨迹解析模块2.3.4机械臂逆解模块2.3.5写字板模块 三、机械臂的建…

0门槛限制!快来领取你的专属元宇宙虚拟展厅!

数字化时代中&#xff0c;元宇宙虚拟展厅仿佛成为了一种新的潮流&#xff0c;虚拟展厅的出现为我们呈现出了一个超越现实的全新世界。元宇宙虚拟展厅以其多样性、互动性、沉浸式展示为特点&#xff0c;同产品进行交互&#xff0c;创造出逼真的虚拟环境&#xff0c;为广大用户打…

八股——const 关键字

1.const作用 作用&#xff1a;const用于保护指针指向数据不被修改 测试代码1 显示数组的函数不小心修改了指针指向的值&#xff0c;这时候没有加const关键字&#xff0c;编译器不会报错 #include <stdio.h> void showar(int ar[]);int main(void) {int ar[4]{2,3,4,5…

Springboot后端跨域处理

跨域 当一台服务器资源从另一台服务器&#xff08;不同的域名或者端口&#xff09;请求一个资源或者接口&#xff0c;就会发起一个跨域HTTP请求。 同源&#xff1a;协议、域名、端口都相同 只要一个不同&#xff0c;就是跨域。 例子 请求方响应方是否跨域原因http://www.ba…

element-ui switch开关组件二次封装,添加loading效果,点击时调用接口后改变状态

先看效果&#xff1a; element-ui中的switch开关无loading属性&#xff08;在element-plus时加入了&#xff09;&#xff0c;而且点击时开关状态就会切换&#xff0c;这使得在需要调用接口后再改变开关状态变得比较麻烦。 思路&#xff1a;switch开关外包一层div&#xff0c;给…

LeetCode518. 零钱兑换 II 以及 动态规划相关的排列组合问题

文章目录 一、题目二、题解方法一&#xff1a;完全背包问题的变体&#xff08;版本1&#xff09;方法二&#xff1a;完全背包问题变体&#xff08;版本2&#xff09; 三、拓展&#xff1a;先遍历物品后遍历背包vs先遍历背包后遍历物品先遍历物品后遍历背包&#xff08;组合问题…