红黑二叉树:
a. 根节点是黑色的;
b. 红色节点的儿子节点是黑色的;
c. 任何一个节点到空节点的所有路径上必包含相同数据的黑色节点;
d. 叶子节点的子节点是黑色节点
假设一颗红黑树的黑色节点个数为R,那么这棵树的最短高度为R,最大长度为2R
所以h<2R,R的最大值为log(n+1),所以红黑树的高度<2log(n+1)
先看插入:第一步,先找到新节点在红黑二叉排序树中的位置,并插入:
- int cmp;
- Entry<K,V> parent;
- // split comparator and comparable paths
- Comparator<? super K> cpr = comparator;
- if (cpr != null) {
- do {
- parent = t;
- cmp = cpr.compare(key, t.key);
- if (cmp < 0)
- t = t.left;
- else if (cmp > 0)
- t = t.right;
- else
- return t.setValue(value);
- } while (t != null);
- }
- else {
- if (key == null)
- throw new NullPointerException();
- Comparable<? super K> k = (Comparable<? super K>) key;
- do {
- parent = t;
- cmp = k.compareTo(t.key);
- if (cmp < 0)
- t = t.left;
- else if (cmp > 0)
- t = t.right;
- else
- return t.setValue(value);
- } while (t != null);
- }
- Entry<K,V> e = new Entry<K,V>(key, value, parent);
- if (cmp < 0)
- parent.left = e;
- else
- parent.right = e;
第二步,因为二叉树中新增了一个节点,如果红黑树的平衡性受了影响,就需要进行调整:
新插入的节点先标成红色,如果该插入的节点不是根节点而且新节点的父节点是黑色的话,那么新加节点后,没有违法上述abcd四点,红黑二叉排序树就不需要再做任何调整。所以只有当新插入节点非root节点或者其对应的父节点是红色节点的时候才需要重新调整:
- while (x != null && x != root && x.parent.color == RED)
取到新节点父节点的兄弟节点y,如果兄弟节点是红色,那么就将x和y标记为黑色,x父节点和其兄弟节点y对应的父节点标记为红色,此时因为他们的父节点的父节点也有可能是红色的,所以需要将父节点赋值给x,重新对x向上进行调整:
- Entry<K,V> y = rightOf(parentOf(parentOf(x)));
- if (colorOf(y) == RED) {
- setColor(parentOf(x), BLACK);
- setColor(y, BLACK);
- setColor(parentOf(parentOf(x)), RED);
- x = parentOf(parentOf(x));
- }
如果此时x是左节点的话,就要将x父节点设置为黑色,x父节点的父节点设置为红色,然后对x父节点的父节点右旋即可,即:将x父节点的父节点极其孩子节点设置为x父节点的右子树,如果x是右节点的话,要先将x对应的父节点设置为x的左孩子,此时问题就和x是左节点是一样的了,只是此时的x已经是x的父节点而已,其代码如下
- else {
- if (x == rightOf(parentOf(x))) {//如果x是右节点的话,要先将x对应的父节点设置为x的左孩子
- x = parentOf(x);
- rotateLeft(x);
- }
- setColor(parentOf(x), BLACK);//x父节点设置为黑色
- setColor(parentOf(parentOf(x)), RED);//x父节点的父节点设置为红色
- rotateRight(parentOf(parentOf(x)));//x父节点的父节点右旋成为x父节点的右孩子
- }
如果新节点x对应的父节点是右孩子节点的话,其调整规则和上述是雷同的,在此不再累述,见详细代码:
- else {
- Entry<K,V> y = leftOf(parentOf(parentOf(x)));
- if (colorOf(y) == RED) {
- setColor(parentOf(x), BLACK);
- setColor(y, BLACK);
- setColor(parentOf(parentOf(x)), RED);
- x = parentOf(parentOf(x));
- } else {
- if (x == leftOf(parentOf(x))) {
- x = parentOf(x);
- rotateRight(x);
- }
- setColor(parentOf(x), BLACK);
- setColor(parentOf(parentOf(x)), RED);
- rotateLeft(parentOf(parentOf(x)));
- }
- }
- private void rotateRight(Entry<K,V> p) {
- if (p != null) {
- Entry<K,V> l = p.left;
- p.left = l.right;
- if (l.right != null) l.right.parent = p;
- l.parent = p.parent;
- if (p.parent == null)
- root = l;
- else if (p.parent.right == p)
- p.parent.right = l;
- else p.parent.left = l;
- l.right = p;
- p.parent = l;
- }
- }
效果就是将当前节点的设置为其子节点的右节点,并将子节点节点原来的右节点设置为当前节点的左子节点;
左旋转的效果就是将当前节点设置为右子节点的左节点,右子节点原来的左子节点设置为当前节点的右子节点,代码如下:
- private void rotateLeft(Entry<K,V> p) {
- if (p != null) {
- Entry<K,V> r = p.right;
- p.right = r.left;
- if (r.left != null)
- r.left.parent = p;
- r.parent = p.parent;
- if (p.parent == null)
- root = r;
- else if (p.parent.left == p)
- p.parent.left = r;
- else
- p.parent.right = r;
- r.left = p;
- p.parent = r;
- }
- }
TreeMap删除元素:
和新增元素一样,删除元素也可能会破坏二叉树的平衡性,所以也可能需要做调整,查找过程就不说了,看看删除的时候我们如何调整来维护其平衡性的:
假如当前要被删除节点有两个孩子节点,那么只需要将其后继节点查找出来,将其内容换成后继节点的内容,然后问题就转变成删除后继节点的问题.这里是找到当前节点的右子树中最小的那个节点的内容替代之,
- Entry<K,V> p = t.right;
- while (p.left != null)
- p = p.left;
- return p;
这样删除有两个孩子节点的节点问题就转换成删除带有一个子节点或者删除叶子节点的问题。接下来一段代码是:
- // Start fixup at replacement node, if it exists.
- Entry<K,V> replacement = (p.left != null ? p.left : p.right);
- replacement.parent = p.parent;
- if (p.parent == null)
- root = replacement;
- else if (p == p.parent.left)
- p.parent.left = replacement;
- else
- p.parent.right = replacement;
- while (x != root && colorOf(x) == BLACK) {
- if (x == leftOf(parentOf(x))) {//如果x是左子节点
- Entry<K,V> sib = rightOf(parentOf(x));
- if (colorOf(sib) == RED) {//如果其兄弟节点为红色
- setColor(sib, BLACK);//兄弟节点设为黑色
- setColor(parentOf(x), RED);//父节点设置为红色
- rotateLeft(parentOf(x));//父节点左旋
- sib = rightOf(parentOf(x));
- }
- if (colorOf(leftOf(sib)) == BLACK &&
- colorOf(rightOf(sib)) == BLACK) {//如果兄弟节点的两个儿子都是黑色
- setColor(sib, RED);//兄弟节点设置为红色
- x = parentOf(x);
- } else {
- if (colorOf(rightOf(sib)) == BLACK) {
- setColor(leftOf(sib), BLACK);
- setColor(sib, RED);
- rotateRight(sib);
- sib = rightOf(parentOf(x));
- }
- setColor(sib, colorOf(parentOf(x)));
- setColor(parentOf(x), BLACK);
- setColor(rightOf(sib), BLACK);
- rotateLeft(parentOf(x));
- x = root;
- }
- }
- setColor(x, BLACK);
另外,如果replacement值是空的,就说明被删除节点没有子节点,如果是黑色节点需要进行调整再删除,如果不是就直接删除即可
- 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;
- }
TreeMap的一个特殊用途是它可以实现数据的区间范围查询