在java jdk8中对HashMap的源码进行了优化,在jdk7中,HashMap处理“碰撞”的时候,都是采用链表来存储,当碰撞的结点很多时,查询时间是O(n)。
在jdk8中,HashMap处理“碰撞”增加了红黑树这种数据结构,当碰撞结点较少时,采用链表存储,当较大时,采用红黑树(特点是查询时间是O(logn))存储(有一个阀值控制,大于阀值,将链表存储转换成红黑树存储)
HashMap的样子就变成下图的样子:
下面对源码进行分析:
public class HashMap<K,V> extends AbstractMap<K,V> implements Map<K,V>, Cloneable, Serializable
由HashMap的类声明中,可知继承自AbstractMap抽象类,实现Map、Cloneable、Serializable接口。
public abstract class AbstractMap<K,V> implements Map<K,V>
AbstractMap抽象类实现Map接口。
观察常量:
static final int DEFAULT_INITIAL_CAPACITY = 1 << 4;
默认table初始容量16
static final int MAXIMUM_CAPACITY = 1 << 30;
table最大容量2的30次方
static final float DEFAULT_LOAD_FACTOR = 0.75f;
默认负载因子0.75
static final int TREEIFY_THRESHOLD = 8;
结点冲突数达到8时,就会对hash表进行调整,如果table容量小于64,那么会进行table扩容,如果不小于64,那么会将冲突数达8那个单链表调整为红黑树
static final int UNTREEIFY_THRESHOLD = 6;
如果原先是红黑树的,resize后冲突结点数少于6了,就把红黑色恢复成单链表
static final int MIN_TREEIFY_CAPACITY = 64;
如果table的容量少于64,那么即使冲突结点数达到TREEIFY_THRESHOLD后不会把该单链表调整成红黑数,而是将table扩容
Node静态类是HashMap中单链表的结点:
不难理解,就不详细讲了
static class Node<K,V> implements Map.Entry<K,V> { final int hash; 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; } public final String toString() { return key + "=" + value; } public final int hashCode() { return Objects.hashCode(key) ^ Objects.hashCode(value); } public final V setValue(V newValue) { V oldValue = value; value = newValue; return oldValue; } public final boolean equals(Object o) { if (o == this) return true; if (o instanceof Map.Entry) { Map.Entry<?,?> e = (Map.Entry<?,?>)o; if (Objects.equals(key, e.getKey()) && Objects.equals(value, e.getValue())) return true; } return false; } }
四个构造方法:
public HashMap(int initialCapacity, float loadFactor)
指定初始容量和负载因子
public HashMap(int initialCapacity)
指定初始容量,使用默认负载因子0.75
public HashMap()
使用默认初始容量16,默认负载因子0.75
public HashMap(Map<? extends K, ? extends V> m)
将一对Key-Value加入HashMap中
public V put(K key, V value) {
//第四个参数是onlyIfAbsent,表示当找到同key的键值对,是否更新value的值,false是更 //新,第五个参数是evict,在afterNodeInsertion方法中使用,但这个函数是空函数 return putVal(hash(key), key, value, false, true); }
实际调用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;
//如果table为空或者未分配空间,则resize,放入第一个K-V时总是先resize if ((tab = table) == null || (n = tab.length) == 0) n = (tab = resize()).length;
//(n-1)&hash计算K-V存的table位置,如果首节点为null,代表该位置还没放入过结点 if ((p = tab[i = (n - 1) & hash]) == null)
//调用newNode新建一个结点放入该位置 tab[i] = newNode(hash, key, value, null); else {//到这里,表示有冲突,开始处理冲突 Node<K,V> e; K k;
//这时候p是指向table[i]的首个Node,判断table[i]是否与待插入节点有相同的
//hash,key值,如果是则e=p
if (p.hash == hash && ((k = p.key) == key || (key != null && key.equals(k)))) e = p;
//到这里说明table[i]的第一个Node与待插入Node的hash或key不同,那么要在
//这个节点之后的链表节点或者树节点中查找
else if (p instanceof TreeNode)//如果之后是树节点,使用树节点的插入方法
//插入成功后e等于null e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value); else {//说明之后是链表节点 for (int binCount = 0; ; ++binCount) {//逐个向后查找
if ((e = p.next) == null) {//如果下一个节点是null,表示没有找到
//同hash或同key的节点 p.next = newNode(hash, key, value, null);//新建一个节点
//放在链表的最后
//如果冲突的节点数已经达到8个,看是否需要改变冲突节点的存储结构,
//treeifyBin首先判断当前hashMap的长度,如果不足64,只进行
//resize,扩容table,如果达到64,那么将冲突的存储结构为红黑树
if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st treeifyBin(tab, hash); break; }//如果找到同hash或同key的节点,那么直接退出循环,
//此时e等于冲突Node if (e.hash == hash && ((k = e.key) == key || (key != null && key.equals(k)))) break; p = e;//调整p节点,以继续查找 } }
//退出循环后,先判断e是否为null,为null表示已经插入成功,不为null表示有冲突 if (e != null) { // existing mapping for key V oldValue = e.value;//保存旧冲突点的value if (!onlyIfAbsent || oldValue == null)
//判断是否需要修改value的值,默认是修改,如果旧冲突点value是null,一定
//是修改的 e.value = value; afterNodeAccess(e);//空函数 return oldValue; } } ++modCount;//当插入了新节点才会运行到这里,HashMap结构调整次数+1 if (++size > threshold)//如果HashMap插入新节点后大于threshold,进行扩容 resize(); afterNodeInsertion(evict);//空函数 return null; }
resize方法
final Node<K,V>[] resize() { Node<K,V>[] oldTab = table; //将原hash表赋值给oldTab临时变量 int oldCap = (oldTab == null) ? 0 : oldTab.length;//原hash表容量赋值给
//oldCap,如果第一次resize,oldCap为0 int oldThr = threshold;//原扩容阀值赋给oldThr临时变量 int newCap, newThr = 0; if (oldCap > 0) {//如果不是第一次resize if (oldCap >= MAXIMUM_CAPACITY) {//如果原table容量已经达到最大值 threshold = Integer.MAX_VALUE; return oldTab;//无法继续扩容,只能返回原table } else if ((newCap = oldCap << 1) < MAXIMUM_CAPACITY && oldCap >= DEFAULT_INITIAL_CAPACITY)//扩容将原容量*2 newThr = oldThr << 1; // double threshold,新扩容阀值*2 }
//原容量为0,但原阀值不为0,那么新容量为原阀值 else if (oldThr > 0) // initial capacity was placed in threshold newCap = oldThr;
//如果原容量和原阀值都为0,用默认值进行初始化 else { // zero initial threshold signifies using defaults newCap = DEFAULT_INITIAL_CAPACITY; newThr = (int)(DEFAULT_LOAD_FACTOR * DEFAULT_INITIAL_CAPACITY); } if (newThr == 0) {//如果新阀值为0 float ft = (float)newCap * loadFactor;//临时变量存储计算出的新阀值 newThr = (newCap < MAXIMUM_CAPACITY && ft < (float)MAXIMUM_CAPACITY ? (int)ft : Integer.MAX_VALUE);//重新设置新阀值 } threshold = newThr;//阀值被重新赋值 @SuppressWarnings({"rawtypes","unchecked"})
//使用新容量创建一个新hash表 Node<K,V>[] newTab = (Node<K,V>[])new Node[newCap]; table = newTab;//旧hash表被新创建的表替换 if (oldTab != null) {//如果旧hash表不为null for (int j = 0; j < oldCap; ++j) {//遍历旧hash表,目的是重新分配所有
//结点到新hash表中 Node<K,V> e; if ((e = oldTab[j]) != null) {//如果oldTab[j]不为null,取出第一个 //结点赋值给e oldTab[j] = null;//oldTab[j]赋值为null if (e.next == null)//如果只有一个结点,表示之前没有碰撞 newTab[e.hash & (newCap - 1)] = e;//把该结点存入新表 else if (e instanceof TreeNode)//如果e是树结点,表示要移动一棵 //树
//调整这棵树 ((TreeNode<K,V>)e).split(this, newTab, j, oldCap); else { //到这里表示e后面带着个单链表,需要遍历单链表,将每个结点重 //新计算在新表的位置,并进行搬运 Node<K,V> loHead = null, loTail = null; Node<K,V> hiHead = null, hiTail = null; Node<K,V> next; do { next = e.next;//记录下一个结点
//新表是旧表的两倍容量,实例上就把单链表拆分为两队,
//e.hash&oldCap为偶数一队,e.hash&oldCap为奇数一对
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) {//lo队不为null,放在新表原位置 loTail.next = null; newTab[j] = loHead; } if (hiTail != null) {//hi队不为null,放在新表j+oldCap位置 hiTail.next = null; newTab[j + oldCap] = hiHead; } } } } } return newTab;//返回调整好的新表 }
treeifBin方法是冲突结点达到8个,判断是要进行扩容还是链表转换成红黑色
final void treeifyBin(Node<K,V>[] tab, int hash) { int n, index; Node<K,V> e;
//table长度还未达到64,仅对table resize扩容 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树结点,并构建双向链表,为构建红黑 //树作准备 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);//转换成红黑树 } }
如果想了解Java 8 的HashMap对比Java 7的HashMap,性能提升,请看http://it.deepinmind.com/%E6%80%A7%E8%83%BD/2014/04/24/hashmap-performance-in-java-8.html