由 java 8 HashMap 中的红黑树，引发出对2叉树、 B树、B+树的分析

最新推荐文章于 2022-08-03 11:57:33 发布

weixin_34310127

最新推荐文章于 2022-08-03 11:57:33 发布

阅读量249

点赞数

文章标签： java 数据结构与算法 python

原文链接：https://my.oschina.net/ovirtKg/blog/758858

版权

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

前些天在网上偶然间看到一篇关于java 8的 HashMap的分析文章，其对java7进行了大量改进，核心是引进了红黑树，提高了hashcode碰撞严重时的查找性能。这一点引发了我对红黑树的兴趣。小白的我表示之前对2叉树也是一之半解，更不要提复杂的红黑树了。因此下定决心分析下红黑树。

首先介绍几个我在这周过程中在网上看到的一些比较好的文章如下：

关于红黑树：

http://www.kuqin.com/shuoit/20160630/352539.html 对插入和删除分析的比较到位

http://blog.csdn.net/v_july_v/article/details/6105630 比较详细的介绍了红黑树的相关知

http://blog.csdn.net/v_JULY_v/article/details/6284050 july的系列文章，

http://www.cs.usfca.edu/~galles/visualization/Algorithms.html 算法的动态过程

java 中的数据结构分析HashMap TreeMap

http://tech.meituan.com/java-hashmap.html 美团分析的重新认识HashMap

http://www.importnew.com/21818.html TreeMap 实现

https://www.ibm.com/developerworks/cn/java/j-lo-tree/ TreeMap

http://yemengying.com/2016/05/07/threadsafe-hashmap/ HashMap并发性问题

接下来，对我这周的分析进行总结：

红黑树

首先介绍下红黑树的相关知识，其是一颗平衡的二叉树。其在二叉查找树的基础上增加了红黑色和相关的性质使得红黑树相对平衡，从而保证了红黑树的查找、插入、删除的时间复杂度最坏为O(log n)。

具体的5个性质简单要介绍，1）结点要么为黑要么为红、2）根结点为黑色、3）每个叶节点（包括null节点）为黑色、4）红色结点的叶结点必须为黑色、5）所有的节点到叶节点的黑结点数相同。

从上面五个性质，我们可以得出如下结点可以是连续黑色的，n个结点的高度为logN。

红黑树的插入和删除是通过结点的旋转（左右旋转），节点变色来重新平衡。接下来重点介绍下插入的恢复过程：

插入必为红色，其叔叔为红色，将叔，父变黑，祖父变红。叔叔为黑色，根据其父是左结点还是右结点，进行相应的左旋与右旋。而后将父节点变黑，祖父变红。如此直到根结点。

TreeMap 与HashMap

之前我就在想，hashMap如果用到了红黑树，可以直接引用treemap中的红黑树算法，但实际是HashMap中也自己实现了一套HashMap。首先分析下treeMap的算法实现:

TreeMap在构造有一定要求，如果在构造时没有比较器。则在put时，如果key没有实现comparable接口，则会抛出异常。因此要么在构造时指定，要么key实现comparable接口。

TreeMap(Comparator<? super K> comparator) {
    this.comparator = comparator;
}

接着就是通过一个标准的2叉树查找实现，找到插入点。

Entry<K,V> t = root;
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);

查到插入点parent 后，插入节点，而后通过fixAfterInsertion（e）自修复

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;

修复过程如下：

首先将节点变为红色，

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) {
            setColor(parentOf(x), BLACK);
            setColor(y, BLACK);
            setColor(parentOf(parentOf(x)), RED);
            x = parentOf(parentOf(x));
        } else {
            if (x == rightOf(parentOf(x))) {
                x = parentOf(x);
                rotateLeft(x);
            }
            setColor(parentOf(x), BLACK);
            setColor(parentOf(parentOf(x)), RED);
            rotateRight(parentOf(parentOf(x)));
        }
    } else {
       与上面类似，只是方向不同，这里的逻辑对应的是父节点是祖父节点的右孩子，
       所以如果当前节点是左孩子，先要右旋。形成两个连续的红色，接着祖父变红，父变黑，因为之前是右孩子，左旋。
    }}
    root.color = Baclk;
}

最后当前结点为root时，将root结点改为Black，插入结束。

删除时类似，先找到节点，删除后恢复，类似。

HashMap

HashMap 在1.8 之前，由一个链表的数组。不是一个单纯的树。而java8 后，同样还是数组，但是数组中的值有了变化，并一定是链表，默认的设置当前超过8时，链表转换成红黑树的结构，但是还是会保持一个以红黑树的root的节点为链表头的链表。

其中treeMap中与另外一个与TreeMap中不同的是，key不要求实现comparable，或者指定比较器，而是用hashcode值的大小来比较，注意不是hashcode的转换后，比数据长度&后的值，因为同一个链表中的该值是相等的，而hashcode不一定一样，但是如果相等，则会通过 object对象的，如下方法，如果没有重写hashcode，则默认用如下方法