王剑编程网

分享专业编程知识与实战技巧

Java源码解析篇:HashMap

相信大家对HashMap的底层原理或多或少都有了解,这里简单提一下,jdk1.8版本之后HashMap内部主要通过数组+链表+红黑树的数据结构来存放元素,本文我们来分析HashMap的几个主要方法,看看HashMap的源码具体是如何实现的,以及jdk做了哪些优化来提升性能。

Node节点

static class Node<K,V> implements Map.Entry<K,V> {
    //hash值
    final int hash;
    //元素的key
    final K key;
    //元素的值
    V value;
    //指向下一个节点
    Node<K,V> next;

    //此处省略其他代码。。。
}
复制代码

Node类是以HashMap的内部类形式存在,代表放入HashMap的元素,字段含义都比较好理解,这里的next字段表示一个指针,指向下一个节点,等会在put方法里分析这个字段的使用场景

put方法

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

put方法直接调用了putVal方法,可以看到第一个参数传入了hash(key),那我们先来看一下hash方法

static final int hash(Object key) {
    int h;
    return (key == null) ? 0 : (h = key.hashCode()) ^ (h >>> 16);
}
复制代码

这个hash方法就是用来计算放入map方法的元素的hash值,重点是这段代码(h = key.hashCode()) ^ (h >>> 16),先通过key.hashCode()获取key的hash值,然后再异或(h >>> 16)计算出最终的hash值,h>>>16表示将hash值右移16位

如果对右移>>>和异或^运算不熟悉的同学,看这行代码可能会比较疑惑,举个栗子:

h值:
1111 1111 1111 1111 1111 1010 0111 1100

h>>>16值:
0000 0000 0000 0000 1111 1111 1111 1111

h ^ (h >>> 16)值:
1111 1111 1111 1111 0000 0101 1000 0011

最终通过h ^ (h >>> 16)计算出来的值才是hash方法获取到的hash值,那么这里为什么要先右移再异或来获取hash值,而不是直接返回hashCode方法拿到的hash值呢?一句话解释就是:为了降低hash冲突,至于为什么能降低hash冲突,我们在后面讲定位元素在数组上的位置的时候再来解释,这里只要知道通过异或操作能将原始hash的高低16位的特征都保留下来,反应在最终的hash值上

hash方法分析完了之后,我们来看下putVal方法,这个方法比较长,里面分支逻辑也比较复杂,我们一点一点来看

 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.数组为空,先调用resize方法生成一个数组
        if ((tab = table) == null || (n = tab.length) == 0)
            n = (tab = resize()).length;
        //2.先定位元素在数组中的位置,如果当前位置元素为空,构建一个Node放入当前位置
        if ((p = tab[i = (n - 1) & hash]) == null)
            tab[i] = newNode(hash, key, value, null);
        else {
            Node<K,V> e; K k;
            //3.当前位置已经存在元素且key相同,直接覆盖当前元素
            if (p.hash == hash &&
                ((k = p.key) == key || (key != null && key.equals(k))))
                e = p;
            //4.如果当前元素是一个树节点,将新元素挂到树上
            else if (p instanceof TreeNode)
                e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);
            //5.将元素挂到链表上
            else {
                //6.遍历链表元素
                for (int binCount = 0; ; ++binCount) {
                    if ((e = p.next) == null) {
                        //7.构建一个节点,这里就是用next指针来链接节点
                        p.next = newNode(hash, key, value, null);
                        //8.如果链表上的元素超过8个,转成红黑树
                        if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st
                            treeifyBin(tab, hash);
                        break;
                    }
                    //9.如果当前元素和新元素key相同,直接覆盖链表上的节点
                    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;
    }
复制代码

上面的代码中,我在各个重要代码上加了注释,应该很容易理解,我们来看下几个比较重要的点:

1.定位元素位置:p = tab[i = (n - 1) & hash]

还记得前面留下的问题吗,为什么hash方法要移位和异或操作,这里就来解释清楚,在获取到hash值之后,(n - 1) & hash将hash和数组长度减1进行与操作定位元素在数组上的位置,n在代码里保证了是2的幂次方,比如16、32、64等等,看完这个方法也就明白了为什么要是2的幂次方,同样,我们来举个栗子看看这个操作:

n值假设是16,那么n-1就是15,二进制值为:1111

hash值就用上面的1111 1111 1111 1111 0000 0101 1000 0011,两者取与,结果是11,所以这个元素就会落在索引为11的位置上,这里用与操作而不是对数组取模,主要还是因为性能上的考虑。

那么为什么要用异或操作呢,我们假定前面那个元素叫a,原始hash值是1111 1111 1111 1111 1111 1010 0111 1100,还有另一个元素b的原始hash值是1111 1111 1111 1110 1111 1010 0111 1100,注意和之前元素的区别,它们只有第16位不一样,a是1,b是0

如果没有进行移位和异或操作,那么它俩进行(n - 1) & hash计算之后值是一样的,都是16,也就是会产生冲突,我们再来看看b经过移位和异或操作之后的hash值是什么

b的h值:
1111 1111 1111 1110 1111 1010 0111 1100

h>>>16值:
0000 0000 0000 0000 1111 1111 1111 1110

h ^ (h >>> 16)值:
1111 1111 1111 1111 0000 0101 1000 0010

所以b最终的hash值为1111 1111 1111 1111 0000 0101 1000 0010,求与(n - 1) & hash之后就是10,也就是定位到了索引为10的数组位置,和a不冲突了,感受到了么,这就是移位和异或的作用,降低元素在数组位置上冲突的概率。

2.链表转红黑树: treeifyBin(tab, hash); 这个方法简单来说就是将链表转成了一颗红黑树,内部实现还是很复杂的,有兴趣的可以自己去看源码。

resize方法

Node<K,V> loHead = null, loTail = null;
Node<K,V> hiHead = null, hiTail = null;
Node<K,V> next;
do {
    next = e.next;
    //1.hash值&老数组长度等于0
    if ((e.hash & oldCap) == 0) {
        if (loTail == null)
            loHead = e;
        else
            loTail.next = e;
        loTail = e;
    }
    //2.hash值&老数组长度不等于0
    else {
        if (hiTail == null)
            hiHead = e;
        else
            hiTail.next = e;
        hiTail = e;
    }
} while ((e = next) != null);
if (loTail != null) {
    loTail.next = null;
    //3.放到新数组的原位置
    newTab[j] = loHead;
}
if (hiTail != null) {
    hiTail.next = null;
    //4.放到(老数组长度+原位置)的新位置
    newTab[j + oldCap] = hiHead;
}
复制代码

HashMap扩容的时候会先生成一个容量为老数组2倍的新数组,然后将老元素拷贝到新数组上,上面代码是从resize方法里摘出来的确定扩容之后新元素位置的代码,注释1那里通过元素hash值和原数组长度取余是否为0就可以快速确定放在新数组的位置,如果为0,就放在老位置,如果不为0就放在新位置,新位置通过老数组长度+原位置来确定,同样,我们来举个栗子理解一下:

假设,hash为1111 1111 1111 1111 0000 0101 1000 0011,数组从16扩容到32,通过e.hash & oldCap计算值为0,所以在新旧数组上的位置都是11

另有hash为1111 1111 1111 1111 0000 0101 1001 0011的元素,数组从16扩容到32,通过e.hash & oldCap计算值为1,所以在新数组上的位置就变成了11 + 16 = 27,不再是原来是的11

总结

本文主要讲述了HashMap以下几个实现:
1.计算hash值(移位、异或操作)
2.寻址算法优化(&操作代替取模)
3.put()方法实现(挂链表以及链表转红黑树)
4.resize方法实现(rehash定位新数组中的位置)

以上,希望对你理解HashMap有帮助,有不同看法或疑问欢迎随时和我讨论


作者:haixiang
链接:
https://juejin.cn/post/6986447578111934500

来源:掘金
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

控制面板
您好,欢迎到访网站!
  查看权限
网站分类
最新留言