HashMap 是 Java 中 Map 的一個實現類,它是一個雙列結構(數據+鏈表),這樣的結構使得它的查詢和插入效率都很高。HashMap 容許 null 鍵和值,它的鍵惟一,元素的存儲無序,而且它是線程不安全的。html
因爲 HashMap 的這些特性,它在 Java 中被普遍地使用,下面咱們就基於 Java 8 分析一下 HashMap 的源碼。前端
首先 HashMap 是一個雙列結構,它是一個散列表,存儲方式是鍵值對。 它繼承了 AbstractMap,實現了 Map<K,V> Cloneable Serializable 接口。node
HashMap 的雙列結構是數組 Node[]+鏈表,咱們知道數組的查詢很快,可是修改很慢,由於數組定長,因此添加或者減小元素都會致使數組擴容。而鏈表結構偏偏相反,它的查詢慢,由於沒有索引,須要遍歷鏈表查詢。可是它的修改很快,不須要擴容,只須要在首或者尾部添加便可。HashMap 正是應用了這兩種數據結構,以此來保證它的查詢和修改都有很高的效率。算法
HashMap 在調用 put() 方法存儲元素的時候,會根據 key 的 hash 值來計算它的索引,這個索引有什麼用呢?HashMap 使用這個索引來將這個鍵值對儲存到對應的數組位置,好比若是計算出來的索引是 n,則它將存儲在 Node[n] 這個位置。數據庫
HashMap 在計算索引的時候儘可能保證它的離散,但仍是會有不一樣的 key 計算出來的索引是同樣的,那麼第二次 put 的時候,key 就會產生衝突。HashMap 用鏈表的結構解決這個問題,當 HashMap 發現當前的索引下已經有不爲 null 的 Node 存在時,會在這個 Node 後面添加新元素,同一索引下的元素就組成了鏈表結構,Node 和 Node 之間如何聯繫能夠看下面 Node 類的源碼分析。編程
先了解一下 HashMap 裏數組的幾個參數:數組
DEFAULT_INITIAL_CAPACITY,默認初始長度,16:安全
static final int DEFAULT_INITIAL_CAPACITY = 1 << 4; // aka 16
MAXIMUM_CAPACITY,最大長度,2^30:網絡
static final int MAXIMUM_CAPACITY = 1 << 30;
DEFAULT_LOAD_FACTOR,默認加載因子,0.75:數據結構
static final float DEFAULT_LOAD_FACTOR = 0.75f; final float loadFactor;
threshold,閾值,擴容的臨界值(capacity * load factor)
int threshold;
再看看 HashMap 構造函數
// 初始長度 加載因子 public HashMap(int initialCapacity, float loadFactor) public HashMap(int initialCapacity) // 無參構造 public HashMap() // 初始化一個Map public HashMap(Map<? extends K, ? extends V> m)
下邊是很是重要的一個內部類 Node ,它實現了 Map.Entry,Node 是 HashMap 中的基本元素,每一個鍵值對都儲存在一個 Node 對象裏, Node 類有四個成員變量:hash key 的哈希值、鍵值對 key 與 value,以及 next 指針。next 也是 Node 類型,這個 Node 指向的是鏈表下一個鍵值對,這也就是前文提到的 hash 衝突時 HashMap 的處理辦法。
Node 類內部實現了 Map.Entry 接口中的 getKey()、getValue() 等方法,因此在遍歷 Map 的時候咱們能夠用 Map.entrySet() 。
static class Node<K,V> implements Map.Entry<K,V> { final int hash; // 哈希值 final K key; V value; // 鏈表結構, 這裏的next將指向鏈表的下一個Node鍵值對 Node<K,V> next; Node(int hash, K key, V value, Node<K,V> next) { ... } public final K getKey() { return key; } public final V getValue() { return value; } }
put() 方法
put() 主要是將 key 和 value 保存到 Node 數組中,HashMap 根據 key 的 hash 值來肯定它的索引,源碼裏 put 方法將調用內部的 putVal() 方法。
public V put(K key, V value) { return putVal(hash(key), key, value, false, true); }
HashMap 在 put 鍵值對的時候會調用 hash() 方法計算 key 的 hash 值,hash() 方法會調用 Object 的 native 方法 hashCode() 而且將計算以後的 hash 值高低位作異或運算,以增長 hash 的複雜度。(Java 裏一個 int 類型佔 4 個字節,一個字節是 8 bit,因此下面源碼中的 h 與 h 右移 16 位就至關於高低位異或)
static final int hash(Object key) { int h; return (key == null) ? 0 : (h = key.hashCode()) ^ (h >>> 16); } //key.hashCode() 是Object類的native方法, 實現是將內部地址轉換成一個integer, 可是並非由Java實現的 public native int hashCode();
putVal() 方法
這部分是主要 put 的邏輯
若是以上條件都不成立,代表 tab[i] 上有其它 key 元素存在,而且沒有轉成紅黑樹結構,這時只需調用 tab[i].next 來遍歷此鏈表,找到鏈表的尾而後將元素存到當前鏈表的尾部。
transient Node<K,V>[] table; final V putVal(int hash, K key, V value, boolean onlyIfAbsent, boolean evict) { Node<K,V>[] tab; Node<K,V> p; int n, i; // 根據當前的map size計算容量大小capacity, 主要實現是在resize()中計算capacity,須要擴容的時候, 長度左移一位(二倍) if ((tab = table) == null || (n = tab.length) == 0) n = (tab = resize()).length; // 這裏就是常常說的桶結構了, 看過HashMap介紹的都知道它的內部有不一樣的桶, 這個桶實際上就是一個鏈表結構 // 在這個地方, HashMap先判斷key所屬的桶是否存在。 (n - 1) & hash 至關於計算桶的序號, 根據桶序號來找到對應的桶 // 這裏的table 是HashMap的數組, 數組爲空就新建一個數組 newNode(hash, key, value, null) if ((p = tab[i = (n - 1) & hash]) == null) tab[i] = newNode(hash, key, value, null); else { //數組不爲空, 先判斷key是否存在, 存在 就覆蓋value Node<K,V> e; K k; if (p.hash == hash && ((k = p.key) == key || (key != null && key.equals(k)))) e = p; // 若是此鏈表是紅黑樹結構(TreeNode) else if (p instanceof TreeNode) e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value); else { // 循環當前鏈表, 找出p.next爲空的位置就是鏈表的末端, 添加上 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) // put以後,若是元素個數大於當前的數組容量了,進行數組擴容 resize(); afterNodeInsertion(evict); return null; }
get() 方法會調用 getNode() 方法,這是 get() 的核心,getNode() 方法的兩個參數分別是 hash 值和 key。
public V get(Object key) { Node<K,V> e; return (e = getNode(hash(key), key)) == null ? null : e.value; }
這裏重點來看 getNode() 方法,前面講到過,HashMap 是經過 key 生成的 hash 值來存儲到數組的對應索引上,HashMap 在 get 的時候也是用這種方式來查找元素的。
final Node<K,V> getNode(int hash, Object key) { // tab: HashMap的數組 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); // 不是紅黑樹, 遍歷桶, 直到找到對應的key, 返回 do { // 1. 判斷hash值是否相等; 2. 判斷key相等。 防止hash碰撞發生 if (e.hash == hash && ((k = e.key) == key || (key != null && key.equals(k)))) return e; } while ((e = e.next) != null); } } return null; }
前面提到,當 HashMap 在 put 元素的時候,HashMap 會調用 resize() 方法來從新計算數組容量,數組擴容以後,數組長度發生變化。咱們知道 HashMap 是根據 key 的 hash 和數組長度計算元素位置的,那當數組長度發生變化時,若是不從新計算元素的位置,當咱們 get 元素的時候就找不到正確的元素了,因此 HashMap 在擴容的同時也從新對數組元素進行了計算。
這時還有一個問題,re-hash 的時候同一個桶(bucket)上的鏈表會從新排列仍是鏈表仍然在同一桶上。先考慮一下當它擴容的時候同一個桶上的元素再與新數組長度作與運算 & 時,可能計算出來的數組索引不一樣。假如數組長度是 16,擴容後的數組長度將是 32。
下邊用二進制說明這個問題:
最終的結果是 0000 1111,和用 oldLen 計算的結果同樣,其實看上式能夠發現真正能改變索引值的是 hash 第 5 位(從右向左)上的值,也就是 length 的最高非零位,因此,同一個鏈表上的元素在擴容後,它們的索引只有兩種可能,一種就是保持原位(最高非零位是 0),另外一種就是 length+ 原索引 i (第五位是 1,結果就相等於 25+原索引 i,也就是 length+i)。
下邊所示的 HashMap 源碼中就是用這個思路來 re-hash 一個桶上的鏈表,e.hash & oldCap == 0 判斷 hash 對應 length 的最高非 0 位是不是 1,是 1 則把元素存在原索引,不然將元素存在 length+原索引的位置。HashMap 定義了四個 Node 對象,lo 開頭的是低位的鏈表(原索引),hi 開頭的是高位的鏈表(length+原索引,因此至關因而新 length 的高位)。
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) { // 最高非零位與操做結果是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; } if (hiTail != null) { hiTail.next = null; // 這部分的鏈表索引起生變化,將鏈表放到新索引上 newTab[j + oldCap] = hiHead; }
另外對比一下 HashMap 與 HashTable:
樊騰飛,有開源精神,樂於分享,但願經過寫博客認識更多志同道合的人。我的博客:rollsbean.com
本文系做者投稿文章。歡迎投稿。