HashMap中链表转换成红黑树的操作——treeifyBin()

前一篇文章中提到,当执行put操作的时候,会出现桶碰撞的情况,这时候桶索引值相同的键值对会以一个链表的形式存在于hash桶中,但是当链表长度很长的时候,查找的性能会很低,JDK1.8中对此进行了优化,当链表的长度超过TREEIFY_THRESHOLD(8)的时候,链表会树化,即通过treeifyBin()方法转换成红黑树。

在看代码之前,我们回忆一下二叉搜索树的插入逻辑insert(Node node, Node root),将node结点插入到root树中

1、如果root == null,则root = node

2、从根结点开始遍历,如果node.value < 当前结点.value,插入到左子树,否则插入到右子树,

3、如果当前结点的左儿子为空,且node.value < 当前结点.value,插入到当前结点的左儿子,

4、如果当前结点的右儿子为空,且node.value > 当前结点.value,插入到当前结点的右儿子。

现在具体来看看treeifyBin()的代码:

static final int MIN_TREEIFY_CAPACITY = 64;

final void treeifyBin(Node<K,V>[] tab, int hash) {
        int n, index; Node<K,V> e;
        //这里还有一个限制条件,当table的长度小于MIN_TREEIFY_CAPACITY(64)时,只是进行扩容
        if (tab == null || (n = tab.length) < MIN_TREEIFY_CAPACITY)   
            resize();                                                                                                   //而扩容的时候,链表的长度有可能会变短
        else if ((e = tab[index = (n - 1) & hash]) != null) {
            TreeNode<K,V> hd = null, tl = null;
            do {        
                //将链表中的结点转换为树结点,形成一个新链表
                TreeNode<K,V> p = replacementTreeNode(e, null);
                if (tl == null)
                    hd = p;
                else {
                    p.prev = tl;
                    tl.next = p;
                }
                tl = p;
            } while ((e = e.next) != null);

            if ((tab[index] = hd) != null)   //将新的树结点链表赋给第index个桶
                hd.treeify(tab);            //执行 TreeNode中的treeify()方法
        }
}

TreeNode<K,V> replacementTreeNode(Node<K,V> p, Node<K,V> next) {
        return new TreeNode<>(p.hash, p.key, p.value, next);
}

从上面的代码可以看出,链表长度大于8而且整个map中的键值对大于等于MIN_TREEIFY_CAPACITY (64)时,才进行链表到红黑树的转换。

上面的代码很简单,重点看一下treeify( )方法:

static final class TreeNode<K,V> extends LinkedHashMap.Entry<K,V> {
        TreeNode<K,V> parent;  // red-black tree links
        TreeNode<K,V> left;
        TreeNode<K,V> right;
        TreeNode<K,V> prev;    // needed to unlink next upon deletion
        boolean red;
        TreeNode(int hash, K key, V val, Node<K,V> next) {
            super(hash, key, val, next);
        }
        final void treeify(Node<K,V>[] tab) {//将链表转换成红黑树
            TreeNode<K,V> root = null;
            for (TreeNode<K,V> x = this, next; x != null; x = next) {//遍历链表中的每一个TreeNode,当前结点为x
                next = (TreeNode<K,V>)x.next;
                x.left = x.right = null;
                if (root == null) {         //对于第一个树结点,当前红黑树的root == null,所以第一个结点是树的根,设置为黑色
                    x.parent = null;
                    x.red = false;
                    root = x;
                }
                else { //对于余下的结点:
                    K k = x.key;
                    int h = x.hash;
                    Class<?> kc = null;
                    for (TreeNode<K,V> p = root; ; ) {//从根结点开始遍历,寻找当前结点x的插入位置
                        int dir, ph;
                        K pk = p.key;
                        if ((ph = p.hash) > h)   //如果当前结点的hash值小于根结点的hash值,方向dir = -1;
                            dir = -1;
                        else if (ph < h)                //如果当前结点的hash值大于根结点的hash值,方向dir = 1;
                            dir = 1;
                        else if ((kc == null &&         //如果x结点的key没有实现comparable接口,或者其key和根结点的key相等(k.compareTo(x) == 0)仲裁插入规则
                                  (kc = comparableClassFor(k)) == null) ||      //只有k的类型K直接实现了Comparable<K>接口,才返回K的class,否则返回null,间接实现也不行。
                                 (dir = compareComparables(kc, k, pk)) == 0)
                            dir = tieBreakOrder(k, pk);         //仲裁插入规则

                        TreeNode<K,V> xp = p;
                        if ((p = (dir <= 0) ? p.left : p.right) == null) {      //如果p的左右结点都不为null,继续for循环,否则执行插入
                            x.parent = xp;
                            if (dir <= 0)           //dir <= 0,插入到左儿子
                                xp.left = x;
                            else            //否则插入到右结点
                                xp.right = x;
                            root = balanceInsertion(root, x);   //插入后进行树的调整,使之符合红黑树的性质
                            break;
                        }
                    }
                }
            }
            moveRootToFront(tab, root);         //Ensures that the given root is the first node of its bin.
        }
}

/**
    * Tie-breaking utility for ordering insertions when equal
  * hashCodes and non-comparable. We don't require a total
  * order, just a consistent insertion rule to maintain
  * equivalence across rebalancings. Tie-breaking further than
  * necessary simplifies testing a bit.
  */
static int tieBreakOrder(Object a, Object b) {
  int d;
  if (a == null || b == null ||
      (d = a.getClass().getName().
       compareTo(b.getClass().getName())) == 0)
    d = (System.identityHashCode(a) <= System.identityHashCode(b) ?
         -1 : 1);
  return d;
}

从上面的代码可以看出,将结点插入到红黑树的时候,比较的是hash值,然后根据比较结果判断插入左子树还是右子树。

总结

  • 整个过程大体上分为两部分:插入树结点元素,具体可以参考二叉搜索树的插入操作;插入完之后开始调整红黑树,使之符合红黑树的特性。

  • 插入的时候,先比较key的hash值来决定插入方向,如果hash值相等的话,再比较compare方法,如果key所属对象没有直接实现Comparable接口,或者compare方法返回0,执行tieBreakOrder,比较两个key所属Class的name,如果还相等,也就是两个对象是同一个类型,那么调用本地方法为两个对象生成hashCode值,再进行比较,hashCode相等的话返回-1。

参考

1、https://blog.csdn.net/qpzkobe/article/details/79533237
2、https://blog.csdn.net/weixin_42340670/article/details/80673127

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 159,219评论 4 362
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,363评论 1 293
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,933评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,020评论 0 206
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,400评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,640评论 1 219
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,896评论 2 313
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,597评论 0 199
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,327评论 1 244
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,581评论 2 246
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,072评论 1 261
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,399评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,054评论 3 236
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,083评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,849评论 0 195
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,672评论 2 274
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,585评论 2 270

推荐阅读更多精彩内容

  • 1.HashMap是一个数组+链表/红黑树的结构,数组的下标在HashMap中称为Bucket值,每个数组项对应的...
    谁在烽烟彼岸阅读 994评论 2 2
  • 一、基本数据类型 注释 单行注释:// 区域注释:/* */ 文档注释:/** */ 数值 对于byte类型而言...
    龙猫小爷阅读 4,213评论 0 16
  • HashMap是一个散列表,通过键值对的形式对元素进行存储,HashMap是线程不安全的,key-value都是可...
    小川君阅读 1,578评论 1 1
  • 不好意思就是慢性自杀:乞丐不好意思要饭,结果饿死了;商户不好意思要账,结果自己门店关门了。不好意思向心仪的人表白,...
    潍坊谷德李平霞阅读 638评论 4 2
  • 我的朋友圈里很干净。 我几乎不发朋友圈。 有时心血来潮会狂发一通,文字极长,还会配上心仪的插图。 但过不了几天我又...
    爬山虎之力阅读 168评论 0 1