最全的HashMap源码解析!

HashMap源码解析

HashMap采用键值对形式的存储结构,每个key对应唯一的value,查询和修改的速度很快,能到到O(1)的平均复杂度。他是非线程安全的,且不能保证元素的存储顺序。

他的关系结构图

在这里插入图片描述

HashMap继承了AbstractMap,而AbstractMap的父类又是Map接口,所以HashMap也间接实现了Map接口,并且实现了Serializable接口,能被序列化,还实现了Cloneable接口可被克隆(浅拷贝)。

在Java8中HashMap采用了数组+链表+红黑树的数据结构。数组的每个元素又被称作桶,Node<key,value>。在添加元素的时候,会根据hash算法计算出不同的key对应的数组下标,注意这里我可没说计算出来的下标都是不同的,因为hash算法也有优略之分,不同的key可能对应同一个数组下标,这种现象被称为哈希冲突。为了解决哈希冲突带来带来的问题,所以在jdk1.8之前,当根据key计算出的数组下标上,已经有了元素,这个位置已经被占用了,这时候就把该元素放在此数组下标的链表的尾部。

但是一个链表的长度达到一定级别时,查询有时需要遍历整个链表,时间复杂度是O(n)。所以在jdk1.8后,当链表的长度大于等于8并且整个数组的大小大于等于64时,将这个链表转化成红黑树,红黑树的查询效率是O(logn)相比链表有提高了查询速率。

红黑树的特点

  • 节点是红色或是黑色
  • 根节点是黑色
  • 叶子结点是黑色(空节点)
  • 从任意节点到每个叶子结点的路径上包含相同个数的黑色节点
  • 每个红色节点的两个子节点都是黑色。也就是说从叶子结点到根节点的路径上不能出现两个连续的红节点。

存储结构

在这里插入图片描述

源码分析

重要属性


//默认初始容量是16
    static final int DEFAULT_INITIAL_CAPACITY = 1 << 4; // aka 16

  //最大容量是2的30次方
    static final int MAXIMUM_CAPACITY = 1 << 30;

    //当构造方法中没有指定负载因子时用这个默认的
    static final float DEFAULT_LOAD_FACTOR = 0.75f;

   //当链表中桶的个数大于等于8就会变成红黑树
    static final int TREEIFY_THRESHOLD = 8;

   //当红黑树大小小于等于6时红黑树会转化成链表
    static final int UNTREEIFY_THRESHOLD = 6;

   //当数组容量大于64时链表才能转成红黑树
    static final int MIN_TREEIFY_CAPACITY = 64;

    //数组,数组的大小是2的n次幂
    transient Node<K,V>[] table;
    //存放具体集合
    transient Set<Map.Entry<K,V>> entrySet;

   //存放元素的个数
    transient int size;

    //每次更改结构的计数器
    transient int modCount;

    //当HashMap所能容纳键值对数量的最大值,超过这个值,需要扩容
    int threshold;

   //负载因子
    final float loadFactor;

loadFactory加载因子
loadFactory加载因子是通知数组存放数据的疏密程度,他的值越接近于1,存放在数组到的数据也就越密集,也就是说这个值越大,链表的长度会增加的越快。当他的值趋近于0,存放在entry中的数据就会越少,也就越稀疏。

threshold
给定的默认容量大小是16,负载因子时0.75,Map在使用过程中不断的往里面存放数据,当存放的桶的个数达到了12=16*0.75,就会扩容,扩容的时候比较消耗性能。

你也许有疑问,12是怎算出来的,其实这个就是threshold属性,threshold=capacity*loadFactoy,当Size大于等于threshold这个值,就需要对数组进行扩容操作了。

从上面的源码我们可以知道,容量最大是2的30次方,当数组大小 大于等于64并且链表长度大于等于8时,链表就会转化成红黑树。当红黑树的桶的个数 小于等于6时就会变回链表。

链表Node节点

//Node 节点实现了Map.Entry<K,V>
static class Node<K,V> implements Map.Entry<K,V> {
        final int hash; //hashCod
        final K key;//键
        V value;//值
        Node<K,V> next;//指向下一个节点
        //构造函数
        Node(int hash, K key, V value, Node<K,V> next) {
            this.hash = hash;
            this.key = key;
            this.value = value;
            this.next = next;
        }

        public final K getKey()        { return key; }
        public final V getValue()      { return value; }
        //重写toString方法
        public final String toString() { return key + "=" + value; }
        //重写hashCode方法
        public final int hashCode() {
            return Objects.hashCode(key) ^ Objects.hashCode(value);
        }

        public final V setValue(V newValue) {
            V oldValue = value;
            value = newValue;
            return oldValue;
        }
        //重写equals方法
        public final boolean equals(Object o) {
            //如果内存地址一致直接返回true
            if (o == this)
                return true;
                //比较的节点必须实现这个Map.Entry
            if (o instanceof Map.Entry) {
                Map.Entry<?,?> e = (Map.Entry<?,?>)o;
                if (Objects.equals(key, e.getKey()) &&
                    Objects.equals(value, e.getValue()))
                    //键和值都相等才能返回true
                    return true;
            }
            return false;
        }
    }

红黑树节点结构

//TreeNode继承了LinkedHashMap中的Entry节点
static final class TreeNode<K,V> extends LinkedHashMap.Entry<K,V> {
        TreeNode<K,V> parent;  // 红黑树的父亲节点
        TreeNode<K,V> left; //左子树
        TreeNode<K,V> right;//右子树
        TreeNode<K,V> prev;    // 链表中的节点,在删除元素的时候可以快速找到他的前驱节点
        boolean red; //是否变色,
        //构造方法
        TreeNode(int hash, K key, V val, Node<K,V> next) {
            super(hash, key, val, next);
        }

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

构造方法

/**
* 构造一个空的HashMap并指定初始容量和负载因子。
* 
**/
public HashMap(int initialCapacity, float loadFactor) {
        //如果初始容量小于0,抛出非法参数异常
        if (initialCapacity < 0)
            throw new IllegalArgumentException("Illegal initial capacity: " +
                                               initialCapacity);
         //如果初始容量大于最大的容量也就是2^30,那么就按照最大的初始容量赋值。
        if (initialCapacity > MAXIMUM_CAPACITY)
            initialCapacity = MAXIMUM_CAPACITY;
         //如果负载因子小于0或者是NaN(float NaN = 0.0f / 0.0f;)也会抛出非法参数异常
        if (loadFactor <= 0 || Float.isNaN(loadFactor))
            throw new IllegalArgumentException("Illegal load factor: " +
                                               loadFactor);
        this.loadFactor = loadFactor;
        this.threshold = tableSizeFor(initialCapacity);
    }
//如果只是指定了初始容量那么负载因子就是默认的0.75
public HashMap(int initialCapacity) {
        this(initialCapacity, DEFAULT_LOAD_FACTOR);
    }

    /**
     * 如果什么都没指定,也就是无参构造,则初始容量是16和负载因子是0.75
     */
    public HashMap() {
        this.loadFactor = DEFAULT_LOAD_FACTOR; // all other fields defaulted
    }

    /**
     * 包含另一个Map的映射,如果被映射的Map是一个null会抛出空指针异常
     * 负载因子时默认的
     */
    public HashMap(Map<? extends K, ? extends V> m) {
        this.loadFactor = DEFAULT_LOAD_FACTOR;
        putMapEntries(m, false);
    }

final void putMapEntries(Map<? extends K, ? extends V> m, boolean evict) {
        int s = m.size();//s是map的大小
        if (s > 0) {
            if (table == null) { // pre-size
                float ft = ((float)s / loadFactor) + 1.0F;
                int t = ((ft < (float)MAXIMUM_CAPACITY) ?
                         (int)ft : MAXIMUM_CAPACITY);
                if (t > threshold)//如果t大于扩容的阀值就初始化阀值
                    threshold = tableSizeFor(t);
            }
            //如果这个map中的元素个数大于扩容的阀值就得扩容
            else if (s > threshold)
                resize();
                //将map中的key和value都添加到HashMap中
            for (Map.Entry<? extends K, ? extends V> e : m.entrySet()) {
                K key = e.getKey();
                V value = e.getValue();
                putVal(hash(key), key, value, false, evict);
            }
        }
    }



put方法

向HashMap中添加元素



    public V put(K key, V value) {
        return putVal(hash(key), key, value, false, true);
    }
    
    //扰动函数(哈希函数)
    static final int hash(Object key) {
        int h;
        //如果key是null,那么映射出来下标是0,否则将key的hashCode和h无符号右移16位做疑惑操作,使计算出的hash更加分散。
        return (key == null) ? 0 : (h = key.hashCode()) ^ (h >>> 16);
    }
    
//onlyifAbsent默认是false,表示即使key存在也可以覆盖旧值。
    final V putVal(int hash, K key, V value, boolean onlyIfAbsent,
                   boolean evict) {
                   //n表示数组的长度,i表示数组的下标,p表示下标 i对应的Node
        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;//e表示临时节点
            if (p.hash == hash &&  
                ((k = p.key) == key || (key != null && key.equals(k))))
                 // 如果key的hash和key本身的值都相等,直接把当前新增家电赋值给临时变量 e
                e = p;
                //这里开始判断使用哪种类型的数据结构开始添加节点,如果是红黑树,就用红黑树的新增方式
            else if (p instanceof TreeNode)
                e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value);
            else {
            //否则采用链表的方式新增节点,遍历到链表的末尾,追加到后面
                for (int binCount = 0; ; ++binCount) {
                   //p.next==null表示到链表尾部
                   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;
                //当onlyIfAbsent=false时才会覆盖
                if (!onlyIfAbsent || oldValue == null)
                    e.value = value;
                afterNodeAccess(e);
                //返回旧的值
                return oldValue;
            }
        }
        //结构修改计数器加一
        ++modCount;
        //如果实际大小 大于扩容的阀值就会再次扩容
        if (++size > threshold)
            resize();
            //插入后回调,具体实现交给了LinkedHashMap
        afterNodeInsertion(evict);
        return null;
    }

红黑树新增节点的过程


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;
            //找到根节点
            TreeNode<K,V> root = (parent != null) ? root() : this;
            for (TreeNode<K,V> p = root;;) {
                int dir, ph; K pk;
                //如果p的hash大于 h,那p就在h的右边
                if ((ph = p.hash) > h)
                    dir = -1;
                //如果p的hash小于 h,那p就在h的左边
                else if (ph < h)
                    dir = 1;
                //如果将要新增的key已经在树中存在了那就直接返回p,这里代表插入的key没有实现Comparable接口,那就通过equals方法和值 比较是不是同一个key
                else if ((pk = p.key) == k || (k != null && k.equals(pk)))
                    return p;
                    //如果key实现了Comparable的话,不再用hashcode比较,需要用compareTo方法
                else if ((kc == null &&
                //如果key没有实现Comparable接口,那么 comparableClassFor(k)返回的就是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)
                    //如果dir=-1 新节点放在左子树
                        xp.left = x;
                    else
                    //如果dir=1 新节点放在右子树
                        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));
             //balanceInsertion 对红黑树进行着色或旋转,以达到更多的查找效率,着色或旋转的几种场景如下
            //着色:新节点总是为红色;如果新节点的父亲是黑色,则不需要重新着色;如果父亲是红色,那么必须                 通过重新着色或者旋转的方法,再次达到红黑树的5个约束条件
            //旋转: 父亲是红色,叔叔是黑色时,进行旋转
            //如果当前节点是父亲的右节点,则进行左旋
            //如果当前节点是父亲的左节点,则进行右旋
          
            //moveRootToFront 方法是 把算出来的root放到根节点上
                    return null;
                }
            }
        }

  1. 首先判断新增的节点在红黑树上是不是已存在如果已存在就不在新增
    1. 如果节点没有实现Comparable接口,使用equals方法判断
    2. 如果节点已经实现Comparable接口,使用CompareTo判断
  2. 新增节点如果已经在红黑树上,直接返回;不在的话判断新增节点是在当前节点的左边还是右边,左边比当前值小,右边比当前值大。
  3. 递归前两步,知道当前节点的左子树或右子树为空时,停止递归,当前节点就是我们将要新增的父亲节点。
  4. 将新增节点放在当前节点的左边或是右边,与当前节点建立父子几点关系。
  5. 进行着色和旋转。

get方法



public V get(Object key) {
        Node<K,V> e;
        return (e = getNode(hash(key), key)) == null ? null : e.value;
    }

    /**
     * Implements Map.get and related methods
     *
     * @param hash hash for key
     * @param key the key
     * @return the node, or null if none
     */
    final Node<K,V> getNode(int hash, Object key) {
        Node<K,V>[] tab; Node<K,V> first, e; int n; K k;
        //如果桶数组不为空,并且桶的长度>0,才进行下面的操作,否则就返回null
        if ((tab = table) != null && (n = tab.length) > 0 &&
            (first = tab[(n - 1) & hash]) != null) {
            //总是先检查第一个节点,如果第一个是我们要找的key就直接返回
            if (first.hash == hash && // always check first node
                ((k = first.key) == key || (key != null && key.equals(k))))
                return first;
            if ((e = first.next) != null) {
            //如果first是TreeNode类型那就调用红黑树获取节点的方法
                if (first instanceof TreeNode)
                    return ((TreeNode<K,V>)first).getTreeNode(hash, key);
                //否则就是一链表的方式查找值
                do {
                //循环遍历,如果桶的hash和key与当前遍历的桶相等就返回。
                    if (e.hash == hash &&
                        ((k = e.key) == key || (key != null && key.equals(k))))
                        return e;
                } while ((e = e.next) != null);
            }
        }
        //以上都没找到就返回null
        return null;
    }

查找主要分为三个步骤

  1. 根据hash算法定位数组的索引位置,找到key及其第一个元素。
  2. 通过equals方法判断第一个节点是否是我们需要的key,是的话直接返回,不是的话,往后遍历
  3. 判断当前节点的next是不是null,如果不是的话,再判断属于哪个类型,如果是红黑树就采用遍历红黑树的方法查找节点,否则就以遍历链表的方式查找

resize方法

每次进行扩容,就会伴随着一次重新分配,并且会遍历hash表中的所有元素是非常耗时的。


final Node<K,V>[] resize() {
        Node<K,V>[] oldTab = table;
        //如果旧的数组是null那他的容量就是0
        int oldCap = (oldTab == null) ? 0 : oldTab.length;
       //旧的扩容阀值
       int oldThr = threshold;
        int newCap, newThr = 0;
        if (oldCap > 0) {
        //如果已超过最大值不在进行扩容 ,直接返回旧数组
            if (oldCap >= MAXIMUM_CAPACITY) {
                threshold = Integer.MAX_VALUE;
                return oldTab;
            }//没有超过最大值就扩容为原来的2倍,左移1位相当于乘2 
            //比如旧的容量是19,19>16,所以扩容为原来的两倍
            else if ((newCap = oldCap << 1) < MAXIMUM_CAPACITY &&
                     oldCap >= DEFAULT_INITIAL_CAPACITY)
                newThr = oldThr << 1; // double threshold
        }
        else if (oldThr > 0) // initial capacity was placed in threshold
        //
            newCap = oldThr;
        else {               // 调用默认构造方法创建的map第一次新增会走到这里进行判断
        //如果旧的数组的容量和扩容阀值都是0,就需要对新的map进行初始化,全部设置成默认值
            newCap = DEFAULT_INITIAL_CAPACITY;
            //threshold = loadFactor * capacity
            newThr = (int)(DEFAULT_LOAD_FACTOR * DEFAULT_INITIAL_CAPACITY);
        }
        if (newThr == 0) {
       //如果新map的扩容阀值等于0,需要计算出扩容阀值,计算出的 阀值肯定不能大于最大容量,即使大于,最大也是Integer.MAX_VALUE
            float ft = (float)newCap * loadFactor;
            newThr = (newCap < MAXIMUM_CAPACITY && ft < (float)MAXIMUM_CAPACITY ?
                      (int)ft : Integer.MAX_VALUE);
        }
        //进行赋值
        threshold = newThr;
        //这里又新建了一个桶数组
        @SuppressWarnings({"rawtypes","unchecked"})
            Node<K,V>[] newTab = (Node<K,V>[])new Node[newCap];
        table = newTab;
        //如果旧的数组不为空,开始依次搬运桶
        if (oldTab != null) {
            for (int j = 0; j < oldCap; ++j) {
                Node<K,V> e;//临时变量e
                if ((e = oldTab[j]) != null) {
                //如果当前桶不为空将它赋值给临时变量, 并且将旧的数组中对应的值置为null,便于GC
                    oldTab[j] = null;
                    //当第一个桶搬到新的数组中时他后面还没有其他桶
                    if (e.next == null)
                        newTab[e.hash & (newCap - 1)] = e;
                    else if (e instanceof TreeNode)
                    //如果搬运的是树节点,将树分割成两棵树放入新数组中
                        ((TreeNode<K,V>)e).split(this, newTab, j, oldCap);
                    else { // preserve order
                    //如果这个链表的长度>0且还没转换成树结构,那么扩容的时候就需要将这条链表转成两条链表
                        Node<K,V> loHead = null, loTail = null;
                        Node<K,V> hiHead = null, hiTail = null;
                        Node<K,V> next;
                        do {
                            next = e.next;
                            //放在低位链表中
                            if ((e.hash & oldCap) == 0) {
                                if (loTail == null)
                                    loHead = e;
                                else
                                    loTail.next = e;
                                loTail = e;
                            }
                            else {
                            //放在高位链表中
                                if (hiTail == null)
                                    hiHead = e;
                                else
                                    hiTail.next = e;
                                hiTail = e;
                            }
                        } while ((e = next) != null);
                        //低位链表的桶在新数组的位置和之前一样
                        if (loTail != null) {
                            loTail.next = null;
                            newTab[j] = loHead;
                        }
                        //高位链表在新数组的位置时原来的位置+oldCap
                        if (hiTail != null) {
                            hiTail.next = null;
                            newTab[j + oldCap] = hiHead;
                        }
                    }
                }
            }
        }
        return newTab;
    }

总结

  • HashMap在java8中的存储结构是: 数组+链表+红黑树
  • HashMap默认初始容量是 16,默认负载因子时0.75f,容量总是2的n次方,最大容量是2的30次方
  • HashMap扩容时每次容量变为原来的两倍
  • 当链表长度大于等于8,但是桶的数量(数组的长度)小于64时,不会变成红黑树只会进行扩容
  • 当红黑树的大小小于6时,会转变成链表
  • HashMap是线程不安全的。在多线程使用场景中,应该尽量使用线程安全的ConcurrentHashMap替换HashMap,HashMap允许null做key。这样的key只能有一个但是null作为value可以有多个。
  • HashMap查找和添加元素的时间复杂度都是O(1)
原文地址:https://www.cnblogs.com/dataoblogs/p/14121931.html