从java8之后HashMap的put方法分析HashMap是怎么实现的

最新推荐文章于 2023-03-09 14:37:14 发布

才_先生

最新推荐文章于 2023-03-09 14:37:14 发布

阅读量281

点赞数

分类专栏： java 文章标签：哈希算法 java 散列表

本文链接：https://blog.csdn.net/qq_28227405/article/details/125963081

版权

java 专栏收录该内容

11 篇文章 0 订阅

订阅专栏

本文详细解析了Java 11中HashMap的putVal方法，阐述了HashMap如何处理键值对的插入，包括数组初始化、哈希碰撞处理、链表与红黑树的转换等核心逻辑。当哈希冲突时，HashMap使用链表解决，超过一定长度会转为红黑树以优化查找效率。

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

public class TestHashMap {
    public static void main(String[] args) {
        Map<String, String> hasmap = new HashMap<>();
        System.out.println(hasmap.put("abc", "1"));
        System.out.println(hasmap.put("abc", "2"));
        System.out.println(hasmap.put("abc", "3"));
    }
}

java11的HashMap源码如下：

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

final V putVal(int hash, K key, V value, boolean onlyIfAbsent,
                   boolean evict) {
        Node<K,V>[] tab; Node<K,V> p; int n, i;
        if ((tab = table) == null || (n = tab.length) == 0)
            n = (tab = resize()).length;
        if ((p = tab[i = (n - 1) & hash]) == null)
            tab[i] = newNode(hash, key, value, null);
        else {
            Node<K,V> e; K k;
            if (p.hash == hash &&
                ((k = p.key) == key || (key != null && key.equals(k))))
                e = p;
            else if (p instanceof TreeNode)
                e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);
            else {
                for (int binCount = 0; ; ++binCount) {
                    if ((e = p.next) == null) {
                        p.next = newNode(hash, key, value, null);
                        if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st
                            treeifyBin(tab, hash);
                        break;
                    }
                    if (e.hash == hash &&
                        ((k = e.key) == key || (key != null && key.equals(k))))
                        break;
                    p = e;
                }
            }
            if (e != null) { // existing mapping for key
                V oldValue = e.value;
                if (!onlyIfAbsent || oldValue == null)
                    e.value = value;
                afterNodeAccess(e);
                return oldValue;
            }
        }
        ++modCount;
        if (++size > threshold)
            resize();
        afterNodeInsertion(evict);
        return null;
    }

putVal方法解读：
1.每次执行put方法时都会先执行if ((tab = table) == null || (n = tab.length) == 0) 判断数组对象是否为空，如果空的时候会先执行n = (tab = resize()).length;这个方法是创建数组；
2.p = tab[i = (n - 1) & hash]：n是数组的默认长度大小，一般是16，hash是指对key的has值，做了&运算后找到key在数组中的下标位置。
3.第二步的p值如果是空的，则执行tab[i] = newNode(hash, key, value, null)创建一个元素。在本例子来说当执行到hasmap.put("abc", "1")方法时，abc这个key的hash值在数组中不存在，则创建一个元素放到数组中的这个位置。因为if (e != null)是null，不成立，这段if里面的逻辑就不执行。最后就是返回一个null。
4.当继续执行hasmap.put("abc", "2") 时，if ((tab = table) == null || (n = tab.length) == 0) 就是false,(p = tab[i = (n - 1) & hash]) == null 也是false, 然后就会执行if (p.hash == hash && ((k = p.key) == key || (key != null && key.equals(k)))) e = p; 这段代码；p这个变量实际上是非空的，因为在(p = tab[i = (n - 1) & hash]) == null的时候已经把数组的值给取出来了。e是指当前的p对象。最后再执行这段代码if (e != null) { // existing mapping for key V oldValue = e.value; if (!onlyIfAbsent || oldValue == null) e.value = value; afterNodeAccess(e); return oldValue; } 把新值放到key的value且返回key的旧值。

for (int binCount = 0; ; ++binCount) {
     if ((e = p.next) == null) {
         p.next = newNode(hash, key, value, null);
         if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st
             treeifyBin(tab, hash);
         break;
     }
     if (e.hash == hash &&
         ((k = e.key) == key || (key != null && key.equals(k))))
         break;
     p = e;
 }

5.关于上面的这段代码怎么理解呢？实际上就是当key的hash出现冲突了，且p不是TreeNode类型还没转换为红黑树时才会执行这段代码。hash冲突是指不同的key经过hash之后的值有可能是一样的。在数组中的每个元素的hash值不一样，相同的hash值如果有多个就是链表的形式即当前的node持有下一个node。这段代码的作用就是在链表中查找需要的key值，如果没找到就在链表的最后一个节点新增一个节点。且对链表的长度进行校验一旦长度大于8就会把链表转换为红黑树if (binCount >= TREEIFY_THRESHOLD - 1) treeifyBin(tab, hash); 这里这样说实际上又不是很正确， treeifyBin(tab, hash);里面会对数组长度进行判断，当数组的长度大于64的时候才会转换为红黑树，小于64的时候还是采用扩容方式。treeifyBin源码如下所示：

final void treeifyBin(Node<K,V>[] tab, int hash) {
        int n, index; Node<K,V> e;
        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);
        }
    }