HashMap中put方法(jdk8)

  • Post author:
  • Post category:其他


一、put方法流程

首先看一下网上一个不错的流程图

1.先判断数组是否为空,如果为空则对数组进行resize()扩容

2.然后把插入元素的哈希值和数组长度-1进行与操作,如果table[i] == null,则直接将元素插入到数组对应位置中,否则进行3

3.判断当前元素的key在数组中是否存在(通过hash并equals方法判断),然后判断节点类型,如果是红黑树,则走4;如果是链表,则走5

4.如果节点类型为红黑树,则执行红黑树插入操作

5.如果节点类型为链表,for循环遍历链表直至链表尾部,然后进行尾插,当链表长度>=8时,会进入链表转红黑树的方法,treeifyBin方法中还会判断数组长度,数组长度>=64,链表长度>=8同时满足,才会将链表转为红黑树;在for循环遍历过程中,如果key相同,则直接插入元素

6.put方法是由返回值的,在插入完成后,如果插入之前已经存在这个key,则返回的是插入之前已存在元素的value

7.记录操作次数变量modCount+1,最后再判断当前map中有多少元素,和阈值做对比,如果超过阈值则进行扩容,没有则返回null

二、put方法源码

首先定义一个HashMap,向map中put一个元素

public class HashMapTest {
    public static void main(String[] args) {
        Map<String,String> map = new HashMap<>();
        map.put("k","v");
    }
}

按住ctrl进入put方法找到HashMap的pu方法

方法传入了hash值,key和value,然后返回了一个putVal方法,put的方法的核心就是putVal方法

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) {
    	//初始化变量--> 提高性能 ; p --> 新插入的元素
        Node<K,V>[] tab; Node<K,V> p; int n, i;
    	//判空,初始化数组
        if ((tab = table) == null || (n = tab.length) == 0)
            //resize() -->  扩容
            n = (tab = resize()).length;
    	//往map中放数据:先判空,如果为空,创建一个新的Node节点 
    	//tab[i = (n - 1) & hash] --> 计算数组下标
    	// 计算方式: 将元素hash值和数组长度-1做与操作
    	// i 就是 key值  范围要在 0 - (n-1)  之间 
        if ((p = tab[i = (n - 1) & hash]) == null)
            tab[i] = newNode(hash, key, value, null);
        else {//如果要插入的元素在这个位置有元素了,执行以下操作
            Node<K,V> e; K k;
            //比较两个元素的key是否相同
            //p --> 插入新元素之前该位置已经存在的元素
            // 比较 哈希值 : 引用地址  : key
            if (p.hash == hash &&
                ((k = p.key) == key || (key != null && key.equals(k))))
                //如果相等则老元素地址指向新元素
                e = p;
            //不相等则判断节点类型:树 TreeNode是Node的一个子类
            else if (p instanceof TreeNode)
                //红黑树的插入操作
                e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);
            //节点类型:链表
            else {
                //for循环:1.遍历到链表尾部,进行尾插
                //2.判断链表长度,超过8将链表改为红黑树
                for (int binCount = 0; ; ++binCount) {
                    //判断节点的下一个节点为空,遍历到链表尾部,进行尾插
                    if ((e = p.next) == null) {
                        //生成一个Node对象,将Node对象作为新节点插入到链表(p.next)
                        p.next = newNode(hash, key, value, null);
                 //如果链表长度 >= TREEIFY_THRESHOLD-1 = 7 因为是从0开始遍历,所以此时链表长度为8
                        if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st
                            //进入转红黑树的方法,treeifyBin方法中还会判断数组长度,数组长度>=64,链表长度>=8同时满足,才会将链表转为红黑树
                            treeifyBin(tab, hash);
                        break;
                    }
                    //key相同时同样插入返回
                    if (e.hash == hash &&
                        ((k = e.key) == key || (key != null && key.equals(k))))
                        break;
                    p = e;
                }
            }
            
            if (e != null) { // existing mapping for key
                //记录已经存在元素key的value
                V oldValue = e.value;
             //onlyIfAbsent put方法的参数,默认为false;如果调用热put方法putIfAbsent,默认为true
                if (!onlyIfAbsent || oldValue == null)
                    //新元素的值将老元素的值覆盖掉
                    e.value = value;
                //HashMap提供给子类的方法
                afterNodeAccess(e);
                //put操作有返回值,返回的是插入之前已经存在的元素的value值
                return oldValue;
            }
        }
    //增加修改次数
        ++modCount;
    	//统计当前map中有多少元素,和阈值对比,判断是否需要扩容
        if (++size > threshold)
            resize();
        afterNodeInsertion(evict);
        return null;
    }


HashMap扩容resize详解



HashMap扩容操作resize_crazyK.的博客-CSDN博客



版权声明:本文为weixin_57504474原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。