HashMap源码解析

1 红黑树

红黑树是一种自平衡的二叉树,他可以避免二分搜索树在极端情况下蜕化成链表的情况。

1.2 2-3树

2-3树构建过程:
在2-3树中插入35和25
image.png
再插入37时,一个节点就包含三个元素,就要进行分裂:
image.png
插入20和33:
image.png
再插入17和43:
image.png
插入27和35,两个节点都能容纳新的元素:
image.png
那么再最后插入22,结果发现,一个节点容纳了3个元素,要进行分裂,但是分裂后,叶子节点的高度不一致了,那么就要再进行聚合操作,如下所示:
image.png

1.3 2-3转红黑树

我们可以采用左倾红黑树的方式,来将2-3树转换为红黑树,转换规则如下:
image.png
2-3树:
image.png

image.png

image.png

1.4 红黑树的性质

① 所有的节点要么是黑色的要么是红色的;
② 根节点是黑色的;
③ 叶子结点是黑色的;
④ 每一个红色节点的子节点都是黑色的;
⑤ 从任意节点到子节点经过的黑色节点个数是相同的。
③中描述的叶子节点一定是黑色,和上图不符合,是因为33中还有两个为空的子节点。
image.png

2 HashMap源码解析

image.png

2.1 成员变量

static final int DEFAULT_INITIAL_CAPACITY = 1 << 4; // aka 默认初始容量大小:2的4次方 16 
static final int MAXIMUM_CAPACITY = 1 << 30; //最大容量:2的30次方,Integer.MAX_VALUE
static final float DEFAULT_LOAD_FACTOR = 0.75f;//默认加载因子
static final int TREEIFY_THRESHOLD = 8; //计数阈值至少为8转化为使用树而不是列表
static final int UNTREEIFY_THRESHOLD = 6; //计数阈值小于6反树化,即红黑树转为列表
static final int MIN_TREEIFY_CAPACITY = 64; //可对桶进行树化的最小表容量
transient Node<K,V>[] table; //表在第一次使用时初始化,大小调整为必要的。在分配时,长度总是2的幂。在某些操作中,我们也允许长度为零。目前不需要的引导机制。) 
transient Set<Map.Entry<K,V>> entrySet; //保存缓存的entrySet()
transient int size; //包含的键值映射的元素数量
transient int modCount; //HashMap在结构上被修改的次数,用于快速失败机制
int threshold; // 调整大小的阈值(容量*负载因子)
final float loadFactor; //哈希表扩容使用的负载因子

2.2 构造函数

public HashMap() {
    // DEFAULT_LOAD_FACTOR = 0.75f;当比例达到0.75时,进行扩容
    this.loadFactor = DEFAULT_LOAD_FACTOR;
}

// 主要对两个参数进行校验
public HashMap(int initialCapacity, float loadFactor) {
    if (initialCapacity < 0) {
        throw new IllegalArgumentException("Illegal initial capacity: " +
                                           initialCapacity);
    }
    if (initialCapacity > MAXIMUM_CAPACITY) {
        initialCapacity = MAXIMUM_CAPACITY;
    }
    if (loadFactor <= 0 || Float.isNaN(loadFactor)) {
        throw new IllegalArgumentException("Illegal load factor: " +
                                           loadFactor);
    }
    this.loadFactor = loadFactor;
    this.threshold = tableSizeFor(initialCapacity);
}

// 直接调用两个参数的构造方法
public HashMap(int initialCapacity) {
    this(initialCapacity, DEFAULT_LOAD_FACTOR);
}

2.3 put方法

直接调用putVal方法,putVal方法解析见:2.5

public V put(K key, V value) {
    return putVal(hash(key), key, value, false, true);
}

2.4 hash(key)

通过扰动函数,重新计算hashCode的值
“>>>”:右移16位,所以原来的低16位消失了,高16位变成了低16为,现在的高16位变成了0
按位异或运算(^):两个数转为二进制,然后从高位开始比较,如果相同则为0,不相同则为1。

static final int hash(Object key) {
int h;
/**
    扰动函数————(h = key.hashCode()) ^ (h >>> 16) 表示:
    将key的hashCode一分为二。其中:
    【高半区16位】数据不变。
    【低半区16位】数据与高半区16位数据进行异或操作,以此来加大低位的随机性。
     注意:如果key的哈希code小于等于16位,那么是没有任何影响的。只有大于16位,才会触发扰动函数的执行效果。
    */
    return (key == null) ? 0 : (h = key.hashCode()) ^ (h >>> 16);
}

2.5 putVal方法

向HashMap中添加k-v元素

/**
 * 向HashMap中添加k-v元素
 *
 * @param onlyIfAbsent 如果是true,则不改变已存在的value值,传入的为false,所以会随时更新
   @param evict        驱逐,赶出,逐出(空方法,未使用)
 */
final V putVal(int hash, K key, V value, boolean onlyIfAbsent, boolean evict) {
    Node<K, V>[] tab;
    Node<K, V> p;
    int n, i;

    /** 1:如果是空的table,那么默认初始化一个长度为16的空的Node[]数组 */
    if ((tab = table) == null || (n = tab.length) == 0)
        // resize()见:2.6
        n = (tab = resize()).length;

    /** 2:如果计算后的下标i,在tab数组中没有数据,那么则直接在下标i处,新增Node节点 */
    // i = (n - 1) & hash:寻址操作,结果:0~n-1,判断有没有哈希冲突
    if ((p = tab[i = (n - 1) & hash]) == null)
        tab[i] = newNode(hash, key, value, null);

    /** 3:如果计算后的下标i,在tab数组中已存在数据,则执行以下逻辑 */
    else {
        Node<K, V> e;
        K k;

        /** 3-1:如果与已存在的Node是相同的key值 */
        // hash值相同,key值相同,说明只是更新value操作
        if (p.hash == hash && ((k = p.key) == key || (key != null && key.equals(k))))
            e = p;

        /** 3-2:如果与已存在的Node是相同的key值,并且是树节点 */
        else if (p instanceof TreeNode)
            e = ((TreeNode<K, V>) p).putTreeVal(this, tab, hash, key, value);

        /** 3-3:如果已存在的Node是链表节点,则循环遍历链表,并对比hash和key,如果都不相同,则将新的Node拼装到链表的末尾;如果相同,则对节点的value进行更新 */
        else {
            // binCount 节点个数
            for (int binCount = 0; ; ++binCount) {
                /** case1:如果遍历到了链表的最后一个Node节点,则创建新的节点并拼接到链表尾部,并跳出循环 */
                if ((e = p.next) == null) {
                    p.next = newNode(hash, key, value, null);
                    // binCount从0开始,也就是链表中第2个node对应binCount=0,
                    // 如果Node链表大于8个Node,那么试图变为红黑树
                    // TREEIFY_THRESHOLD:树化阈值
                    if (binCount >= TREEIFY_THRESHOLD - 1) 
                        // treeifyBin 解析见:2.7
                        treeifyBin(tab, hash);
                    break;
                }

                /** case2:循环链表的过程中找到了key值相同的节点,直接跳出循环,更新value */
                if (e.hash == hash && ((k = e.key) == key || (key != null && key.equals(k))))
                    break;

                p = e;
            }
        }

        /** 3-4:如果存在相同的key值,更新value值 */
        if (e != null) {
            V oldValue = e.value;
            // 如果onlyIfAbsent=false或者oldValue不为空,则将新的value值进行更新
            if (!onlyIfAbsent || oldValue == null)
                e.value = value;
            afterNodeAccess(e); // 空方法
            return oldValue;
        }
    }
    ++modCount;

    /** 4:如果长度超过阈值,则执行扩容操作 */
    if (++size > threshold) resize();

    afterNodeInsertion(evict); // 空方法
    return null;
}

2.6 resize方法

table扩容

/**
* 执行table扩容操作
*
* 常量命名解释:
*      Tab——>Table 表
*      Cap——>Capacity 容量
*      Thr——>Threshold 阈值
*/
final Node<K, V>[] resize() {
    Node<K, V>[] oldTab = table;                        // 旧的table表
    int oldCap = (oldTab == null) ? 0 : oldTab.length;  // 旧table表的容量
    int oldThr = threshold;                             // 旧table表的阈值
    int newCap = 0;  // 新table表的容量
    int newThr = 0;  // 新table表的阈值

    /**
    * Part 1:根据情况,调整新表的容量newCap和阈值newThr
    */
    /** 1-1:如果老table的容量大于0 **/
    if (oldCap > 0) {
        if (oldCap >= MAXIMUM_CAPACITY) {
            threshold = Integer.MAX_VALUE;
            return oldTab;
        }
        else if ((newCap = oldCap << 1) < MAXIMUM_CAPACITY && oldCap >= DEFAULT_INITIAL_CAPACITY)
        newThr = oldThr << 1; // 新table【阈值】和【容量】都扩容为旧table【阈值】和【容量】的2倍
    }

    /** 1-2:oldCap小于等于0 而 oldThr大于0 */
    else if (oldThr > 0) newCap = oldThr;

    /** 1-3:新的HashMap,第一次请求put方法,创建底层table数组及相关变量 */
    else {
        newCap = DEFAULT_INITIAL_CAPACITY; // 默认值:16
        newThr = (int) (DEFAULT_LOAD_FACTOR * DEFAULT_INITIAL_CAPACITY); // 默认为:12
    }

    if (newThr == 0) {
        float ft = (float) newCap * loadFactor;
        newThr = (newCap < MAXIMUM_CAPACITY && ft < (float) MAXIMUM_CAPACITY ? (int) ft : Integer.MAX_VALUE);
    }
    threshold = newThr;

    /** 1-4:构建新的table数组 */
    @SuppressWarnings({"rawtypes", "unchecked"})
    Node<K, V>[] newTab = (Node<K, V>[]) new Node[newCap];
    table = newTab;

    /**
    * Part 2:如果老的table里有数据,则进行数据迁移
    */
    if (oldTab != null) {
        // 复制逻辑,将旧数组数据复制到新数组中
        /** 2-1:遍历存储于旧table数组中的每条链表 */
        for (int j = 0; j < oldCap; ++j) {
            Node<K, V> e;
            if ((e = oldTab[j]) != null) { // 如果在旧table数组中j的位置存在一条链表,则进行数据迁移操作
                oldTab[j] = null;

                /** case1:针对【单节点】进行数据迁移操作*/
                if (e.next == null)
                newTab[e.hash & (newCap - 1)] = e;

                /** case2:针对【红黑树】进行数据迁移操作 */
                else if (e instanceof TreeNode)
                ((TreeNode<K, V>) e).split(this, newTab, j, oldCap);

                /** case3:针对【链表】进行数据迁移操作 */
                else {
                    Node<K, V> loHead = null;   // 【低位】链表【头】指针
                    Node<K, V> loTail = null;   // 【低位】链表【尾】指针
                    Node<K, V> hiHead = null;   // 【高位】链表【头】指针
                    Node<K, V> hiTail = null;   // 【高位】链表【尾】指针
                    Node<K, V> next;
                    do {
                        next = e.next;
                        // 数组扩容后,链表存放的位置和原来的位置可能不一样
                        // 下方的判断逻辑是确定链表将存放在低位还是高位
                        /** case3-1:节点e需要被存放在新table的【低位】链表中 */
                        if ((e.hash & oldCap) == 0) { // eg1:oldCap=16=0010000;&计算结果如果为0,说明e.hash的第5位是0,否则是1
                            if (loTail == null) loHead = e; // 设置【头】指针
                            else loTail.next = e;           // 拼装链表
                            loTail = e;                     // 设置【尾】指针
                        }

                        /** case3-2:节点e需要被存放在新table的【高位】链表中 */
                        else {
                            if (hiTail == null) hiHead = e; // 设置【头】指针
                            else hiTail.next = e;           // 拼装链表
                            hiTail = e;                     // 设置【尾】指针
                        }
                    }
                    while ((e = next) != null); // 获取链表的下一个节点

                    /** case3-3:如果存在【低位】链表,则将其赋值到新table中 */
                    if (loTail != null) {
                        loTail.next = null;
                        newTab[j] = loHead;
                    }

                    /** case3-4:如果存在【高位】链表,则将其赋值到新table中 */
                    if (hiTail != null) {
                        hiTail.next = null;
                        newTab[j + oldCap] = hiHead;
                    }
                }
            }
        }
    }
    return newTab;
}

2.7 treeifyBin方法

在putVal中调用是试图转化成红黑树,并不是一定会转化,看下面代码中的1,还有一次判断,tab为空或者长度小于MIN_TREEIFY_CAPACITY时,最小树容量:MIN_TREEIFY_CAPACITY=64

/**
 * 替换bin中给定哈希索引处的所有链接节点,除非表太小,在这种情况下调整大小。
 */
final void treeifyBin(Node<K, V>[] tab, int hash) {
    int n;
    int index;
    Node<K, V> e;

    /** 1:当表tab的长度小于64时,只扩展数组大小,不转换为红黑树 */
    if (tab == null || (n = tab.length) < MIN_TREEIFY_CAPACITY)
        resize();

    /** 2:如果新增的node要插入的数组位置已经有链表存在了,则将该链表转换为红黑树 */
    else if ((e = tab[index = (n - 1) & hash]) != null) {
        /** 2-1:遍历链表中每个节点,将Node转化为TreeNode */
        TreeNode<K, V> hd = null; // 指向头节点
        TreeNode<K, V> tl = null; // 指向尾节点
        do {
            // replacementTreeNode 只是将Node转为TreeNode
            // new TreeNode<>(p.hash, p.key, p.value, next)
            TreeNode<K, V> p = replacementTreeNode(e, null);
            // 将每个Node转换为TreeNode,以hd为头节点,
            // 每个TreeNode用prev和next连接成新的TreeNode链表(双向链表)
            if (tl == null) {
                hd = p;
            } else {
                p.prev = tl;
                tl.next = p;
            }
            tl = p;
        } while ((e = e.next) != null);

        /** 32-2:如果头节点hd不为null,则TreeNode链表进行树型化操作 */
        if ((tab[index] = hd) != null)
            // treeify解析见:2.8
            hd.treeify(tab);
    }
}

2.8 treeify方法

将TreeNode双向连边转换成树

final void treeify(Node<K,V>[] tab) {
    TreeNode<K,V> root = null;
    // 1. 遍历双向链表中的每一个TreeNode
    for (TreeNode<K,V> x = this, next; x != null; x = next) {
        next = (TreeNode<K,V>)x.next;
        x.left = x.right = null;
        // 2. 如果根节点为空,就将第一个TreeNode作为root节点
        if (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;;) {
                // 3. 通过dir的值确定插入的大致位置,是当前节点的左侧还是右侧
                int dir, ph;
                K pk = p.key;
                // 父节点大于当前节点
                if ((ph = p.hash) > h)
                    dir = -1;
                else if (ph < h)
                    dir = 1;
                else if ((kc == null &&
                          (kc = comparableClassFor(k)) == null) ||
                         (dir = compareComparables(kc, k, pk)) == 0)
                    dir = tieBreakOrder(k, pk);

                // 4. 通过上面dir的值插入节点
                TreeNode<K,V> xp = p;
                if ((p = (dir <= 0) ? p.left : p.right) == null) {
                    x.parent = xp;
                    if (dir <= 0)
                        xp.left = x;
                    else
                        xp.right = x;
                    // 红黑树平衡调整,解析见:2.9
                    root = balanceInsertion(root, x);
                    break;
                }
            }
        }
    }
    moveRootToFront(tab, root);
}

�2.9 balanceInsertion方法

红黑树平衡调整

static <K,V> TreeNode<K,V> balanceInsertion(TreeNode<K,V> root, TreeNode<K,V> x) {
  x.red = true;
  // xp:x的父节点,xpp:x父节点的父节点......
  for (TreeNode<K,V> xp, xpp, xppl, xppr;;) {
    // 没有根节点,将自己变为根节点
    if ((xp = x.parent) == null) {
      x.red = false;
      return x;
    }
    // x父节点为黑色,或者没有祖父节点
    else if (!xp.red || (xpp = xp.parent) == null)
      return root;
    // x的父节点在x祖父节点的左侧
    if (xp == (xppl = xpp.left)) {
      // 祖父节点的右子节点是红色节点,则将祖父节点的左右两个节点都变为黑色,祖父节点变为红色
      // 图解见图2.9.1场景一
      if ((xppr = xpp.right) != null && xppr.red) {
        xppr.red = false;
        xp.red = false;
        xpp.red = true;
        x = xpp;
      }
      // 旋转+变色,如果xpp不存在右子节点,或者该节点是黑色
      else {
        if (x == xp.right) {
          // 【左旋+变色】,图解见:图2.9.2,步骤一
          root = rotateLeft(root, x = xp);
          xpp = (xp = x.parent) == null ? null : xp.parent;
        }
        if (xp != null) {
          xp.red = false;
          if (xpp != null) {
            xpp.red = true;
            // 右旋+变色,图解见:图2.9.2,步骤二
            root = rotateRight(root, xpp);
          }
        }
      }
    }
    // x的父节点在x祖父节点的右侧
    else {
      // 祖父节点的左子节点是红色节点,则将祖父节点的左右两个节点都变为黑色,祖父节点变为红色
      // 图解见图2.9.1场景二
      if (xppl != null && xppl.red) {
        xppl.red = false;
        xp.red = false;
        xpp.red = true;
        x = xpp;
      }
      // 【旋转+变色】操作:如果xpp不存在左子节点 或者 该节点是黑色的
      else {
        // 【右旋+变色】,如果x的父节点,图解见:图2.9.3,步骤一
        if (x == xp.left) {
          root = rotateRight(root, x = xp);
          xpp = (xp = x.parent) == null ? null : xp.parent;
        }
        if (xp != null) {
          xp.red = false;
          if (xpp != null) {
            xpp.red = true;
            // 左旋+变色,图解见:图2.9.3,步骤二
            root = rotateLeft(root, xpp);
          }
        }
      }
    }
  }
}

图2.9.1 变色操作
image.png
图2.9.2 变色+旋转
image.png
图2.9.3 变色+旋转
image.png

3 红黑树插入节点转换过程

image.png

  • 3
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值