treemap源码解析之个人小笔记
2015-10-05 00:42
351 查看
添加entry时,核心函数fixAfterInsertion源码如下:
具体图示请分别参照http://www.cnblogs.com/fornever/archive/2011/12/02/2270692.html 中的插入和删除情形。
private void fixAfterInsertion(Entry<K,V> x) {
x.color = RED;
while (x != null && x != root && x.parent.color == RED) {
if (parentOf(x) == leftOf(parentOf(parentOf(x)))) {
Entry<K,V> y = rightOf(parentOf(parentOf(x)));
if (colorOf(y) == RED) {
//parent为red且parent的sibling为red,参照情形3
setColor(parentOf(x), BLACK);
setColor(y, BLACK);
setColor(parentOf(parentOf(x)), RED);
x = parentOf(parentOf(x));
} else {
//parent为red且parent的sibling为black,参照情形5
if (x == rightOf(parentOf(x))) {
x = parentOf(x);
rotateLeft(x);
}
//parent为red且parent的sibling为black,参照情形4
setColor(parentOf(x), BLACK);
setColor(parentOf(parentOf(x)), RED);
rotateRight(parentOf(parentOf(x)));
}
} else {
Entry<K,V> y = leftOf(parentOf(parentOf(x)));
if (colorOf(y) == RED) {
//parent为red且parent的sibling为red,和情形3对称
setColor(parentOf(x), BLACK);
setColor(y, BLACK);
setColor(parentOf(parentOf(x)), RED);
x = parentOf(parentOf(x));
} else {
//parent为red且parent的sibling为black,和情形5对称
if (x == leftOf(parentOf(x))) {
x = parentOf(x);
rotateRight(x);
}
setColor(parentOf(x), BLACK);
setColor(parentOf(parentOf(x)), RED);
rotateLeft(parentOf(parentOf(x)));
}
}
}
root.color = BLACK;
}
删除entry时,
private void fixAfterDeletion(Entry<K,V> x) {
while (x != root && colorOf(x) == BLACK) {
if (x == leftOf(parentOf(x))) {
Entry<K,V> sib = rightOf(parentOf(x));
if (colorOf(sib) == RED) {
//x为black且x的sibling为red,和情形1对称,x对应n
setColor(sib, BLACK);
setColor(parentOf(x), RED);
rotateLeft(parentOf(x));
sib = rightOf(parentOf(x));
}
if (colorOf(leftOf(sib)) == BLACK &&
colorOf(rightOf(sib)) == BLACK) {
//sib为之前x的兄弟,当sib的两个子元素均为黑时,在上步条件满足时则和情形3对应;若不满足上面条件,则和情形2对应
setColor(sib, RED);
x = parentOf(x);
} else {
if (colorOf(rightOf(sib)) == BLACK) {
//sib的左孩子为红,右孩子为黑,和情形5对应
setColor(leftOf(sib), BLACK);
setColor(sib, RED);
rotateRight(sib);
sib = rightOf(parentOf(x));
}
//sib的右孩子为红,和情形4对应;且上面情况必须导致这种情形。
setColor(sib, colorOf(parentOf(x)));
setColor(parentOf(x), BLACK);
setColor(rightOf(sib), BLACK);
rotateLeft(parentOf(x));
x = root;
}
} else { // symmetric
//下面情形和上面是左右对称的。
Entry<K,V> sib = leftOf(parentOf(x));
if (colorOf(sib) == RED) {
setColor(sib, BLACK);
setColor(parentOf(x), RED);
rotateRight(parentOf(x));
sib = leftOf(parentOf(x));
}
if (colorOf(rightOf(sib)) == BLACK &&
colorOf(leftOf(sib)) == BLACK) {
setColor(sib, RED);
x = parentOf(x);
} else {
if (colorOf(leftOf(sib)) == BLACK) {
setColor(rightOf(sib), BLACK);
setColor(sib, RED);
rotateLeft(sib);
sib = leftOf(parentOf(x));
}
setColor(sib, colorOf(parentOf(x)));
setColor(parentOf(x), BLACK);
setColor(leftOf(sib), BLACK);
rotateRight(parentOf(x));
x = root;
}
}
}
setColor(x, BLACK);
}
具体图示请分别参照http://www.cnblogs.com/fornever/archive/2011/12/02/2270692.html 中的插入和删除情形。
private void fixAfterInsertion(Entry<K,V> x) {
x.color = RED;
while (x != null && x != root && x.parent.color == RED) {
if (parentOf(x) == leftOf(parentOf(parentOf(x)))) {
Entry<K,V> y = rightOf(parentOf(parentOf(x)));
if (colorOf(y) == RED) {
//parent为red且parent的sibling为red,参照情形3
setColor(parentOf(x), BLACK);
setColor(y, BLACK);
setColor(parentOf(parentOf(x)), RED);
x = parentOf(parentOf(x));
} else {
//parent为red且parent的sibling为black,参照情形5
if (x == rightOf(parentOf(x))) {
x = parentOf(x);
rotateLeft(x);
}
//parent为red且parent的sibling为black,参照情形4
setColor(parentOf(x), BLACK);
setColor(parentOf(parentOf(x)), RED);
rotateRight(parentOf(parentOf(x)));
}
} else {
Entry<K,V> y = leftOf(parentOf(parentOf(x)));
if (colorOf(y) == RED) {
//parent为red且parent的sibling为red,和情形3对称
setColor(parentOf(x), BLACK);
setColor(y, BLACK);
setColor(parentOf(parentOf(x)), RED);
x = parentOf(parentOf(x));
} else {
//parent为red且parent的sibling为black,和情形5对称
if (x == leftOf(parentOf(x))) {
x = parentOf(x);
rotateRight(x);
}
setColor(parentOf(x), BLACK);
setColor(parentOf(parentOf(x)), RED);
rotateLeft(parentOf(parentOf(x)));
}
}
}
root.color = BLACK;
}
删除entry时,
private void fixAfterDeletion(Entry<K,V> x) {
while (x != root && colorOf(x) == BLACK) {
if (x == leftOf(parentOf(x))) {
Entry<K,V> sib = rightOf(parentOf(x));
if (colorOf(sib) == RED) {
//x为black且x的sibling为red,和情形1对称,x对应n
setColor(sib, BLACK);
setColor(parentOf(x), RED);
rotateLeft(parentOf(x));
sib = rightOf(parentOf(x));
}
if (colorOf(leftOf(sib)) == BLACK &&
colorOf(rightOf(sib)) == BLACK) {
//sib为之前x的兄弟,当sib的两个子元素均为黑时,在上步条件满足时则和情形3对应;若不满足上面条件,则和情形2对应
setColor(sib, RED);
x = parentOf(x);
} else {
if (colorOf(rightOf(sib)) == BLACK) {
//sib的左孩子为红,右孩子为黑,和情形5对应
setColor(leftOf(sib), BLACK);
setColor(sib, RED);
rotateRight(sib);
sib = rightOf(parentOf(x));
}
//sib的右孩子为红,和情形4对应;且上面情况必须导致这种情形。
setColor(sib, colorOf(parentOf(x)));
setColor(parentOf(x), BLACK);
setColor(rightOf(sib), BLACK);
rotateLeft(parentOf(x));
x = root;
}
} else { // symmetric
//下面情形和上面是左右对称的。
Entry<K,V> sib = leftOf(parentOf(x));
if (colorOf(sib) == RED) {
setColor(sib, BLACK);
setColor(parentOf(x), RED);
rotateRight(parentOf(x));
sib = leftOf(parentOf(x));
}
if (colorOf(rightOf(sib)) == BLACK &&
colorOf(leftOf(sib)) == BLACK) {
setColor(sib, RED);
x = parentOf(x);
} else {
if (colorOf(leftOf(sib)) == BLACK) {
setColor(rightOf(sib), BLACK);
setColor(sib, RED);
rotateLeft(sib);
sib = leftOf(parentOf(x));
}
setColor(sib, colorOf(parentOf(x)));
setColor(parentOf(x), BLACK);
setColor(leftOf(sib), BLACK);
rotateRight(parentOf(x));
x = root;
}
}
}
setColor(x, BLACK);
}
相关文章推荐
- TreeMap源码解析
- ROS学习笔记------ROS深度解析----- day 7 2019/3/16 帅某(Cartographer源码阅读(1):程序入口)
- Spring源码解析笔记3——自定义标签的解析
- Hadoop学习笔记14:./bin/hadoop namenode -format之源码解析
- TreeMap关键源码解析-红黑树操作
- 第50讲:Scala中Variance变化点及其在Spark中的应用源码解析学习笔记
- 从源码解析TreeMap
- [TensorFlow 学习笔记-08]tf.pad函数源码解析
- Laravel学习笔记之Session源码解析(上)
- ROS学习笔记------ROS深度解析----- day 8 2019/3/16 帅某(Cartographer源码阅读(2):Node和MapBuilder对象)
- JSON解析----个人学习笔记
- 安卓学习笔记---Android源码解析--Material Design之水波纹点击效果RippleEffect使用
- Java 集合系列12之 TreeMap详细介绍(源码解析)和使用示例
- JUC.Condition学习笔记[附详细源码解析]
- jQuery源码解析(架构与依赖模块)笔记一
- 阿里巴巴开发手册解析个人笔记(五)Mysql规约(一)建表规约
- Hive源码解析与开发实战笔记--讲师贾杰
- Java 集合系列12之 TreeMap源码解析
- Java TreeMap 源码解析
- Java【集合系列】-12- TreeMap详细介绍(源码解析)和使用示例