阅读博客
1,
java提高篇(二三)-----HashMap
这一篇由chenssy发表于2014年1月,是根据JDK1.6的源码讲的。
2,Java类集框架之HashMap(JDK1.8)源码剖析
这一篇由push_pop发表于2015年5月,根据JDK1.8讲的。
先说1.6的HashMap
1.6的HashMap代码较少,写的比较容易看懂。
HashMap里存的对象是Entry,HashMap实际上是Entry数组。数组的大小是2的n次方,默认为16,对应属性是capacity。
HashMap有4个构造函数,但其实最终调用的都是HashMap(int initialCapacity,float loadFactor)这个。可以看到,构造HashMap需要两个参数 容量initialCapacity和负载因子loadFactor。
initialCapacity这个参数并不是说 HashMap能装多少组元素,而是HashMap里有多少个桶,也就是Entry数组的大小。它并不会总使用你设定的值,而是会取一个比initialCapacity大的2的n次方。
比如你设置initialCapacity=9,实际上桶的大小会是16。桶大小不是一成不变的,会根据HashMap中总元素的增加而成倍的增加。比如现在capacity是16,下一次调整就是32,再增加就是64.
HashMap在构造时,会设置扩容的阈值shreshold=initialCapacity*loadFactor。
下面说常用的put方法。1.6源码
1 public V put(K key, V value) { 2 if (key == null) 3 return putForNullKey(value); 4 int hash = hash(key.hashCode()); 5 int i = indexFor(hash, table.length); 6 for (Entry<K,V> e = table[i]; e != null; e = e.next) { 7 Object k; 8 if (e.hash == hash && ((k = e.key) == key || key.equals(k))) { 9 V oldValue = e.value; 10 e.value = value; 11 e.recordAccess(this); 12 return oldValue; 13 } 14 } 15 16 modCount++; 17 addEntry(hash, key, value, i); 18 return null; 19 }
1.6根据key是否为null,会走不同的分支,key==null的,会单独走一个方法putForNullKey(). null这个key会固定放到table[0]
hash值,等于key.hashCode再进入一个hash函数算出来的值,这个hash()函数做了几次移位和取或运算。
indexFor() 等于用hash对table的长度取模,来得到将来放到数组的哪个位置。这里用按位与,速度比%要快多。
put函数是有返回值的。
这时会有三种情况
- 情况1,i位置上没有元素,那么for循环不会进去,直接新建一个Entry节点,放在i位置上,返回null。
- 情况2,i位置上已经有元素了,而且是一列元素,链表存储方式,但是都是不同的key,那么,就新建一个Entry节点,放在i的位置,之前这里的链表挂在新建的节点后面,或者说,新建的节点指向以前挂在这里的链表,返回null
这里有2点注意,
1,新建的节点,是挂在原来列表的头上,而不是末尾。这样就省得遍历到末尾了。
2,如何判断key相同呢?
e.hash == hash && ((k = e.key) == key || key.equals(k))
key的hash值相等,且key的equals方法相等,或者根本就是一个key(引用相同),这也是为什么一个对象要做Key必须要重载hashCode()和equals()方法的原因吧。
- 情况3,其实前面两种情况都是一个新key加入,只不过1是没有发生hash碰撞,2是发生碰撞了。 情况3是说 对一个已经存在key,再次put会新值怎么样。首先新值覆盖旧值,然后返回旧值。
所以通过put返回值,可以判断put的是一个新key,还是旧key。
get方法
1.6源码
public V get(Object key) { if (key == null) return getForNullKey(); int hash = hash(key.hashCode()); for (Entry<K,V> e = table[indexFor(hash, table.length)]; e != null; e = e.next) { Object k; if (e.hash == hash && ((k = e.key) == key || key.equals(k))) return e.value; } return null; }
如果key==null会单独走一个方法。
其他的 先算出hash值,找到对应的桶,遍历。
如果有相同的key就返回对应的value ,否则返回null。
get比较简单。
size属性,记录HashMap中有多少元素(Entry)。每次添加Entry时,size都会自增1,同时会和阈值shreshold比较,当size>=shreshold时,就是调用resize方法 扩容
void addEntry(int hash, K key, V value, int bucketIndex) { Entry<K,V> e = table[bucketIndex]; table[bucketIndex] = new Entry<K,V>(hash, key, value, e); if (size++ >= threshold) resize(2 * table.length); }
resize方法
void resize(int newCapacity) { Entry[] oldTable = table; int oldCapacity = oldTable.length; if (oldCapacity == MAXIMUM_CAPACITY) { threshold = Integer.MAX_VALUE; return; } Entry[] newTable = new Entry[newCapacity]; transfer(newTable); table = newTable; threshold = (int)(newCapacity * loadFactor); }
每次容量扩成之前的2倍
先new一个新数组,
再通过transfer方法把之前的元素都重新hash到新的数组中,
阈值按照新的容量重新计算。
remove()
还有删除的方法,删除也有返回值,如果返回null,说明要删除的key不存在,如果不是null,就是删除元素的value。
可以看到1.6的HashMap采用的是数组形式,哈希值取模后就是数组的索引,所以在不冲突时,时间复杂度为1,当冲突时,多个节点一个接一个形成一个链结构,后来者插入到链表头。
===分割线=======================================
上面是jdk1.6的HashMap源代码
下面聊聊jdk1.8的HashMap,这个版本的HashMap变化还是比较大的,整个源文件从1000行增加到了2300行(包含所有注释空行)。
第一个改变是把以前的Entry类 改名字 成了Node。
Entry和Node都实现了Map.Entry接口。 这个接口有4个static的方法,不需要实现。
在Java8中,接口中支持了静态方法,这个静态方法需要有body,不需要实现。
HashMap还是有4个构造函数,最主要的是HashMap(int,float) , 1.8中,不会在构造时就new table数组,而是会延迟到第一次put数据时。构造函数只设定了loadFactor和threshold参数,threshold的计算方法和之前不同,它调用了一个tableSizeFor的方法,这个方法返回大于initialCapacity的2的n次幂
/** * Returns a power of two size for the given target capacity. */ static final int tableSizeFor(int cap) { int n = cap - 1; n |= n >>> 1; n |= n >>> 2; n |= n >>> 4; n |= n >>> 8; n |= n >>> 16; return (n < 0) ? 1 : (n >= MAXIMUM_CAPACITY) ? MAXIMUM_CAPACITY : n + 1; }
hash值,依然是对 key的hashCode进行再hash, 不过hash函数和1.6的又有不同。
put方法
1.8的put实际上是调用putVal方法,这个方法多了一个onlyIfAbsent参数,貌似考虑在将来在put的时候,如果该key已经存在,就不覆盖了。目前这个参数还是false,每次put如果key存在,会用新值替换旧值。
1.8源码,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; 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; if (p.hash == hash && ((k = p.key) == key || (key != null && key.equals(k)))) e = p; else if (p instanceof TreeNode) e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value); else { for (int binCount = 0; ; ++binCount) { 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; if (!onlyIfAbsent || oldValue == null) e.value = value; afterNodeAccess(e); return oldValue; } } ++modCount; if (++size > threshold) resize(); afterNodeInsertion(evict); return null; }
1.8的put比1.6的看起来复杂多了。代码写的更紧凑,不容易看懂,许多赋值和比较都在一个语句中写着。可能是为了执行效率吧,阅读起来体验下降。
新版的resize方法是一个自适应的方法,不需要参数,这个resize比以前复杂,有初始化table的任务。
没有对key=null去特殊处理。
情况1,如果table位上没有节点,new一个新节点挂上即可,返回null。
情况2,table[i]为上有节点,且第一个节点的key就是本次要put的key,那么用新值覆盖旧值,返回旧值。
后面就是要遍历这个位置上的挂接节点了。1.8的这些节点有两种组织方式 ,对于7个一下的,还是链表的结构。对于7个以上的,会转化成红黑树的结构来存储。
情况3,后面节点是树的结构。那么就调用TreeNode的put方法,有存在key,就替换,返回旧值,不存在就新建节点,返回null。
情况4,后面节点是链表的结构。如果遍历这些节点,存在相同的key,那么同情况2,新值覆盖旧值,返回旧值。如果遍历完,没有存在key,就新建节点,这个节点是挂在链表尾,返回null。1.6是挂在链表头。然后,如果链表长度大于阈值7,那么需要将这个链表结构,转化成树的结构。
1.8的HashMap新增一个内部类TreeNode,是继承LinkHashMap.Entry而来。当一个桶内的元素数量大于阈值(TREEIFY_THRESHOLD=8),会转化为树形,小于阈值(
UNTREEIFY_THRESHOLD)又会转换成链表结构。 这对于大量hash碰撞的情况有查找的优化,对于hash碰撞较小时,则不会用到TreeNode结构。
get()方法
1.8源码
public V get(Object key) { Node<K,V> e; return (e = getNode(hash(key), key)) == null ? null : e.value; } final Node<K,V> getNode(int hash, Object key) { Node<K,V>[] tab; Node<K,V> first, e; int n; K k; if ((tab = table) != null && (n = tab.length) > 0 && (first = tab[(n - 1) & hash]) != null) { if (first.hash == hash && // always check first node ((k = first.key) == key || (key != null && key.equals(k)))) return first; if ((e = first.next) != null) { if (first instanceof TreeNode) return ((TreeNode<K,V>)first).getTreeNode(hash, key); do { if (e.hash == hash && ((k = e.key) == key || (key != null && key.equals(k)))) return e; } while ((e = e.next) != null); } } return null; }
get方法,如果找到了key,则返回value,否则,返回null。
会根据节点结构是红黑树,还是链表用不用的方式去遍历。
put的函数名是putVal,而get的函数名是getNode,这里不统一,有点奇怪。
remove()
删除方法,如果找到了key,返回删除的value,否则返回null。
根据node的结构,选择在树中遍历,还是以链表的方式遍历。
1.8源码
final Node<K,V> removeNode(int hash, Object key, Object value, boolean matchValue, boolean movable) { Node<K,V>[] tab; Node<K,V> p; int n, index;
//如果桶内没元素,直接返回null if ((tab = table) != null && (n = tab.length) > 0 && (p = tab[index = (n - 1) & hash]) != null) { Node<K,V> node = null, e; K k; V v;
//第一个节点就找到了key if (p.hash == hash && ((k = p.key) == key || (key != null && key.equals(k)))) node = p; else if ((e = p.next) != null) {
//后面是TreeNode结构 if (p instanceof TreeNode) node = ((TreeNode<K,V>)p).getTreeNode(hash, key); else {
//后面是链表结构 do { if (e.hash == hash && ((k = e.key) == key || (key != null && key.equals(k)))) { node = e; break; } p = e; } while ((e = e.next) != null); } }
if (node != null && (!matchValue || (v = node.value) == value || (value != null && value.equals(v)))) {
//匹配到了key,根据是链表还是树,用不同方式删除。 if (node instanceof TreeNode) ((TreeNode<K,V>)node).removeTreeNode(this, tab, movable); else if (node == p) tab[index] = node.next; else p.next = node.next; ++modCount; --size; afterNodeRemoval(node); return node; } } return null; }
1.8中增加的代码主要都是因为引入了红黑树而引起的。导致许多地方都要考虑两种情况。
HashMap碰撞有没有那么频繁呢?如果碰撞真的这么厉害,这也算是一种优化吧。