Map源码解析之TreeMap

Map源码解析之HashMap
Map源码解析之HashMap红黑树
Map源码解析之HashMap补充：集合、迭代器、compute、merge、replace
Map源码解析之LinkedHashMap

前面几篇文章分别解析了HashMap和LinkedHashMap，这一片文章将解析Map的另一种数据结构–TreeMap。
TreeMap基于红黑树实现的，因此TreeMap是有序的（LinkedHashMap的有序是指插入顺序或访问顺序，TreeMap的有序是指大小顺序），且TreeMap没有comparator时节点key值不可为null（value值可为null）。
TreeMap比较大小的方式主要有两种：comparator和comparable。
（1）如果成员变量comparator不为null，使用Comparator#compare进行判断。
（2）如果成员变量comparator为null，且key值的类型为class T implement Comparable<T>，则可以使用其Comparable$compareTo方法进行大小判断。此时key值不能为null。

一、主要的成员变量

比较器，null值时采用key值的自然顺序：private final Comparator<? super K> comparator;
红黑树的根节点：private transient Entry<K,V> root;
节点数量：private transient int size = 0;
修改次数：private transient int modCount = 0;

二、主要的内部类

1. TreeMap.Entry类

TreeMap的节点类，包括key值、value值、左儿子、右儿子、父节点和节点颜色，节点颜色默认黑色。

static final class Entry<K,V> implements Map.Entry<K,V> {
	K key;
	V value;
    Entry<K,V> left;
    Entry<K,V> right;
    Entry<K,V> parent;
    boolean color = BLACK;
}

三、红黑树的修正

红黑树有4条性质
（1）每个节点要么是黑色，要么是红色。
（2）根节点是黑色。
（3）每个叶子节点都是黑色。（在这里，叶子结点指的是空节点NIL）。
（4）如果一个叶子节点是红色的，则其子节点必然为黑色。
（5）每个节点到该节点的子孙叶子节点（NIL节点）的路径上的黑色节点数量相同。
因此，在插入节点或删除节点后要对红黑树进行修正，保证其依旧是一颗红黑树。
红黑树的节点增加和删除分成两部分。

1. TreeMap#fixAfterInsertion方法

用于插入节点后对红黑树进行修正，使其依旧是一颗红黑树，类似于HashMap.TreeNode#balanceInsertion方法。

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) {
        		//对应场景（1）
                setColor(parentOf(x), BLACK);
                setColor(y, BLACK);
                setColor(parentOf(parentOf(x)), RED);
                x = parentOf(parentOf(x));
            } else {
                if (x == rightOf(parentOf(x))) {
        			//对应场景（2）
                    x = parentOf(x);
                    rotateLeft(x);
                }
        		//对应场景（3）
                setColor(parentOf(x), BLACK);
                setColor(parentOf(parentOf(x))