死磕 java集合之TreeMap源码分析(三)- 内含红黑树分析全过程

2019独角兽企业重金招聘Python工程师标准>>> hot3.png

欢迎关注我的公众号“彤哥读源码”,查看更多源码系列文章, 与彤哥一起畅游源码的海洋。


删除元素

删除元素本身比较简单,就是采用二叉树的删除规则。

(1)如果删除的位置有两个叶子节点,则从其右子树中取最小的元素放到删除的位置,然后把删除位置移到替代元素的位置,进入下一步。

(2)如果删除的位置只有一个叶子节点(有可能是经过第一步转换后的删除位置),则把那个叶子节点作为替代元素,放到删除的位置,然后把这个叶子节点删除。

(3)如果删除的位置没有叶子节点,则直接把这个删除位置的元素删除即可。

(4)针对红黑树,如果删除位置是黑色节点,还需要做再平衡。

(5)如果有替代元素,则以替代元素作为当前节点进入再平衡。

(6)如果没有替代元素,则以删除的位置的元素作为当前节点进入再平衡,平衡之后再删除这个节点。

public V remove(Object key) {// 获取节点Entry<K,V> p = getEntry(key);if (p == null)return null;V oldValue = p.value;// 删除节点deleteEntry(p);// 返回删除的valuereturn oldValue;
}private void deleteEntry(Entry<K,V> p) {// 修改次数加1modCount++;// 元素个数减1size--;if (p.left != null && p.right != null) {// 如果当前节点既有左子节点,又有右子节点// 取其右子树中最小的节点Entry<K,V> s = successor(p);// 用右子树中最小节点的值替换当前节点的值p.key = s.key;p.value = s.value;// 把右子树中最小节点设为当前节点p = s;// 这种情况实际上并没有删除p节点,而是把p节点的值改了,实际删除的是p的后继节点}// 如果原来的当前节点(p)有2个子节点,则当前节点已经变成原来p的右子树中的最小节点了,也就是说其没有左子节点了// 到这一步,p肯定只有一个子节点了// 如果当前节点有子节点,则用子节点替换当前节点Entry<K,V> replacement = (p.left != null ? p.left : p.right);if (replacement != null) {// 把替换节点直接放到当前节点的位置上(相当于删除了p,并把替换节点移动过来了)replacement.parent = p.parent;if (p.parent == null)root = replacement;else if (p == p.parent.left)p.parent.left  = replacement;elsep.parent.right = replacement;// 将p的各项属性都设为空p.left = p.right = p.parent = null;// 如果p是黑节点,则需要再平衡if (p.color == BLACK)fixAfterDeletion(replacement);} else if (p.parent == null) {// 如果当前节点就是根节点,则直接将根节点设为空即可root = null;} else {// 如果当前节点没有子节点且其为黑节点,则把自己当作虚拟的替换节点进行再平衡if (p.color == BLACK)fixAfterDeletion(p);// 平衡完成后删除当前节点(与父节点断绝关系)if (p.parent != null) {if (p == p.parent.left)p.parent.left = null;else if (p == p.parent.right)p.parent.right = null;p.parent = null;}}
}

删除再平衡

经过上面的处理,真正删除的肯定是黑色节点才会进入到再平衡阶段。

因为删除的是黑色节点,导致整颗树不平衡了,所以这里我们假设把删除的黑色赋予当前节点,这样当前节点除了它自已的颜色还多了一个黑色,那么:

(1)如果当前节点是根节点,则直接涂黑即可,不需要再平衡;

(2)如果当前节点是红+黑节点,则直接涂黑即可,不需要平衡;

(3)如果当前节点是黑+黑节点,则我们只要通过旋转把这个多出来的黑色不断的向上传递到一个红色节点即可,这又可能会出现以下四种情况:

(假设当前节点为父节点的左子节点)

情况策略
1)x是黑+黑节点,x的兄弟是红节点(1)将兄弟节点设为黑色;<br>(2)将父节点设为红色;<br>(3)以父节点为支点进行左旋;<br>(4)重新设置x的兄弟节点,进入下一步;
2)x是黑+黑节点,x的兄弟是黑节点,且兄弟节点的两个子节点都是黑色(1)将兄弟节点设置为红色;<br>(2)将x的父节点作为新的当前节点,进入下一次循环;
3)x是黑+黑节点,x的兄弟是黑节点,且兄弟节点的右子节点为黑色,左子节点为红色(1)将兄弟节点的左子节点设为黑色;<br>(2)将兄弟节点设为红色;<br>(3)以兄弟节点为支点进行右旋;<br>(4)重新设置x的兄弟节点,进入下一步;
3)x是黑+黑节点,x的兄弟是黑节点,且兄弟节点的右子节点为红色,左子节点任意颜色(1)将兄弟节点的颜色设为父节点的颜色;<br>(2)将父节点设为黑色;<br>(3)将兄弟节点的右子节点设为黑色;<br>(4)以父节点为支点进行左旋;<br>(5)将root作为新的当前节点(退出循环);

(假设当前节点为父节点的右子节点,正好反过来)

情况策略
1)x是黑+黑节点,x的兄弟是红节点(1)将兄弟节点设为黑色;<br>(2)将父节点设为红色;<br>(3)以父节点为支点进行右旋;<br>(4)重新设置x的兄弟节点,进入下一步;
2)x是黑+黑节点,x的兄弟是黑节点,且兄弟节点的两个子节点都是黑色(1)将兄弟节点设置为红色;<br>(2)将x的父节点作为新的当前节点,进入下一次循环;
3)x是黑+黑节点,x的兄弟是黑节点,且兄弟节点的左子节点为黑色,右子节点为红色(1)将兄弟节点的右子节点设为黑色;<br>(2)将兄弟节点设为红色;<br>(3)以兄弟节点为支点进行左旋;<br>(4)重新设置x的兄弟节点,进入下一步;
3)x是黑+黑节点,x的兄弟是黑节点,且兄弟节点的左子节点为红色,右子节点任意颜色(1)将兄弟节点的颜色设为父节点的颜色;<br>(2)将父节点设为黑色;<br>(3)将兄弟节点的左子节点设为黑色;<br>(4)以父节点为支点进行右旋;<br>(5)将root作为新的当前节点(退出循环);

让我们来看看TreeMap中的实现:

/*** 删除再平衡*(1)每个节点或者是黑色,或者是红色。*(2)根节点是黑色。*(3)每个叶子节点(NIL)是黑色。(注意:这里叶子节点,是指为空(NIL或NULL)的叶子节点!)*(4)如果一个节点是红色的,则它的子节点必须是黑色的。*(5)从一个节点到该节点的子孙节点的所有路径上包含相同数目的黑节点。*/
private void fixAfterDeletion(Entry<K,V> x) {// 只有当前节点不是根节点且当前节点是黑色时才进入循环while (x != root && colorOf(x) == BLACK) {if (x == leftOf(parentOf(x))) {// 如果当前节点是其父节点的左子节点// sib是当前节点的兄弟节点Entry<K,V> sib = rightOf(parentOf(x));// 情况1)如果兄弟节点是红色if (colorOf(sib) == RED) {// (1)将兄弟节点设为黑色setColor(sib, BLACK);// (2)将父节点设为红色setColor(parentOf(x), RED);// (3)以父节点为支点进行左旋rotateLeft(parentOf(x));// (4)重新设置x的兄弟节点,进入下一步sib = rightOf(parentOf(x));}if (colorOf(leftOf(sib))  == BLACK &&colorOf(rightOf(sib)) == BLACK) {// 情况2)如果兄弟节点的两个子节点都是黑色// (1)将兄弟节点设置为红色setColor(sib, RED);// (2)将x的父节点作为新的当前节点,进入下一次循环x = parentOf(x);} else {if (colorOf(rightOf(sib)) == BLACK) {// 情况3)如果兄弟节点的右子节点为黑色// (1)将兄弟节点的左子节点设为黑色setColor(leftOf(sib), BLACK);// (2)将兄弟节点设为红色setColor(sib, RED);// (3)以兄弟节点为支点进行右旋rotateRight(sib);// (4)重新设置x的兄弟节点sib = rightOf(parentOf(x));}// 情况4)// (1)将兄弟节点的颜色设为父节点的颜色setColor(sib, colorOf(parentOf(x)));// (2)将父节点设为黑色setColor(parentOf(x), BLACK);// (3)将兄弟节点的右子节点设为黑色setColor(rightOf(sib), BLACK);// (4)以父节点为支点进行左旋rotateLeft(parentOf(x));// (5)将root作为新的当前节点(退出循环)x = root;}} else { // symmetric// 如果当前节点是其父节点的右子节点// sib是当前节点的兄弟节点Entry<K,V> sib = leftOf(parentOf(x));// 情况1)如果兄弟节点是红色if (colorOf(sib) == RED) {// (1)将兄弟节点设为黑色setColor(sib, BLACK);// (2)将父节点设为红色setColor(parentOf(x), RED);// (3)以父节点为支点进行右旋rotateRight(parentOf(x));// (4)重新设置x的兄弟节点sib = leftOf(parentOf(x));}if (colorOf(rightOf(sib)) == BLACK &&colorOf(leftOf(sib)) == BLACK) {// 情况2)如果兄弟节点的两个子节点都是黑色// (1)将兄弟节点设置为红色setColor(sib, RED);// (2)将x的父节点作为新的当前节点,进入下一次循环x = parentOf(x);} else {if (colorOf(leftOf(sib)) == BLACK) {// 情况3)如果兄弟节点的左子节点为黑色// (1)将兄弟节点的右子节点设为黑色setColor(rightOf(sib), BLACK);// (2)将兄弟节点设为红色setColor(sib, RED);// (3)以兄弟节点为支点进行左旋rotateLeft(sib);// (4)重新设置x的兄弟节点sib = leftOf(parentOf(x));}// 情况4)// (1)将兄弟节点的颜色设为父节点的颜色setColor(sib, colorOf(parentOf(x)));// (2)将父节点设为黑色setColor(parentOf(x), BLACK);// (3)将兄弟节点的左子节点设为黑色setColor(leftOf(sib), BLACK);// (4)以父节点为支点进行右旋rotateRight(parentOf(x));// (5)将root作为新的当前节点(退出循环)x = root;}}}// 退出条件为多出来的黑色向上传递到了根节点或者红节点// 则将x设为黑色即可满足红黑树规则setColor(x, BLACK);
}

删除元素举例

假设我们有下面这样一颗红黑树。

treemap-delete1

我们删除6号元素,则从右子树中找到了最小元素7,7又没有子节点了,所以把7作为当前节点进行再平衡。

我们看到7是黑节点,且其兄弟为黑节点,且其兄弟的两个子节点都是红色,满足情况4),平衡之后如下图所示。

treemap-delete2

我们再删除7号元素,则从右子树中找到了最小元素8,8有子节点且为黑色,所以8的子节点9是替代节点,以9为当前节点进行再平衡。

我们发现9是红节点,则直接把它涂成黑色即满足了红黑树的特性,不需要再过多的平衡了。

treemap-delete3

这次我们来个狠的,把根节点删除,从右子树中找到了最小的元素5,5没有子节点,所以把5作为当前节点进行再平衡。

我们看到5是黑节点,且其兄弟为红色,符合情况1),平衡之后如下图所示,然后进入情况2)。

treemap-delete4

对情况2)进行再平衡后如下图所示。

treemap-delete5

然后进入下一次循环,发现不符合循环条件了,直接把x涂为黑色即可,退出这个方法之后会把旧x删除掉(见deleteEntry()方法),最后的结果就是下面这样。

treemap-delete6


未完待续,下一节我们一起探讨红黑树遍历元素的操作。

现在公众号文章没办法留言了,如果有什么疑问或者建议请直接在公众号给我留言。


欢迎关注我的公众号“彤哥读源码”,查看更多源码系列文章, 与彤哥一起畅游源码的海洋。

qrcode

转载于:https://my.oschina.net/u/4108008/blog/3032739

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

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

相关文章

Linux:进程实例信息(/proc)

https://blog.csdn.net/test1280/article/details/73632333 Linux:进程实例信息&#xff08;/proc&#xff09; 问几个问题&#xff1a; 1.怎么知道一个进程对应哪个可执行文件&#xff1f; 2.怎么知道一个进程的资源限制&#xff1f; 3.怎么知道一个进程所处的环境&#xff1f…

四元素理解

旋转变换_四元数 2017年03月29日 11:59:38 csxiaoshui 阅读数&#xff1a;5686 1.简介 四元数是另一种描述三维旋转的方式&#xff0c;四元数使用4个分量来描述旋转&#xff0c;四元数的描述方式如下&#xff1a; qsxiyjzk,(s,x,y,z∈ℝ&#xff09;i2j2k2ijk−1 四元数的由…

31、SAM文件中flag含义解释工具--转载

转载&#xff1a;http://www.cnblogs.com/nkwy2012/p/6362996.html SAM是Sequence Alignment/Map 的缩写。像bwa等软件序列比对结果都会输出这样的文件。samtools网站上有专门的文档介绍SAM文件。具体地址&#xff1a;http://samtools.sourceforge.net/SAM1.pdf很多人困惑SAM文…

《Head First设计模式》批注系列(一)——观察者设计模式

最近在读《Head First设计模式》一书&#xff0c;此系列会引用源书内容&#xff0c;但文章内容会更加直接&#xff0c;以及加入一些自己的理解。 观察者模式&#xff08;有时又被称为模型-视图&#xff08;View&#xff09;模式、源-收听者(Listener)模式或从属者模式&#xff…

PYPL 4 月排行:Python 最流行,Java 还行不行?

开发四年只会写业务代码&#xff0c;分布式高并发都不会还做程序员&#xff1f; PYPL 发布了 4 月份的编程语言排行榜。 前五的分别是&#xff1a;Python、Java、Javascript、C# 和 PHP。可以看到&#xff0c;榜单没有什么大变化&#xff0c;但是相比去年 4 月份&#xff0c;…

两个向量的旋转矩阵与四元素

两向量的夹角 2017年06月20日 17:38:11 csxiaoshui 阅读数&#xff1a;36764 怎么计算两个向量间的夹角呢&#xff1f; 这里主要分两种情况&#xff0c;对于二维向量和三维向量来分别讨论。 1. 二维向量 二维向量的情况相对简单&#xff0c;根据向量间的点乘关系 v1⋅v2|…

顺序表

一、数据是如何在内存中存储的&#xff1f; 32位系统中char&#xff0c;int型数据在内存中的存储方式&#xff1a; char占1byte&#xff08;8bit&#xff09;int占4byte&#xff08;32bit&#xff09;假设我们有一个int类型的值&#xff0c;它从0x01开始&#xff0c;一个int占据…

Establishing SSL connection without server's identity verification is not recommended.

完全描述:Establishing SSL connection without servers identity verification is not recommended. According to MySQL 5.5.45, 5.6.26 and 5.7.6 requirements SSL connection must be established by default if explicit option isnt set. For compliance with existing …

四元素的真面目..........简单粗暴

作者&#xff1a;Yang Eninala 链接&#xff1a;https://www.zhihu.com/question/23005815/answer/33971127 来源&#xff1a;知乎 著作权归作者所有。商业转载请联系作者获得授权&#xff0c;非商业转载请注明出处。 根据我的理解&#xff0c;大多数人用汉密尔顿四元数就只…

2.自定义变量调节器

① 使用registerPlugin()方法来扩充变量调节器 该方法接收3个参数 1. 字符串modifier 2. 插件函数的名字 3. PHP回调函数 示例&#xff1a;自定义一个变量调节器&#xff0c;可以改变文字的颜色和大小 第一步&#xff1a;调用smarty对象的registerPlugin&#xff08;&#x…

SpringBoot2构建基于RBAC权限模型的驾校代理小程序后端

本项目是使用SpringBoot2构建的一套基于RBAC权限模型的后台管理系统&#xff0c;前端是微信小程序。 项目地址: github.com/fuyunwang/D… 项目的缘由 最近接了个外包,主要是针对于驾校开发一个代理小程序。目的是为了方便驾校的管理来招揽学员,同时方便维护学员和代理信息。 项…

while read line的问题

循环中的重定向或许你应该在其他脚本中见过下面的这种写法&#xff1a;while read linedo…done < file刚开始看到这种结构时&#xff0c;很难理解< file是如何与循环配合在一起工作的。因为循环内有很多条命令&#xff0c;而我们之前接触的重定向都是为一条命令工作的。…

Linemod;理解

Linemod 代码笔记 2019年03月11日 16:18:30 haithink 阅读数&#xff1a;197 最近了解到 Linemod 这个模板匹配算法&#xff0c;印象不错 准备仔细学习一下&#xff0c;先做点代码笔记&#xff0c;免得后面不好回顾 目前的笔记基本上把 核心流程都分析得比较清楚了&#xff0…

Swift3中数组创建方法

转载自&#xff1a;http://blog.csdn.net/bwf_erg/article/details/70858865 数组是由一组类型相同的元素构成的有序数据集合。数组中的集合元素是有 序的&#xff0c;而且可以重复出现。 1 数组创建 在Swift语言中&#xff0c;数组的类型格式为&#xff1a; Array<ElementT…

BZOJ 5249: [2018多省省队联测]IIIDX(贪心 + 线段树)

题意 这一天&#xff0c;\(\mathrm{Konano}\) 接到了一个任务&#xff0c;他需要给正在制作中的游戏 \(\mathrm{《IIIDX》}\) 安排曲目 的解锁顺序。游戏内共有\(n\) 首曲目&#xff0c;每首曲目都会有一个难度 \(d\) &#xff0c;游戏内第 \(i\) 首曲目会在玩家 Pass 第 \(\lf…

手眼标定

Eye-in-hand和Eye-to-hand问题求解和实验 2018年12月07日 00:00:40 百川木易 阅读数 3018 2018/12/5 By Yang Yang&#xff08;yangyangipp.ac.cn&#xff09; 本文所有源码和仿真场景文件全部公开&#xff0c;点击Gitee仓库链接。 文章目录 问题描述Eye-in-hand问题求解公式…

RNN总结

RNN既可以表述为循环神 经网络&#xff08;recurrent neural network&#xff09;&#xff0c;也可以表述为递归神经网络&#xff08;recursive neural network&#xff09;&#xff0c;前者一般用于处理以时间序列为输入的问题&#xff08;比如把一个句子看成词组成的序列&…

Problem 2. number题解

number&#xff1a;数学二分图匹配 首先&#xff0c;如果S<N,那么S1&#xff0c;S2...N这些数直接放在S1,S2...N的位置上(如果其他数x放在这些位置上面&#xff0c;这些数不放在对应位置&#xff0c;那么x一定能放在这些数放的位置&#xff0c;所以直接交换即可)所以可以直接…

SSD列子

一、介绍 本博文主要介绍实现通过SSD物体检测方式实现工件裂纹检测。裂纹图像如下所示&#xff1a; 二、关于SSD算法 具体算法不再阐述&#xff0c;详细请参考&#xff1a; https://blog.csdn.net/u013989576/article/details/73439202 https://blog.csdn.net/xiaohu2022/arti…

linux硬链接与软链接

Linux 系统中有软链接和硬链接两种特殊的“文件”。 软链接可以看作是Windows中的快捷方式&#xff0c;可以让你快速链接到目标档案或目录。 硬链接则透过文件系统的inode来产生新档名&#xff0c;而不是产生新档案。 创建方法都很简单&#xff1a; 软链接&#xff08;符号链接…