2021-10-15 HashMap的put方法

2021-10-15  本文已影响0人  归去来ming
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未初始化或者长度为0,进行扩容
    if ((tab = table) == null || (n = tab.length) == 0)
        n = (tab = resize()).length;
    /** 
    (n - 1) & hash 确定元素存放在哪个桶中,桶为空,新生成结点放入桶中(此时,这个结点是放在数组中)
    哈希值赋值给i,tab[i]赋值给p,即 p = tab[i = (n - 1) & hash],虽然是在if分支里,但p变量在方法内第一行,所以能在后续else中使用
    相当于在if之前赋值,if和else都能使用此赋值
    p = tab[i = (n - 1) & hash];
    if (p == null)
        tab[i] = newNode(hash, key, value, null);
    else {
        ......
    }
    */
    if ((p = tab[i = (n - 1) & hash]) == null)
        tab[i] = newNode(hash, key, value, null);
    // 桶中已经存在元素:hash值相同
    else {
        Node<K,V> e; K k;
        // 比较桶中第一个元素(数组中的结点), 如果key相等
        if (p.hash == hash &&
            ((k = p.key) == key || (key != null && key.equals(k))))
                // 将第一个元素赋值给e,用e来记录,后面将e进行旧值覆盖
                e = p;
        // 如果key不相等且是红黑树结点
        else if (p instanceof TreeNode)
            // 放入树中
            e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);
        // 如果key不相等,不是红黑树结点,那就是链表结点
        else {
            // 在链表最末插入结点
            for (int binCount = 0; ; ++binCount) {
                // 到达链表的尾部
                /*
                  等同于:
                  e = p.next;
                  if (e = null) {
                */
                if ((e = p.next) == null) {
                    // 在尾部插入新结点,虽然p.next现在不为null,但e仍是null
                    p.next = newNode(hash, key, value, null);
                    // 结点数量达到阈值,转化为红黑树
                    if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st
                        treeifyBin(tab, hash);
                    // 跳出循环
                    break;
                }
                // 没到达尾部:判断链表中结点的key值与插入的元素的key值是否相等
                if (e.hash == hash &&
                    ((k = e.key) == key || (key != null && key.equals(k))))
                    // 相等,跳出循环
                    break;

                /*
                用于遍历桶中的链表,与前面的e = p.next组合,可以遍历链表。
                没到达尾部,在链接中间某结点仍没找到key值相等的结点,就继续往后遍历
               */
                p = e;
            }
        }
        // 表示在桶中找到key值、hash值与插入元素相等的结点,进行旧值覆盖
        if (e != null) { 
            // 记录e的value
            V oldValue = e.value;
            // onlyIfAbsent为false或者旧值为null
            if (!onlyIfAbsent || oldValue == null)
                //用新值替换旧值
                e.value = value;
            // 访问后回调
            afterNodeAccess(e);
            // 返回旧值
            return oldValue;
        }
    }
    // 结构性修改
    ++modCount;
    // 实际大小大于阈值则扩容
    if (++size > threshold)
        resize();
    // 插入后回调
    afterNodeInsertion(evict);
    return null;
} 

有一个疑问:为什么if ((e = p.next) == null) {时new了一个node,e还是null?于是做了一个测试:

public static void main(String[] args){
        ListNode node1 = new ListNode(1);
        ListNode node2 = new ListNode(2);
        // 分开写,与下面注释代码写在if里面是一样的效果
        node1 = node2.next;
        if (node1 == null) {
            // 虽然为node2.next赋值了,不为空了,但是node1指向的是null,不是node2,所以node1仍为null
            node2.next = new ListNode(3);
        }
//        if ((node1 = node2.next) == null) {
//            node2.next = new ListNode(3);
//        }
        // 这句报空指针异常
        System.out.println(node1.val);
    }
image.png

所以if里面虽然为p.next赋值了,但是e此时指向的是null,不是一个非空对象引用,仍然是null,不会走下面 if (e != null)的逻辑

上一篇 下一篇

猜你喜欢

热点阅读