TreeMap源码解读,初识红黑树

TreeMap源码解读,初识红黑树

  • 一:红黑树简介
  • 二:TreeMap简介

一:红黑树简介

1、红黑树是一种特殊的二叉查找树。红黑树的每个结点上都有存储位表示结点的颜色,可以是红
(Red)或黑(Black)。

2、红黑树的每个结点是黑色或者红色。当是不管怎么样他的根结点是黑色。每个叶子结点(叶子结点
代表终结、结尾的节点)也是黑色 [注意:这里叶子结点,是指为空(NIL或NULL)的叶子结点!

3、如果一个结点是红色的,则它的子结点必须是黑色的。

4、每个结点到叶子结点NIL所经过的黑色结点的个数一样的。[确保没有一条路径会比其他路径长出俩
倍,所以红黑树是相对接近平衡的二叉树的!

5、红黑树的基本操作是添加、删除。在对红黑树进行添加或删除之后,都会用到旋转方法。为什么
呢?道理很简单,添加或删除红黑树中的结点之后,红黑树的结构就发生了变化,可能不满足上面
三条性质,也就不再是一颗红黑树了,而是一颗普通的树。而通过旋转和变色,可以使这颗树重新
成为红黑树。简单点说,旋转和变色的目的是让树保持红黑树的特性。
TreeMap源码解读,初识红黑树_第1张图片

二:TreeMap简介

1、TreeMap 是一个有序的 key-value 集合,它是通过红黑树实现的。
2、TreeMap基于红黑树(Red-Black tree)实现。该映射根据其键的自然顺序进行排序,或者根据创建映射时提供的 Comparator 进行排序,具体取决于使用的构造方法。
3、TreeMap是线程非同步的。

TreeMap<String,Integer> treeMap = new TreeMap<>(new Comparator<String>() {
            @Override
            public int compare(String o1, String o2) {
                return o1.length() - o2.length();
            }
        });

        treeMap.put("maliu",18);
        treeMap.put("zhangsan",25);
        treeMap.put("lisi",20);
        treeMap.put("wangwu",30);
        treeMap.put("zhaoliu",20);
        treeMap.put("sunqi",30);

显示结果:
TreeMap源码解读,初识红黑树_第2张图片
可以看到确实是按照key的长度由小到大进行排序

TreeMap源码解读,初识红黑树_第3张图片
TreeMap源码解读,初识红黑树_第4张图片
TreeMap的 put 方法

   public V put(K key, V value) {
        Entry<K,V> t = root;
        if (t == null) {
            compare(key, key); // type (and possibly null) check

            root = new Entry<>(key, value, null);
            size = 1;
            modCount++;
            return null;
        }
        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();
            @SuppressWarnings("unchecked")
                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<>(key, value, parent);
        if (cmp < 0)
            parent.left = e;
        else
            parent.right = e;

         //进行红黑树的一些标识吧。。。这一块还不是很清楚
        fixAfterInsertion(e);
        size++;
        modCount++;
        return null;
    }

1、添加第一个元素
TreeMap源码解读,初识红黑树_第5张图片
2、添加第二个元素

这个时候就要开始启用比较器进行比较排序了
TreeMap源码解读,初识红黑树_第6张图片
TreeMap源码解读,初识红黑树_第7张图片
所以如果在TreeMap中不传入比较器时显示的是这样的:
TreeMap源码解读,初识红黑树_第8张图片
其实他就是默认按照 String 类中的比交较器进行对比的
TreeMap源码解读,初识红黑树_第9张图片
比如说要加入 zhangsan
进行后的结果就是:
TreeMap源码解读,初识红黑树_第10张图片

然后就这样的流程依次的走下去:
最终会得到一个这样的红黑树。水平有限,红黑树的标识这里就先不研究了,以后弄懂之后再写
TreeMap源码解读,初识红黑树_第11张图片

然后对其进行中序遍历即可得到输出的结果

看到这里也许细心的同学已经发现了,sunqi 怎么不见了,我们往下再看

sunqi 和 maliu 的长度都为 5 ,经过比较器后结果为 0
TreeMap源码解读,初识红黑树_第12张图片
TreeMap源码解读,初识红黑树_第13张图片
TreeMap源码解读,初识红黑树_第14张图片
但是它的 value 确是替换了

你可能感兴趣的:(JDK源码学习记录,数据结构,TreeMap,java)