HashMap put流程源码解读

阿胜yss

已于 2023-03-12 17:54:04 修改

阅读量79

点赞数

文章标签： java 数据结构链表

于 2023-03-10 16:17:16 首次发布

本文链接：https://blog.csdn.net/weixin_46591962/article/details/129434967

版权

文章详细阐述了HashMap的put方法的工作原理，包括计算key的哈希值、处理哈希冲突、链表与红黑树的转换以及扩容机制。在遇到哈希冲突时，HashMap会根据节点类型采用链表或红黑树存储，并在达到一定阈值时将链表转换为红黑树以优化查找性能。

摘要由CSDN通过智能技术生成

// // 返回值: previous value, or null if none
public V put(K key, V value) {
	 // put方法直接调用putVal方法 
	 // 第一个参数是获取key的hash值，具体方法在下面，
	 // 第二个参数是key，第三个是value，
	 // 第四个参数是 是否替换原value false 是 true 否
     return putVal(hash(key), key, value, false, true);
 }
 .....
 
 static final int hash(Object key) {
     int h;
     return (key == null) ? 0 : (h = key.hashCode()) ^ (h >>> 16);
 }

我们继续来看putVal方法

// 返回值: previous value, or null if none
final V putVal(int hash, K key, V value, boolean onlyIfAbsent,
                   boolean evict) {
         // 声明一些变量  
        Node<K,V>[] tab; Node<K,V> p; int n, i;
        // table 是数组 在首次初始化时要调整其数值为2的幂。
        // 如果table为null或者table的长度为0，则进行初始化扩容。
        if ((tab = table) == null || (n = tab.length) == 0)
            n = (tab = resize()).length;
            // key的hash 与 table的长度 -1 按位取与& 的值作为下标去数组中找到相应节点   如果相应节点为null 则没有发生hash碰撞，直接新建一个节点放在这个位置
        if ((p = tab[i = (n - 1) & hash]) == null)
            tab[i] = newNode(hash, key, value, null);
        else {
        // 发生hash碰撞
            Node<K,V> e; K k;
            // 判断数组中节点key的hash值 和key 是不是和当前节点的一样，
            // 如果一样，则把当前节点赋给临时节点e
            if (p.hash == hash &&
                ((k = p.key) == key || (key != null && key.equals(k))))
                e = p;
            // key不一样 首先判断节点类型是链表还是树
            else if (p instanceof TreeNode)
            // 树类型则按照树类型的放值方法
                e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);
            else {
            	// 链表类型
                for (int binCount = 0; ; ++binCount) {
                	// p的下一节点为null
                    if ((e = p.next) == null) {
                        p.next = newNode(hash, key, value, null);
                        // static final int TREEIFY_THRESHOLD = 8;
                        // 判断链表长度是不是大于8
                        if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st
                        // 如果链表长度 大于等于 8了，
                        // 则要判断是数组扩容还是链表变红黑树。
                            treeifyBin(tab, hash);
                        break;
                    }
                    // 如果找到key相等以及hash值相同的key
                    if (e.hash == hash &&
                        ((k = e.key) == key || (key != null && key.equals(k))))
                        // 退出循环
                        break;
                       //赋给临时节点p
                    p = e;
                }
            }
            if (e != null) { // existing mapping for key
                V oldValue = e.value;
                if (!onlyIfAbsent || oldValue == null)
                    e.value = value;
                afterNodeAccess(e);
                return oldValue;
            }
        }
        // 结构修改次数+1
        ++modCount;
        // threshold = 容量(默认16) * 负载因子(默认0.75)
        if (++size > threshold)
        	// 扩容
            resize();
        afterNodeInsertion(evict);
        return null;
    }

再来看putTreeVal方法

// 参数一:当前整个map集合
// 参数二: hash数组
// 参数三: key的hash值
// 参数四: k 参数五: v
// 此方法属于红黑树添加节点范畴
final TreeNode<K,V> putTreeVal(HashMap<K,V> map, Node<K,V>[] tab,
                                       int h, K k, V v) {
    Class<?> kc = null;
    boolean searched = false;
    // 通过root() 方法找到其根节点
    TreeNode<K,V> root = (parent != null) ? root() : this;
    // 
    for (TreeNode<K,V> p = root;;) {
        int dir, ph; K pk;
        if ((ph = p.hash) > h)
            dir = -1;
        else if (ph < h)
            dir = 1;
        else if ((pk = p.key) == k || (k != null && k.equals(pk)))
            return p;
        else if ((kc == null &&
                  (kc = comparableClassFor(k)) == null) ||
                 (dir = compareComparables(kc, k, pk)) == 0) {
            if (!searched) {
                TreeNode<K,V> q, ch;
                searched = true;
                if (((ch = p.left) != null &&
                     (q = ch.find(h, k, kc)) != null) ||
                    ((ch = p.right) != null &&
                     (q = ch.find(h, k, kc)) != null))
                    return q;
            }
            dir = tieBreakOrder(k, pk);
        }

        TreeNode<K,V> xp = p;
        if ((p = (dir <= 0) ? p.left : p.right) == null) {
            Node<K,V> xpn = xp.next;
            TreeNode<K,V> x = map.newTreeNode(h, k, v, xpn);
            if (dir <= 0)
                xp.left = x;
            else
                xp.right = x;
            xp.next = x;
            x.parent = x.prev = xp;
            if (xpn != null)
                ((TreeNode<K,V>)xpn).prev = x;
            moveRootToFront(tab, balanceInsertion(root, x));
            return null;
        }
    }
}

// 返回树的根节点
final TreeNode<K,V> root() {
     for (TreeNode<K,V> r = this, p;;) {
         if ((p = r.parent) == null)
         // 当r的parent节点为null时 返回
             return r;
         r = p;
     }
 }

当链表长度大于8时执行的方法

static final int MIN_TREEIFY_CAPACITY = 64;
....
// 第一个参数: hash数组
// 第一个参数: key的hash值
final void treeifyBin(Node<K,V>[] tab, int hash) {
    int n, index; Node<K,V> e;
    // hash数组为null  或者  hash数组的长度小于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)
            hd.treeify(tab);
    }
}