Java:从Map到HashMap的一步步实现!

开发 后端
在 Java 中, Map 提供了键——值的映射关系。映射不能包含重复的键,并且每个键只能映射到一个值。

 一、 Map

1.1 Map 接口

在 Java 中, Map 提供了键——值的映射关系。映射不能包含重复的键,并且每个键只能映射到一个值。

以 Map 键——值映射为基础,java.util 提供了 HashMap(最常用)、 TreeMap、Hashtble、LinkedHashMap 等数据结构。

衍生的几种 Map 的主要特点:

  •  HashMap:最常用的数据结构。键和值之间通过 Hash函数 来实现映射关系。当进行遍历的 key 是无序的
  •  TreeMap:使用红黑树构建的数据结构,因为红黑树的原理,可以很自然的对 key 进行排序,所以 TreeMap 的 key 遍历时是默认按照自然顺序(升序)排列的。
  •  LinkedHashMap: 保存了插入的顺序。遍历得到的记录是按照插入顺序的。

1.2 Hash 散列函数

Hash (散列函数)是把任意长度的输入通过散列算法变换成固定长度的输出。Hash 函数的返回值也称为 哈希值 哈希码 摘要或哈希。Hash作用如下图所示:

Hash 函数可以通过选取适当的函数,可以在时间和空间上取得较好平衡。

解决 Hash 的两种方式:拉链法和线性探测法

1.3 键值关系的实现

  1. interface Entry<K,V> 

在 HashMap 中基于链表的实现 

  1. static class Node<K,V> implements Map.Entry<K,V> {  
  2.         final int hash;  
  3.         final K key;  
  4.         V value;  
  5.         Node<K,V> next;  
  6.         Node(int hash, K key, V value, Node<K,V> next) {  
  7.             this.hash = hash;  
  8.             this.key = key;  
  9.             this.value = value;  
  10.             this.next = next;  
  11.         } 

用树的方式实现: 

  1. static final class TreeNode<K,V> extends LinkedHashMap.Entry<K,V> {  
  2.         TreeNode<K,V> parent;  // red-black tree links  
  3.         TreeNode<K,V> left;  
  4.         TreeNode<K,V> right;  
  5.         TreeNode<K,V> prev;    // needed to unlink next upon deletion  
  6.         boolean red;  
  7.         TreeNode(int hash, K key, V val, Node<K,V> next) {  
  8.             super(hash, key, val, next);  
  9.         } 

1.4 Map 约定的 API

1.4.1 Map 中约定的基础 API

基础的增删改查: 

  1. int size();  // 返回大小  
  2. boolean isEmpty(); // 是否为空  
  3. boolean containsKey(Object key); // 是否包含某个键  
  4. boolean containsValue(Object value); // 是否包含某个值  
  5. V get(Object key); // 获取某个键对应的值   
  6. V put(K key, V value); // 存入的数据   
  7. V remove(Object key); // 移除某个键  
  8. void putAll(Map<? extends K, ? extends V> m); //将将另一个集插入该集合中  
  9. void clear();  // 清除  
  10. Set<K> keySet(); //获取 Map的所有的键返回为 Set集合  
  11. Collection<V> values(); //将所有的值返回为 Collection 集合  
  12. Set<Map.Entry<K, V>> entrySet(); // 将键值对映射为 Map.Entry,内部类 Entry 实现了映射关系的实现。并且返回所有键值映射为 Set 集合。   
  13. boolean equals(Object o);   
  14. int hashCode(); // 返回 Hash 值  
  15. default boolean replace(K key, V oldValue, V newValue); // 替代操作  
  16. default V replace(K key, V value); 

1.4.2 Map 约定的较为高级的 API 

  1. default V getOrDefault(Object key, V defaultValue); //当获取失败时,用 defaultValue 替代。 
  2. default void forEach(BiConsumer<? super K, ? super V> action)  // 可用 lambda 表达式进行更快捷的遍历  
  3. default void replaceAll(BiFunction<? super K, ? super V, ? extends V> function);   
  4. default V putIfAbsent(K key, V value);  
  5. default V computeIfAbsent(K key,  
  6.             Function<? super K, ? extends V> mappingFunction); 
  7. default V computeIfPresent(K key,  
  8.             BiFunction<? super K, ? super V, ? extends V> remappingFunction);  
  9. default V compute(K key,  
  10.             BiFunction<? super K, ? super V, ? extends V> remappingFunction)  
  11. default V merge(K key, V value,  
  12.             BiFunction<? super V, ? super V, ? extends V> remappingFunction)    

1.4.3 Map 高级 API 的使用

  •  getOrDefault() 当这个通过 key获取值,对应的 key 或者值不存在时返回默认值,避免在使用过程中 null 出现,避免程序异常。
  •  ForEach() 传入 BiConsumer 函数式接口,表达的含义其实和 Consumer 一样,都 accept 拥有方法,只是 BiConsumer 多了一个 andThen() 方法,接收一个BiConsumer接口,先执行本接口的,再执行传入的参数的 accept 方法。   
  1. Map<String, String> map = new HashMap<>();  
  2.      map.put("a", "1");  
  3.      map.put("b", "2");  
  4.      map.put("c", "3");  
  5.      map.put("d", "4");  
  6.      map.forEach((k, v) -> {  
  7.          System.out.println(k+"-"+v);  
  8.      });  
  9.  } 

更多的函数用法:

https://www.cnblogs.com/king0/p/runoob.com/java/java-hashmap.html

1.5 从 Map 走向 HashMap

HashMap 是 Map的一个实现类,也是 Map 最常用的实现类。

1.5.1 HashMap 的继承关系 

  1. public class HashMap<K,V> extends AbstractMap<K,V>  
  2.     implements Map<K,V>, Cloneable, Serializable  

在 HashMap 的实现过程中,解决 Hash冲突的方法是拉链法。因此从原理来说 HashMap 的实现就是 数组 + 链表(数组保存链表的入口)。当链表过长,为了优化查询速率,HashMap 将链表转化为红黑树(数组保存树的根节点),使得查询速率为 log(n),而不是链表的 O(n)。

二、HashMap 

  1. /*  
  2.  * @author  Doug Lea  
  3.  * @author  Josh Bloch  
  4.  * @author  Arthur van Hoff  
  5.  * @author  Neal Gafter  
  6.  * @see     Object#hashCode()  
  7.  * @see     Collection  
  8.  * @see     Map  
  9.  * @see     TreeMap  
  10.  * @see     Hashtable  
  11.  * @since   1.2  
  12.  */ 

首先 HashMap 由 Doug Lea 和 Josh Bloch 两位大师的参与。同时 Java 的 Collections 集合体系,并发框架 Doug Lea 也做出了不少贡献。

2.1 基本原理

对于一个插入操作,首先将键通过 Hash 函数转化为数组的下标。若该数组为空,直接创建节点放入数组中。若该数组下标存在节点,即 Hash 冲突,使用拉链法,生成一个链表插入。

引用图片来自 https://blog.csdn.net/woshimaxiao1/article/details/83661464

如果存在 Hash 冲突,使用拉链法插入,我们可以在这个链表的头部插入,也可以在链表的尾部插入,所以在 JDK 1.7 中使用了头部插入的方法,JDK 1.8 后续的版本中使用尾插法。

JDK 1.7 使用头部插入的可能依据是最近插入的数据是最常用的,但是头插法带来的问题之一,在多线程会链表的复制会出现死循环。所以 JDK 1.8 之后采用的尾部插入的方法。关于这点,可以看:Java8之后,HashMap链表插入方式->为何要从头插入改为尾插入

在 HashMap 中,前面说到的 数组+链表 的数组的定义 

  1. transient Node<K,V>[] table; 

链表的定义: 

  1. static class Node<K,V> implements Map.Entry<K,V>  

2.1.2 提供的构造函数 

  1. public HashMap() { // 空参  
  2.       this.loadFactor = DEFAULT_LOAD_FACTOR; // all other fields defaulted  
  3.   }  
  4.   public HashMap(int initialCapacity) { //带有初始大小的,一般情况下,我们需要规划好 HashMap 使用的大小,因为对于一次扩容操作,代价是非常的大的  
  5.       this(initialCapacity, DEFAULT_LOAD_FACTOR);  
  6.   }  
  7.   public HashMap(int initialCapacity, float loadFactor); // 可以自定义负载因子  public HashMap(int initialCapacity, float loadFactor); // 可以自定义负载因子 

三个构造函数,都没有完全的初始化 HashMap,当我们第一次插入数据时,才进行堆内存的分配,这样提高了代码的响应速度。

2.2 HashMap 中的 Hash函数定义 

  1. static final int hash(Object key) {  
  2.         int h;  
  3.         return (key == null) ? 0 : (h = key.hashCode()) ^ (h >>> 16); // 将 h 高 16 位和低 16 位 进行异或操作。  
  4.     }  
  5. // 采用 异或的原因:两个进行位运算,在与或异或中只有异或到的 0 和 1 的概率是相同的,而&和|都会使得结果偏向0或者1。 

这里可以看到,Map 的键可以为 null,且 hash 是一个特定的值 0。

Hash 的目的是获取数组 table 的下标。Hash 函数的目标就是将数据均匀的分布在 table 中。

让我们先看看如何通过 hash 值得到对应的数组下标。第一种方法:hash%table.length()。但是除法操作在 CPU 中执行比加法、减法、乘法慢的多,效率低下。第二种方法 table[(table.length - 1) & hash] 一个与操作一个减法,仍然比除法快。这里的约束条件为  table.length = 2^N。 

  1. table.length =16  
  2. table.length -1 = 15 1111 1111  
  3. //任何一个数与之与操作,获取到这个数的低 8 位,其他位为 0 

上面的例子可以让我们获取到对应的下标,而 (h = key.hashCode()) ^ (h >>> 16) 让高 16 也参与运算,让数据充分利用,一般情况下 table 的索引不会超过 216,所以高位的信息我们就直接抛弃了,^ (h >>> 16) 让我们在数据量较少的情况下,也可以使用高位的信息。如果 table 的索引超过 216, hashCode() 的高 16 为 和 16 个 0 做异或得到的 Hash 也是公平的。

2.3 HashMap 的插入操作

上面我们已经知道如果通过 Hash 获取到 对应的 table 下标,因此我们将对应的节点加入到链表就完成了一个 Map 的映射,的确 JDK1.7 中的 HashMap 实现就是这样。让我们看一看 JDK 为实现现实的 put 操作。

定位到 put() 操作。 

  1. public V put(K key, V value) {  
  2.        return putVal(hash(key), key, value, false, true);  
  3.    } 

可以看到 put 操作交给了 putVal 来进行通用的实现。 

  1. final V putVal(int hash, K key, V value, boolean onlyIfAbsent, boolean evict);  
  2. //onlyIfAbsent  如果当前位置已存在一个值,是否替换,false是替换,true是不替换  
  3. evict // 钩子函数的参数,LinkedHashMap 中使用到,HashMap 中无意义。 

2.3.1 putVal 的流程分析

其实 putVal() 流程的函数非常的明了。这里挑了几个关键步骤来引导。

是否第一次插入,true 调用 resizer() 进行调整,其实此时 resizer() 是进行完整的初始化,之后直接赋值给对应索引的位置。 

  1. if ((tab = table) == null || (n = tab.length) == 0) // 第一次 put 操作, tab 没有分配内存,通过 redize() 方法分配内存,开始工作。  
  2.            n = (tab = resize()).length;  
  3.        if ((p = tab[i = (n - 1) & hash]) == null)  
  4.            tab[i] = newNode(hash, key, value, null); 

如果链表已经转化为树,则使用树的插入。 

  1. else if (p instanceof TreeNode)  
  2.                 e = ((TreeNode<K,V>)p).putTreeVal(this, tab, hash, key, value); 

用遍历的方式遍历每个 Node,如果遇到键相同,或者到达尾节点的next 指针将数据插入,记录节点位置退出循环。若插入后链表长度为 8 则调用 treeifyBin() 是否进行树的转化 。 

  1. for (int binCount = 0; ; ++binCount) {  
  2.                    if ((e = p.next) == null) {  
  3.                        p.next = newNode(hash, key, value, null);  
  4.                        if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st  
  5.                            treeifyBin(tab, hash);  
  6.                        break;  
  7.                    }  
  8.                    if (e.hash == hash &&  
  9.                        ((k = e.key) == key || (key != null && key.equals(k))))  
  10.                        break;  
  11.                    p = e
  12.                 } 

对键重复的操作:更新后返回旧值,同时还取决于onlyIfAbsent,普通操作中一般为 true,可以忽略。   

  1. if (e != null) { // existing mapping for key  
  2.               V oldValue = e.value; 
  3.                if (!onlyIfAbsent || oldValue == null)  
  4.                   e.value = value;  
  5.               afterNodeAccess(e); //钩子函数,进行后续其他操作,HashMap中为空,无任何操作。  
  6.               return oldValue;  
  7.           } 

  1. ++modCount;  
  2.    if (++size > threshold)  
  3.        resize();  
  4.    afterNodeInsertion(evict);  
  5.    return null; 

后续的数据维护。

2.3.2 modCount 的含义

fail-fast 机制是java集合(Collection)中的一种错误机制。当多个线程对同一个集合的内容进行操作时,就可能会产生fail-fast事件。例如:当某一个线程A通过iterator去遍历某集合的过程中,若该集合的内容被其他线程所改变了;那么线程A访问集合时,就会抛出ConcurrentModificationException异常,产生fail-fast事件。一种多线程错误检查的方式,减少异常的发生。

一般情况下,多线程环境 我们使用 ConcurrentHashMap 来代替 HashMap。

2.4 resize() 函数

HashMap 扩容的特点:默认的table 表的大小事 16,threshold 为 12。负载因子 loadFactor .75,这些都是可以构造是更改。以后扩容都是 2 倍的方式增加。

至于为何是0.75 代码的注释中也写了原因,对 Hash函数构建了泊松分布模型,进行了分析。

2.4.1 HashMap 预定义的一些参数 

  1. static final int DEFAULT_INITIAL_CAPACITY = 1 << 4; // aka 16  HashMap 的默认大小。 为什么使用 1 <<4  
  2. static final int MAXIMUM_CAPACITY = 1 << 30; // 最大容量  
  3. static final float DEFAULT_LOAD_FACTOR = 0.75f; // 加载因子,扩容使用  
  4. static final int UNTREEIFY_THRESHOLD = 6;//  树结构转化为链表的阈值  
  5. static final int TREEIFY_THRESHOLD = 8;  //  链表转化为树结构的阈值  
  6. static final int MIN_TREEIFY_CAPACITY = 64; // 链表转变成树之前,还会有一次判断,只有数组长度大于 64 才会发生转换。这是为了避免在哈希表建立初期,多个键值对恰好被放入了同一个链表中而导致不必要的转化。 
  7. // 定义的有关变量  
  8. int threshold;   // threshold表示当HashMap的size大于threshold时会执行resize操作 

这些变量都是和 HashMap 的扩容机制有关,将会在下文中用到。

2.4.2 resize() 方法解析   

  1. Node<K,V>[] oldTab = table 
  2.      int oldCap = (oldTab == null) ? 0 : oldTab.length;   
  3.      int oldThr = threshold 
  4.      int newCap, newThr = 0; // 定义了 旧表长度、旧表阈值、新表长度、新表阈值  
  1. if (oldCap > 0) {  // 插入过数据,参数不是初始化的  
  2.           if (oldCap >= MAXIMUM_CAPACITY) {  // 如果旧的表长度大于 1 << 30 
  3.               threshold = Integer.MAX_VALUE; // threshold 设置 Integer 的最大值。也就是说我们可以插入 Integer.MAX_VALUE 个数据  
  4.               return oldTab; // 直接返回旧表的长度,因为表的下标索引无法扩大了。   
  5.           }  
  6.           else if ((newCap = oldCap << 1< MAXIMUM_CAPACITY && //   
  7.                    oldCap >= DEFAULT_INITIAL_CAPACITY)  //新表的长度为旧表的长度的 2 倍。 
  8.               newThr = oldThr << 1; // double threshold 新表的阈值为同时为旧表的两倍  
  9.       }  
  10.       else if (oldThr > 0) //   public HashMap(int initialCapacity, float loadFactor)   中的  this.threshold = tableSizeFor(initialCapacity);  给正确的位置    
  11.           newCap = oldThr 
  12.       else {               // zero initial threshold signifies using defaults ,如果调用了其他两个构造函数,则下面代码初始化。因为他们都没有对其 threshold 设置,默认为 0, 
  13.           newCap = DEFAULT_INITIAL_CAPACITY 
  14.           newThr = (int)(DEFAULT_LOAD_FACTOR * DEFAULT_INITIAL_CAPACITY);  
  15.       }  
  16.       if (newThr == 0) { // 修正 threshold,例如上面的   else if (oldThr > 0)  部分就没有设置。  
  17.           float ft = (float)newCap * loadFactor;  
  18.           newThr = (newCap < MAXIMUM_CAPACITY && ft < (float)MAXIMUM_CAPACITY ?  
  19.                     (int)ft : Integer.MAX_VALUE);  
  20.       }  
  21.       threshold = newThr 
  22.       @SuppressWarnings({"rawtypes","unchecked"}) 

当一些参数设置正确后便开始扩容。   

  1. Node<K,V>[] newTab = (Node<K,V>[])new Node[newCap];  

当扩容完毕之后,自然就是将原表中的数据搬到新的表中。下面代码完成了该任务。 

  1. if (oldTab != null) {  
  2.    for (int j = 0; j < oldCap; ++j) {  
  3.       ....   
  4.    }  

如何正确的,快速的扩容调整每个键值节点对应的下标?第一种方法:遍历节点再使用 put() 加入一遍,这种方法实现,但是效率低下。第二种,我们手动组装好链表,加入到相应的位置。显然第二种比第一种高效,因为第一种 put() 还存在其他不属于这种情况的判断,例如重复键的判断等。

练手项目,学习强化,点击这里

所以 JDK 1.8 也使用了第二种方法。我们可以继续使用e.hash & (newCap - 1)找到对应的下标位置,对于旧的链表,执行e.hash & (newCap - 1) 操作,只能产生两个不同的索引。一个保持原来的索引不变,另一个变为 原来索引 + oldCap(因为 newCap 的加入产生导致索引的位数多了 1 位,即就是最左边的一个,且该位此时结果为 1,所以相当于 原来索引 + oldCap)。所以可以使用 if ((e.hash & oldCap) == 0) 来确定出索引是否来变化。

因此这样我们就可以将原来的链表拆分为两个新的链表,然后加入到对应的位置。为了高效,我们手动的组装好链表再存储到相应的下标位置上。 

  1. oldCap  = 16  
  2. newCap  = 32  
  3. hash       : 0001 1011  
  4. oldCap-1   : 0000 1111  
  5. 结果为     :  0000 1011  对应的索引的 11  
  6. -------------------------  
  7. e.hash & oldCap 则定于 1,则需要进行调整索引  
  8. oldCap  = 16  
  9. hash       : 0001 1011   
  10. newCap-1   : 0001 1111  
  11. 结果为     :  0001 1011  
  12. 相当于 1011 + 1 0000 原来索引 + newCap 
  1. for (int j = 0; j < oldCap; ++j)  // 处理每个链表  

特殊条件处理 

  1. Node<K,V> e;  
  2.                 if ((e = oldTab[j]) != null) {  
  3.                     oldTab[j] = null;  
  4.                     if (e.next == null)  // 该 链表只有一个节点,那么直接复制到对应的位置,下标由 e.hash & (newCap - 1) 确定  
  5.                         newTab[e.hash & (newCap - 1)] = e;  
  6.                     else if (e instanceof TreeNode) // 若是 树,该给树的处理程序  
  7.                         ((TreeNode<K,V>)e).split(this, newTab, j, oldCap); 

普通情况处理:       

  1. else { // preserve order  
  2.                         Node<K,V> loHead = nullloTail = null;  // 构建原来索引位置 的链表,需要的指针  
  3.                         Node<K,V> hiHead = nullhiTail = null; // 构建 原来索引 + oldCap 位置 的链表需要的指针  
  4.                         Node<K,V> next;  
  5.                         do {  
  6.                             next = e.next;  
  7.                             if ((e.hash & oldCap) == 0) {  
  8.                                 if (loTail == null)  
  9.                                     loHead = e;  
  10.                                 else  
  11.                                     loTail.next = e;  
  12.                                 loTail = e 
  13.                             }  
  14.                             else {  
  15.                                 if (hiTail == null) 
  16.                                      hiHead = e;  
  17.                                 else  
  18.                                     hiTail.next = e;  
  19.                                 hiTail = e 
  20.                             }  
  21.                         } while ((e = next) != null); // 将原来的链表划分两个链表  
  22.                         if (loTail != null) { // 将链表写入到相应的位置 
  23.                             loTail.next = null 
  24.                             newTab[j] = loHead; 
  25.                          }  
  26.                         if (hiTail != null) {  
  27.                             hiTail.next = null 
  28.                             newTab[j + oldCap] = hiHead;  
  29.                         }  
  30.                     } 

到此 resize() 方法的逻辑完成了。总的来说 resizer() 完成了 HashMap 完整的初始化,分配内存和后续的扩容维护工作。

2.5 remove 解析   

  1. public V remove(Object key) {  
  2.        Node<K,V> e;  
  3.        return (e = removeNode(hash(key), key, null, false, true)) == null ?  
  4.            null : e.value;  
  5.    } 

将 remove 删除工作交给内部函数 removeNode() 来实现。 

  1. final Node<K,V> removeNode(int hash, Object key, Object value,  
  2.                                boolean matchValue, boolean movable) {  
  3.         Node<K,V>[] tab; Node<K,V> p; int n, index;  
  4.         if ((tab = table) != null && (n = tab.length) > 0 &&  
  5.             (p = tab[index = (n - 1) & hash]) != null) {  // 获取索引, 
  6.             Node<K,V> node = null, e; K k; V v;  
  7.             if (p.hash == hash &&    
  8.                 ((k = p.key) == key || (key != null && key.equals(k)))) // 判断索引处的值是不是想要的结果  
  9.                 node = p;    
  10.             else if ((e = p.next) != null) { // 交给树的查找算法  
  11.                 if (p instanceof TreeNode)  
  12.                     node = ((TreeNode<K,V>)p).getTreeNode(hash, key);  
  13.                 else {  
  14.                     do { // 遍历查找  
  15.                         if (e.hash == hash &&  
  16.                             ((k = e.key) == key ||  
  17.                              (key != null && key.equals(k)))) {  
  18.                             node = e;  
  19.                             break;  
  20.                         } 
  21.                          p = e 
  22.                     } while ((ee = e.next) != null);  
  23.                 }  
  24.             }  
  25.             if (node != null && (!matchValue || (v = node.value) == value ||  
  26.                                  (value != null && value.equals(v)))) {  
  27.                 if (node instanceof TreeNode)  //树的删除  
  28.                     ((TreeNode<K,V>)node).removeTreeNode(this, tab, movable);  
  29.                 else if (node == p) // 修复链表,链表的删除操作  
  30.                     tab[index] = node.next;   
  31.                 else  
  32.                     p.next = node.next;    
  33.                 ++modCount;  
  34.                 --size;  
  35.                 afterNodeRemoval(node); 
  36.                  return node;  
  37.             }  
  38.         }  
  39.         return null;  
  40.     } 

三、 HashMap 从链表到红黑树的转变

如果链表的长度(冲突的节点数)已经达到8个,此时会调用 treeifyBin() ,treeifyBin() 首先判断当前hashMap 的 table的长度,如果不足64,只进行resize,扩容table,如果达到64,那么将冲突的存储结构为红黑树。在源码还有这样的一个字段。 

  1. static final int UNTREEIFY_THRESHOLD = 6;   
  2. // 这样表明了从红黑树转化为链表的阈值为 6,为何同样不是 8 那?   
  3. //如果插入和删除都在 8 附近,将多二者相互转化将浪费大量的时间,对其性能影响。   
  4. //如果是的二者转化的操作不平衡,偏向一方,则可以避免此类影响。 

3.1 红黑树的数据结构 

  1. static final class TreeNode<K,V> extends LinkedHashMap.Entry<K,V> {  
  2.       TreeNode<K,V> parent;  // red-black tree links  
  3.       TreeNode<K,V> left;  
  4.       TreeNode<K,V> right;  
  5.       TreeNode<K,V> prev;    // 删除后需要取消链接,指向前一个节点(原链表中的前一个节点)  
  6.       boolean red;  
  7.   } 

因为 继承了 LinkedHashMap.Entry<K,V> ,所以存储的数据还是在Entry 中: 

  1. static class Node<K,V> implements Map.Entry<K,V> {  
  2.      final int hash;  
  3.      final K key;  
  4.      V value;  
  5.      Node<K,V> next;  
  6.      Node(int hash, K key, V value, Node<K,V> next) {  
  7.          this.hash = hash;  
  8.          this.key = key;  
  9.          this.value = value;  
  10.          this.next = next;  
  11.      }  

3.2 承上启下的 treeifyBin()

treeifyBin() 决定了一个链表何时转化为一个红黑树。treeifyBin() 有两种格式: 

  1. final void treeifyBin(Node<K,V>[] tab, int hash);  
  2.  final void treeify(Node<K,V>[] tab);   
  1. final void treeifyBin(Node<K,V>[] tab, int hash) { // 简单的 Node 修改为 TreeNode,同时维护了 prev 属性。 
  2.        int n, index; Node<K,V> e;  
  3.        if (tab == null || (n = tab.length) < MIN_TREEIFY_CAPACITY 
  4.            resize();  
  5.        else if ((e = tab[index = (n - 1) & hash]) != null) {  
  6.            TreeNode<K,V> hd = nulltl = null 
  7.            do {  
  8.                TreeNode<K,V> p = replacementTreeNode(e, null); 
  9.                if (tl == null)  
  10.                    hd = p 
  11.                else {  
  12.                    p.prev = tl 
  13.                    tl.next = p 
  14.                }  
  15.                tl = p
  16.             } while ((ee = e.next) != null);  
  17.            if ((tab[index] = hd) != null)  
  18.                hd.treeify(tab);  // 真正生成红黑树的  
  19.        }  
  20.    }  
  1. TreeNode<K,V> replacementTreeNode(Node<K,V> p, Node<K,V> next) {  
  2.     return new TreeNode<>(p.hash, p.key, p.value, next);  
  3. } // 实现 Node 链表节点到 TreeNode 节点的转化。 

下面函数真正实现了链表的红黑树的转变。首先构建一个标准查询二叉树,然后在标准查询二叉树然后调整为一个红黑树。而 balanceInsertion() 实现了调整。       

  1. /**  
  2.         * Forms tree of the nodes linked from this node.  
  3.         */  
  4.        final void treeify(Node<K,V>[] tab) {  
  5.            TreeNode<K,V> root = null 
  6.            for (TreeNode<K,V> x = this, next; x != null; x = next) {  
  7.                next = (TreeNode<K,V>)x.next;  
  8.                xx.left = x.right = null 
  9.                if (root == null) { // 第一次转化过程,将链表的头节点作为根节点。  
  10.                    x.parent = null 
  11.                    x.red = false;  // 红黑树的定义 根节点必须为黑色  
  12.                    root = x 
  13.                }  
  14.                else { 
  15.                     K k = x.key;  
  16.                    int h = x.hash;  
  17.                    Class<?> kc = null 
  18.                    for (TreeNode<K,V> p = root;;) {  
  19.                        int dir, ph;  
  20.                        K ppk = p.key;  
  21.                        if ((pph = p.hash) > h)  //// 通过 Hash 的大小来确定插入顺序  
  22.                            dir = -1; // dir 大小顺序的标识 
  23.                         else if (ph < h 
  24.                            dir = 1;   
  25.                        else if ((kc == null && //当 两个 Hash 的值相同,进行特殊的方法,确定大小。  
  26.                                  (kc = comparableClassFor(k)) == null) || // Returns x's Class if it is of the form "class C implements Comparable ", else null. 如果 key类的 源码书写格式为 C implement Comparable<C> 那么返回该类类型 C, 如果间接实现也不行。 如果是 String 类型,直接返回 String.class 
  27.                                  (dir = compareComparables(kc, k, pk)) == 0)   //    ((Comparable)k).compareTo(pk)); 强制转换后进行对比,若 dir == 0,则 tieBreakOrder(),继续仲裁  
  28.                             dir = tieBreakOrder(k, pk);  // 首先通过二者的类类型进行比较,如果相等的话,使用 (System.identityHashCode(a) <= System.identityHashCode(b) 使用原始的 hashcode,不是重写的在对比。 
  29.                         TreeNode<K,V> xp = p; // 遍历的,上一个节点  
  30.                        if ((p = (dir <= 0) ? p.left : p.right) == null) { //通过 dir,将 p 向下查找,直到 p 为 null,找到一个插入时机  
  31.                            x.parent = xp 
  32.                            if (dir <= 0)  
  33.                                xxp.left = x;  
  34.                            else  
  35.                                xxp.right = x;  
  36.                            root = balanceInsertion(root, x); //进行二叉树的调整  
  37.                            break;  
  38.                        }  
  39.                    }  
  40.                }  
  41.            }  
  42.            moveRootToFront(tab, root);  
  43.        } 

 3.3 将一个二叉树转化为红黑树的操作-balanceInsertion()

当红黑树中新增节点的时候需要调用balanceInsertion方法来保证红黑树的特性。

如果想要了解红黑树的插入过程那么必须对红黑树的性质有一个较为清晰的了解。

红黑树的性质:

  1.  每个结点或是红色的,或是黑色的
  2.  根节点是黑色的
  3.  每个叶结点(NIL)是黑色的
  4.  如果一个节点是红色的,则它的两个儿子都是黑色的。
  5.  对于每个结点,从该结点到其叶子结点构成的所有路径上的黑结点个数相同。 
  1. static <K,V> TreeNode<K,V> balanceInsertion(TreeNode<K,V> root,  
  2.                                                     TreeNode<K,V> x) {  
  3.             x.red = true;  // 插入的子节点必须为 red  
  4.             for (TreeNode<K,V> xp, xpp, xppl, xppr;;) { //// x 当前处理节点 xp父节点 xpp祖父节点 xppl祖父左节点 xppr 祖父右节点  
  5.                 if ((xxp = x.parent) == null) { // 如果 当前处理节点为根节点,满足红黑树的性质,结束循环  
  6.                     x.red = false 
  7.                     return x;  
  8.                 }  
  9.                 else if (!xp.red || (xpxpp = xp.parent) == null)   
  10.                     return root;  
  11.                 if (xp == (xppxppl = xpp.left)) {  
  12.                     if ((xppxppr = xpp.right) != null && xppr.red) {  
  13.                         xppr.red = false
  14.                          xp.red = false 
  15.                         xpp.red = true 
  16.                         x = xpp 
  17.                     }  
  18.                     else {  
  19.                         if (x == xp.right) {  
  20.                             root = rotateLeft(root, x = xp);  
  21.                             xpp = (xxp = x.parent) == null ? null : xp.parent;  
  22.                         }  
  23.                         if (xp != null) { 
  24.                              xp.red = false 
  25.                             if (xpp != null) {  
  26.                                 xpp.red = true 
  27.                                 root = rotateRight(root, xpp);  
  28.                             }  
  29.                         }  
  30.                     }  
  31.                 }  
  32.                 else {  
  33.                     if (xppl != null && xppl.red) {  
  34.                         xppl.red = false 
  35.                         xp.red = false 
  36.                         xpp.red = true 
  37.                         x = xpp 
  38.                     }  
  39.                     else {  
  40.                         if (x == xp.left) {  
  41.                             root = rotateRight(root, x = xp);  
  42.                             xpp = (xxp = x.parent) == null ? null : xp.parent;  
  43.                         }  
  44.                         if (xp != null) {  
  45.                             xp.red = false 
  46.                             if (xpp != null) {  
  47.                                 xpp.red = true 
  48.                                 root = rotateLeft(root, xpp);  
  49.                             }  
  50.                         }  
  51.                     }  
  52.                 }  
  53.             }  
  54.         } 

TreeNode 红黑树总结

TreeNode 完整的实现了一套红黑树的增删改查的规则。实现参考了《算法导论》 

  1. /* ------------------------------------------------------------ */  
  2. // Red-black tree methods, all adapted from CLR 

这里推荐一个红黑树动画演示网站 https://rbtree.phpisfuture.com/

红黑树是一个不严格的平衡二叉查找树,高度近似 log(N)。

四、HashMap 的扩展

Map中 key 有一个性质,就是 key 不能重复,而 Java Set 的含义:集合中不能有重复的元素。HashMap 的实现已经足够的优秀。那么我们是否可以用 key 的性质来实现 Set ?的确 JDK 中的 HashSet 就是这样做的。 

  1. public class HashSet<E>  
  2.     extends AbstractSet<E>  
  3.     implements Set<E>, Cloneable, java.io.Serializable  
  4.  
  5.     private transient HashMap<E,Object> map;  
  6.       // Dummy value to associate with an Object in the backing Map  
  7.     private static final Object PRESENT = new Object();  

PRESENT 就是存进 Map 中的 value,而 key 正是 Set 语义的实现。而且可以判断出 HashSet 中是允许存入 Null 值的。 

 

责任编辑:庞桂玉 来源: Java知音
相关推荐

2016-11-02 18:54:01

javascript

2017-01-19 21:08:33

iOS路由构建

2009-08-14 11:35:01

Scala Actor

2018-12-24 10:04:06

Docker存储驱动

2019-03-05 14:09:27

Docker存储容器

2019-07-09 15:23:22

Docker存储驱动

2017-01-06 15:13:25

LinuxVim源代码

2017-12-25 11:50:57

LinuxArch Linux

2010-03-04 16:28:17

Android核心代码

2011-05-10 10:28:55

2018-06-11 15:30:12

2018-04-23 14:23:12

2018-07-13 15:36:52

2019-04-01 10:15:02

2021-01-03 15:07:16

开发编程语言后端.

2009-12-17 16:36:23

无线路由设置密码

2015-07-27 16:06:16

VMware Thin虚拟化

2023-02-17 14:52:59

谷歌AI

2020-11-09 07:35:59

MySQL数据库业务

2013-01-08 10:01:44

计算模式企业计算HPC
点赞
收藏

51CTO技术栈公众号