阅读博客
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碰撞有没有那么频繁呢?如果碰撞真的这么厉害,这也算是一种优化吧。