HashMap中保证红黑树根节点一定是对应链表头节点moveRootToFront()方法源码解读

红黑树根节点若不为其对应链表的头节点,则按照下述步骤的处理,将根节点向前移动到头节点:

  1. 将根节点从所在链表中删除,即链表的删除操作:修改根节点的前后节点的指向即可,即将根节点的上一节点的下一节点设置为根节点的下一节点,将根节点的下一节点的上一节点设置为根节点的上一节点。
  2. 将根节点所在链表的头节点位置的值设置为根节点;
  3. 将现有头节点的上一个节点设置为根节点,将根节点的下一个节点设置为现有头节点,将根节点的上一个节点设置为null.
    具体源码如下:
/**HashMap$TreeNode的moveRootToFront()方法源码*/
        /*** Ensures that the given root is the first node of its bin.*/
        static <K,V> void moveRootToFront(Node<K,V>[] tab, TreeNode<K,V> root) {
    
    
            int n;
            if (root != null && tab != null && (n = tab.length) > 0) {
    
     
                int index = (n - 1) & root.hash;
                TreeNode<K,V> first = (TreeNode<K,V>)tab[index]; //目前的头节点
                if (root != first) {
    
    //根节点不等于目前的头节点
                    Node<K,V> rn;
                    tab[index] = root; //将根节点设置为头节点
                    TreeNode<K,V> rp = root.prev;
                    if ((rn = root.next) != null)
                        ((TreeNode<K,V>)rn).prev = rp; //将根节点的下一节点的上一节点指向根节点的上一节点
                    if (rp != null)
                        rp.next = rn; //将根节点的上一节点的下一节点指向根节点的下一节点
                    if (first != null)
                        first.prev = root; //将原头节点的上一节点指向根节点
                    root.next = first; //将根节点的下一节点指向原头节点
                    root.prev = null; //将根节点的上一节点设为null
                }
                assert checkInvariants(root); //检查根节点是否满足红黑树节点规则
            }
        }

示意图:
将红黑树的根节点移动到对应链表头节点,假设根节点所处位置如下左图所示,初始链表指向为实线,调整指向的线为虚线;调整后的链表指向如下右图所示:
在这里插入图片描述
PS:由于文档是我在本地编写好之后再赋值过来的,有些文本格式没能完整的体现,故提供下述图片,供大家阅览,以便有更好的阅读体验:
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/u010425839/article/details/106653696