Java集合源码剖析(三)【TreeMap、LinkedHashmap】

时间:2023-02-06 17:18:37


TreeMap源码剖析


前言

    本文不打算延续前几篇的风格(对所有的源码加入注释),因为要理解透TreeMap的所有源码,对博主来说,确实需要耗费大量的时间和经历,目前看来不大可能有这么多时间的投入,故这里意在通过于阅读源码对TreeMap有个宏观上的把握,并就其中一些方法的实现做比较深入的分析。

红黑树简介

    TreeMap是基于红黑树实现的,这里只对红黑树做个简单的介绍,红黑树是一种特殊的二叉排序树,关于二叉排序树,参见:http://blog.csdn.net/ns_code/article/details/19823463,红黑树通过一些限制,使其不会出现二叉树排序树中极端的一边倒的情况,相对二叉排序树而言,这自然提高了查询的效率。

    二叉排序树的基本性质如下:

    1、每个节点都只能是红色或者黑色

    2、根节点是黑色

    3、每个叶节点(NIL节点,空节点)是黑色的。

    4、如果一个结点是红的,则它两个子节点都是黑的。也就是说在一条路径上不能出现相邻的两个红色结点。

    5、从任一节点到其每个叶子的所有路径都包含相同数目的黑色节点。

    正是这些性质的限制,使得红黑树中任一节点到其子孙叶子节点的最长路径不会长于最短路径的2倍,因此它是一种接*衡的二叉树。

    说到红黑树,自然不免要和AVL树对比一番。相比较而言,AVL树是严格的平衡二叉树,而红黑树不算严格意义上的平衡二叉树,只是接*衡,不会让树的高度如BST极端情况那样等于节点的个数。其实能用到红黑树的地方,也都可以用AVL树来实现,但红黑树的应用却非常广泛,而AVL树则很少被使用。在执行插入、删除操作时,AVL树需要调整的次数一般要比红黑树多(红黑树的旋转调整最多只需三次),效率相对较低,且红黑树的统计性能较AVL树要好,当然AVL树在查询效率上可能更胜一筹,但实际上也高不了多少。

    红黑树的插入删除操作很简单,就是单纯的二叉排序树的插入删除操作。红黑树被认为比较变态的地方自然在于插入删除后对红黑树的调整操作(旋转和着色),主要是情况分的很多,限于篇幅及博主的熟悉程度优先,这里不打算详细介绍插入删除后调整红黑树的各种情况及其实现,我们有个宏观上的了解即可,如须详细了解,参见算法导论或一些相关的资料。

TreeMap源码剖析

    存储结构

    TreeMap的排序是基于对key的排序实现的,它的每一个Entry代表红黑树的一个节点,Entry的数据结构如下:

[java]  view plain copy
  1. static final class Entry<K,V> implements Map.Entry<K,V> {    
  2.      // 键    
  3.      K key;    
  4.      // 值    
  5.      V value;    
  6.      // 左孩子    
  7.      Entry<K,V> left = null;    
  8.      // 右孩子    
  9.      Entry<K,V> right = null;    
  10.      // 父节点    
  11.      Entry<K,V> parent;    
  12.      // 当前节点颜色    
  13.      boolean color = BLACK;    
  14.   
  15.      // 构造函数    
  16.      Entry(K key, V value, Entry<K,V> parent) {    
  17.          this.key = key;    
  18.          this.value = value;    
  19.          this.parent = parent;    
  20.      }    
  21.   
  22. 。。。。。  
  23. }  

    构造方法

    先来看下TreeMap的构造方法。TreeMap一共有4个构造方法。

    1、无参构造方法

[java]  view plain copy
  1. public TreeMap() {    
  2.     comparator = null;    
  3. }    
    采用无参构造方法,不指定比较器,这时候,排序的实现要依赖key.compareTo()方法,因此key必须实现Comparable接口,并覆写其中的compareTo方法。

    2、带有比较器的构造方法

[java]  view plain copy
  1. public TreeMap(Comparator<? super K> comparator) {    
  2.     this.comparator = comparator;    
  3. }    
    采用带比较器的构造方法,这时候,排序依赖该比较器,key可以不用实现Comparable接口。

    3、带Map的构造方法

[java]  view plain copy
  1. public TreeMap(Map<? extends K, ? extends V> m) {    
  2.     comparator = null;    
  3.     putAll(m);    
  4. }    
    该构造方法同样不指定比较器,调用putAll方法将Map中的所有元素加入到TreeMap中。putAll的源码如下:

[java]  view plain copy
  1. // 将map中的全部节点添加到TreeMap中    
  2. public void putAll(Map<? extends K, ? extends V> map) {    
  3.     // 获取map的大小    
  4.     int mapSize = map.size();    
  5.     // 如果TreeMap的大小是0,且map的大小不是0,且map是已排序的“key-value对”    
  6.     if (size==0 && mapSize!=0 && map instanceof SortedMap) {    
  7.         Comparator c = ((SortedMap)map).comparator();    
  8.         // 如果TreeMap和map的比较器相等;    
  9.         // 则将map的元素全部拷贝到TreeMap中,然后返回!    
  10.         if (c == comparator || (c != null && c.equals(comparator))) {    
  11.             ++modCount;    
  12.             try {    
  13.                 buildFromSorted(mapSize, map.entrySet().iterator(),    
  14.                             nullnull);    
  15.             } catch (java.io.IOException cannotHappen) {    
  16.             } catch (ClassNotFoundException cannotHappen) {    
  17.             }    
  18.             return;    
  19.         }    
  20.     }    
  21.     // 调用AbstractMap中的putAll();    
  22.     // AbstractMap中的putAll()又会调用到TreeMap的put()    
  23.     super.putAll(map);    
  24. }   
    显然,如果Map里的元素是排好序的,就调用buildFromSorted方法来拷贝Map中的元素,这在下一个构造方法中会重点提及,而如果Map中的元素不是排好序的,就调用AbstractMap的putAll(map)方法,该方法源码如下:

[java]  view plain copy
  1. public void putAll(Map<? extends K, ? extends V> m) {    
  2.     for (Map.Entry<? extends K, ? extends V> e : m.entrySet())    
  3.         put(e.getKey(), e.getValue());    
  4. }   
    很明显它是将Map中的元素一个个put(插入)到TreeMap中的,主要因为Map中的元素是无序存放的,因此要一个个插入到红黑树中,使其有序存放,并满足红黑树的性质。

    4、带有SortedMap的构造方法

[java]  view plain copy
  1. public TreeMap(SortedMap<K, ? extends V> m) {    
  2.     comparator = m.comparator();    
  3.     try {    
  4.         buildFromSorted(m.size(), m.entrySet().iterator(), nullnull);    
  5.     } catch (java.io.IOException cannotHappen) {    
  6.     } catch (ClassNotFoundException cannotHappen) {    
  7.     }    
  8. }    
    首先将比较器指定为m的比较器,这取决于生成m时调用构造方法是否传入了指定的构造器,而后调用buildFromSorted方法,将SortedMap中的元素插入到TreeMap中,由于SortedMap中的元素师有序的,实际上它是根据SortedMap创建的TreeMap,将SortedMap中对应的元素添加到TreeMap中。

    插入删除

    插入操作即对应TreeMap的put方法,put操作实际上只需按照二叉排序树的插入步骤来操作即可,插入到指定位置后,再做调整,使其保持红黑树的特性。put源码的实现:

[java]  view plain copy
  1. public V put(K key, V value) {    
  2.     Entry<K,V> t = root;    
  3.     // 若红黑树为空,则插入根节点    
  4.     if (t == null) {    
  5.     // TBD:    
  6.     // 5045147: (coll) Adding null to an empty TreeSet should    
  7.     // throw NullPointerException    
  8.     //    
  9.     // compare(key, key); // type check    
  10.         root = new Entry<K,V>(key, value, null);    
  11.         size = 1;    
  12.         modCount++;    
  13.         return null;    
  14.     }    
  15.     int cmp;    
  16.     Entry<K,V> parent;    
  17.     // split comparator and comparable paths    
  18.     Comparator<? super K> cpr = comparator;    
  19.     // 找出(key, value)在二叉排序树中的插入位置。    
  20.     // 红黑树是以key来进行排序的,所以这里以key来进行查找。    
  21.     if (cpr != null) {    
  22.         do {    
  23.             parent = t;    
  24.             cmp = cpr.compare(key, t.key);    
  25.             if (cmp < 0)    
  26.                 t = t.left;    
  27.             else if (cmp > 0)    
  28.                 t = t.right;    
  29.             else   
  30.                 return t.setValue(value);    
  31.         } while (t != null);    
  32.     }    
  33.     else {    
  34.         if (key == null)    
  35.             throw new NullPointerException();    
  36.         Comparable<? super K> k = (Comparable<? super K>) key;    
  37.         do {    
  38.             parent = t;    
  39.             cmp = k.compareTo(t.key);    
  40.             if (cmp < 0)    
  41.                 t = t.left;    
  42.             else if (cmp > 0)    
  43.                 t = t.right;    
  44.             else   
  45.                 return t.setValue(value);    
  46.         } while (t != null);    
  47.     }    
  48.     // 为(key-value)新建节点    
  49.     Entry<K,V> e = new Entry<K,V>(key, value, parent);    
  50.     if (cmp < 0)    
  51.         parent.left = e;    
  52.     else   
  53.         parent.right = e;    
  54.     // 插入新的节点后,调用fixAfterInsertion调整红黑树。    
  55.     fixAfterInsertion(e);    
  56.     size++;    
  57.     modCount++;    
  58.     return null;    
  59. }    
    这里的fixAfterInsertion便是节点插入后对树进行调整的方法,这里不做介绍。
    删除操作及对应TreeMap的deleteEntry方法,deleteEntry方法同样也只需按照二叉排序树的操作步骤实现即可,删除指定节点后,再对树进行调整即可。deleteEntry方法的实现源码如下:

[java]  view plain copy
  1. // 删除“红黑树的节点p”    
  2. private void deleteEntry(Entry<K,V> p) {    
  3.     modCount++;    
  4.     size--;    
  5.   
  6.     if (p.left != null && p.right != null) {    
  7.         Entry<K,V> s = successor (p);    
  8.         p.key = s.key;    
  9.         p.value = s.value;    
  10.         p = s;    
  11.     }   
  12.   
  13.     Entry<K,V> replacement = (p.left != null ? p.left : p.right);    
  14.   
  15.     if (replacement != null) {    
  16.         replacement.parent = p.parent;    
  17.         if (p.parent == null)    
  18.             root = replacement;    
  19.         else if (p == p.parent.left)    
  20.             p.parent.left  = replacement;    
  21.         else   
  22.             p.parent.right = replacement;    
  23.   
  24.         p.left = p.right = p.parent = null;    
  25.   
  26.         if (p.color == BLACK)    
  27.             fixAfterDeletion(replacement);    
  28.     } else if (p.parent == null) {   
  29.         root = null;    
  30.     } else {  
  31.         if (p.color == BLACK)    
  32.             fixAfterDeletion(p);    
  33.   
  34.         if (p.parent != null) {    
  35.             if (p == p.parent.left)    
  36.                 p.parent.left = null;    
  37.             else if (p == p.parent.right)    
  38.                 p.parent.right = null;    
  39.             p.parent = null;    
  40.         }    
  41.     }    
  42. }    
    后面的fixAfterDeletion方法便是节点删除后对树进行调整的方法,这里不做介绍。

    其他很多方法这里不再一一介绍。

几点总结

    本文对TreeMap的分析较前几篇文章有些浅尝辄止,TreeMap用的没有HashMap那么多,我们有个宏观上的把我和比较即可。

    1、TreeMap是根据key进行排序的,它的排序和定位需要依赖比较器或覆写Comparable接口,也因此不需要key覆写hashCode方法和equals方法,就可以排除掉重复的key,而HashMap的key则需要通过覆写hashCode方法和equals方法来确保没有重复的key。

    2、TreeMap的查询、插入、删除效率均没有HashMap高,一般只有要对key排序时才使用TreeMap。

    3、TreeMap的key不能为null,而HashMap的key可以为null。


    注:对TreeSet和HashSet的源码不再进行剖析,二者分别是基于TreeMap和HashMap实现的,只是对应的节点中只有key,而没有value,因此对TreeMap和HashMap比较了解的话,对TreeSet和HashSet的理解就会非常容易。




LinkedHashmap源码剖析

LinkedHashMap简介

    LinkedHashMap是HashMap的子类,与HashMap有着同样的存储结构,但它加入了一个双向链表的头结点,将所有put到LinkedHashmap的节点一一串成了一个双向循环链表,因此它保留了节点插入的顺序,可以使节点的输出顺序与输入顺序相同。

    LinkedHashMap可以用来实现LRU算法(这会在下面的源码中进行分析)。

    LinkedHashMap同样是非线程安全的,只在单线程环境下使用。

    LinkedHashMap源码剖析

    LinkedHashMap源码如下(加入了详细的注释):

[java]  view plain copy Java集合源码剖析(三)【TreeMap、LinkedHashmap】 Java集合源码剖析(三)【TreeMap、LinkedHashmap】
  1. package java.util;  
  2. import java.io.*;  
  3.   
  4.   
  5. public class LinkedHashMap<K,V>  
  6.     extends HashMap<K,V>  
  7.     implements Map<K,V>  
  8. {  
  9.   
  10.     private static final long serialVersionUID = 3801124242820219131L;  
  11.   
  12.     //双向循环链表的头结点,整个LinkedHa只哟shMap中只有一个header,  
  13.     //它将哈希表中所有的Entry贯穿起来,header中不保存key-value对,只保存前后节点的引用  
  14.     private transient Entry<K,V> header;  
  15.   
  16.     //双向链表中元素排序规则的标志位。  
  17.     //accessOrder为false,表示按插入顺序排序  
  18.     //accessOrder为true,表示按访问顺序排序  
  19.     private final boolean accessOrder;  
  20.   
  21.     //调用HashMap的构造方法来构造底层的数组  
  22.     public LinkedHashMap(int initialCapacity, float loadFactor) {  
  23.         super(initialCapacity, loadFactor);  
  24.         accessOrder = false;    //链表中的元素默认按照插入顺序排序  
  25.     }  
  26.   
  27.     //加载因子取默认的0.75f  
  28.     public LinkedHashMap(int initialCapacity) {  
  29.         super(initialCapacity);  
  30.         accessOrder = false;  
  31.     }  
  32.   
  33.     //加载因子取默认的0.75f,容量取默认的16  
  34.     public LinkedHashMap() {  
  35.         super();  
  36.         accessOrder = false;  
  37.     }  
  38.   
  39.     //含有子Map的构造方法,同样调用HashMap的对应的构造方法  
  40.     public LinkedHashMap(Map<? extends K, ? extends V> m) {  
  41.         super(m);  
  42.         accessOrder = false;  
  43.     }  
  44.   
  45.     //该构造方法可以指定链表中的元素排序的规则  
  46.     public LinkedHashMap(int initialCapacity,float loadFactor,boolean accessOrder) {  
  47.         super(initialCapacity, loadFactor);  
  48.         this.accessOrder = accessOrder;  
  49.     }  
  50.   
  51.     //覆写父类的init()方法(HashMap中的init方法为空),  
  52.     //该方法在父类的构造方法和Clone、readObject中在插入元素前被调用,  
  53.     //初始化一个空的双向循环链表,头结点中不保存数据,头结点的下一个节点才开始保存数据。  
  54.     void init() {  
  55.         header = new Entry<K,V>(-1nullnullnull);  
  56.         header.before = header.after = header;  
  57.     }  
  58.   
  59.   
  60.     //覆写HashMap中的transfer方法,它在父类的resize方法中被调用,  
  61.     //扩容后,将key-value对重新映射到新的newTable中  
  62.     //覆写该方法的目的是为了提高复制的效率,  
  63.     //这里充分利用双向循环链表的特点进行迭代,不用对底层的数组进行for循环。  
  64.     void transfer(HashMap.Entry[] newTable) {  
  65.         int newCapacity = newTable.length;  
  66.         for (Entry<K,V> e = header.after; e != header; e = e.after) {  
  67.             int index = indexFor(e.hash, newCapacity);  
  68.             e.next = newTable[index];  
  69.             newTable[index] = e;  
  70.         }  
  71.     }  
  72.   
  73.   
  74.     //覆写HashMap中的containsValue方法,  
  75.     //覆写该方法的目的同样是为了提高查询的效率,  
  76.     //利用双向循环链表的特点进行查询,少了对数组的外层for循环  
  77.     public boolean containsValue(Object value) {  
  78.         // Overridden to take advantage of faster iterator  
  79.         if (value==null) {  
  80.             for (Entry e = header.after; e != header; e = e.after)  
  81.                 if (e.value==null)  
  82.                     return true;  
  83.         } else {  
  84.             for (Entry e = header.after; e != header; e = e.after)  
  85.                 if (value.equals(e.value))  
  86.                     return true;  
  87.         }  
  88.         return false;  
  89.     }  
  90.   
  91.   
  92.     //覆写HashMap中的get方法,通过getEntry方法获取Entry对象。  
  93.     //注意这里的recordAccess方法,  
  94.     //如果链表中元素的排序规则是按照插入的先后顺序排序的话,该方法什么也不做,  
  95.     //如果链表中元素的排序规则是按照访问的先后顺序排序的话,则将e移到链表的末尾处。  
  96.     public V get(Object key) {  
  97.         Entry<K,V> e = (Entry<K,V>)getEntry(key);  
  98.         if (e == null)  
  99.             return null;  
  100.         e.recordAccess(this);  
  101.         return e.value;  
  102.     }  
  103.   
  104.     //清空HashMap,并将双向链表还原为只有头结点的空链表  
  105.     public void clear() {  
  106.         super.clear();  
  107.         header.before = header.after = header;  
  108.     }  
  109.   
  110.     //Enty的数据结构,多了两个指向前后节点的引用  
  111.     private static class Entry<K,V> extends HashMap.Entry<K,V> {  
  112.         // These fields comprise the doubly linked list used for iteration.  
  113.         Entry<K,V> before, after;  
  114.   
  115.         //调用父类的构造方法  
  116.         Entry(int hash, K key, V value, HashMap.Entry<K,V> next) {  
  117.             super(hash, key, value, next);  
  118.         }  
  119.   
  120.         //双向循环链表中,删除当前的Entry  
  121.         private void remove() {  
  122.             before.after = after;  
  123.             after.before = before;  
  124.         }  
  125.   
  126.         //双向循环立链表中,将当前的Entry插入到existingEntry的前面  
  127.         private void addBefore(Entry<K,V> existingEntry) {  
  128.             after  = existingEntry;  
  129.             before = existingEntry.before;  
  130.             before.after = this;  
  131.             after.before = this;  
  132.         }  
  133.   
  134.   
  135.         //覆写HashMap中的recordAccess方法(HashMap中该方法为空),  
  136.         //当调用父类的put方法,在发现插入的key已经存在时,会调用该方法,  
  137.         //调用LinkedHashmap覆写的get方法时,也会调用到该方法,  
  138.         //该方法提供了LRU算法的实现,它将最近使用的Entry放到双向循环链表的尾部,  
  139.         //accessOrder为true时,get方法会调用recordAccess方法  
  140.         //put方法在覆盖key-value对时也会调用recordAccess方法  
  141.         //它们导致Entry最近使用,因此将其移到双向链表的末尾  
  142.         void recordAccess(HashMap<K,V> m) {  
  143.             LinkedHashMap<K,V> lm = (LinkedHashMap<K,V>)m;  
  144.             //如果链表中元素按照访问顺序排序,则将当前访问的Entry移到双向循环链表的尾部,  
  145.             //如果是按照插入的先后顺序排序,则不做任何事情。  
  146.             if (lm.accessOrder) {  
  147.                 lm.modCount++;  
  148.                 //移除当前访问的Entry  
  149.                 remove();  
  150.                 //将当前访问的Entry插入到链表的尾部  
  151.                 addBefore(lm.header);  
  152.             }  
  153.         }  
  154.   
  155.         void recordRemoval(HashMap<K,V> m) {  
  156.             remove();  
  157.         }  
  158.     }  
  159.   
  160.     //迭代器  
  161.     private abstract class LinkedHashIterator<T> implements Iterator<T> {  
  162.     Entry<K,V> nextEntry    = header.after;  
  163.     Entry<K,V> lastReturned = null;  
  164.   
  165.     /** 
  166.      * The modCount value that the iterator believes that the backing 
  167.      * List should have.  If this expectation is violated, the iterator 
  168.      * has detected concurrent modification. 
  169.      */  
  170.     int expectedModCount = modCount;  
  171.   
  172.     public boolean hasNext() {  
  173.             return nextEntry != header;  
  174.     }  
  175.   
  176.     public void remove() {  
  177.         if (lastReturned == null)  
  178.         throw new IllegalStateException();  
  179.         if (modCount != expectedModCount)  
  180.         throw new ConcurrentModificationException();  
  181.   
  182.             LinkedHashMap.this.remove(lastReturned.key);  
  183.             lastReturned = null;  
  184.             expectedModCount = modCount;  
  185.     }  
  186.   
  187.     //从head的下一个节点开始迭代  
  188.     Entry<K,V> nextEntry() {  
  189.         if (modCount != expectedModCount)  
  190.         throw new ConcurrentModificationException();  
  191.             if (nextEntry == header)  
  192.                 throw new NoSuchElementException();  
  193.   
  194.             Entry<K,V> e = lastReturned = nextEntry;  
  195.             nextEntry = e.after;  
  196.             return e;  
  197.     }  
  198.     }  
  199.   
  200.     //key迭代器  
  201.     private class KeyIterator extends LinkedHashIterator<K> {  
  202.     public K next() { return nextEntry().getKey(); }  
  203.     }  
  204.   
  205.     //value迭代器  
  206.     private class ValueIterator extends LinkedHashIterator<V> {  
  207.     public V next() { return nextEntry().value; }  
  208.     }  
  209.   
  210.     //Entry迭代器  
  211.     private class EntryIterator extends LinkedHashIterator<Map.Entry<K,V>> {  
  212.     public Map.Entry<K,V> next() { return nextEntry(); }  
  213.     }  
  214.   
  215.     // These Overrides alter the behavior of superclass view iterator() methods  
  216.     Iterator<K> newKeyIterator()   { return new KeyIterator();   }  
  217.     Iterator<V> newValueIterator() { return new ValueIterator(); }  
  218.     Iterator<Map.Entry<K,V>> newEntryIterator() { return new EntryIterator(); }  
  219.   
  220.   
  221.     //覆写HashMap中的addEntry方法,LinkedHashmap并没有覆写HashMap中的put方法,  
  222.     //而是覆写了put方法所调用的addEntry方法和recordAccess方法,  
  223.     //put方法在插入的key已存在的情况下,会调用recordAccess方法,  
  224.     //在插入的key不存在的情况下,要调用addEntry插入新的Entry  
  225.     void addEntry(int hash, K key, V value, int bucketIndex) {  
  226.         //创建新的Entry,并插入到LinkedHashMap中  
  227.         createEntry(hash, key, value, bucketIndex);  
  228.   
  229.         //双向链表的第一个有效节点(header后的那个节点)为近期最少使用的节点  
  230.         Entry<K,V> eldest = header.after;  
  231.         //如果有必要,则删除掉该近期最少使用的节点,  
  232.         //这要看对removeEldestEntry的覆写,由于默认为false,因此默认是不做任何处理的。  
  233.         if (removeEldestEntry(eldest)) {  
  234.             removeEntryForKey(eldest.key);  
  235.         } else {  
  236.             //扩容到原来的2倍  
  237.             if (size >= threshold)  
  238.                 resize(2 * table.length);  
  239.         }  
  240.     }  
  241.   
  242.     void createEntry(int hash, K key, V value, int bucketIndex) {  
  243.         //创建新的Entry,并将其插入到数组对应槽的单链表的头结点处,这点与HashMap中相同  
  244.         HashMap.Entry<K,V> old = table[bucketIndex];  
  245.         Entry<K,V> e = new Entry<K,V>(hash, key, value, old);  
  246.         table[bucketIndex] = e;  
  247.         //每次插入Entry时,都将其移到双向链表的尾部,  
  248.         //这便会按照Entry插入LinkedHashMap的先后顺序来迭代元素,  
  249.         //同时,新put进来的Entry是最近访问的Entry,把其放在链表末尾 ,符合LRU算法的实现  
  250.         e.addBefore(header);  
  251.         size++;  
  252.     }  
  253.   
  254.     //该方法是用来被覆写的,一般如果用LinkedHashmap实现LRU算法,就要覆写该方法,  
  255.     //比如可以将该方法覆写为如果设定的内存已满,则返回true,这样当再次向LinkedHashMap中put  
  256.     //Entry时,在调用的addEntry方法中便会将近期最少使用的节点删除掉(header后的那个节点)。  
  257.     protected boolean removeEldestEntry(Map.Entry<K,V> eldest) {  
  258.         return false;  
  259.     }  
  260. }  

    几点总结

    关于LinkedHashMap的源码,给出以下几点比较重要的总结:

    1、从源码中可以看出,LinkedHashMap中加入了一个head头结点,将所有插入到该LinkedHashMap中的Entry按照插入的先后顺序依次加入到以head为头结点的双向循环链表的尾部。

Java集合源码剖析(三)【TreeMap、LinkedHashmap】Java集合源码剖析(三)【TreeMap、LinkedHashmap】

    实际上就是HashMap和LinkedList两个集合类的存储结构的结合。在LinkedHashMapMap中,所有put进来的Entry都保存在如第一个图所示的哈希表中,但它又额外定义了一个以head为头结点的空的双向循环链表,每次put进来Entry,除了将其保存到对哈希表中对应的位置上外,还要将其插入到双向循环链表的尾部。

    2、LinkedHashMap由于继承自HashMap,因此它具有HashMap的所有特性,同样允许key和value为null。

    3、注意源码中的accessOrder标志位,当它false时,表示双向链表中的元素按照Entry插入LinkedHashMap到中的先后顺序排序,即每次put到LinkedHashMap中的Entry都放在双向链表的尾部,这样遍历双向链表时,Entry的输出顺序便和插入的顺序一致,这也是默认的双向链表的存储顺序;当它为true时,表示双向链表中的元素按照访问的先后顺序排列,可以看到,虽然Entry插入链表的顺序依然是按照其put到LinkedHashMap中的顺序,但put和get方法均有调用recordAccess方法(put方法在key相同,覆盖原有的Entry的情况下调用recordAccess方法),该方法判断accessOrder是否为true,如果是,则将当前访问的Entry(put进来的Entry或get出来的Entry)移到双向链表的尾部(key不相同时,put新Entry时,会调用addEntry,它会调用creatEntry,该方法同样将新插入的元素放入到双向链表的尾部,既符合插入的先后顺序,又符合访问的先后顺序,因为这时该Entry也被访问了),否则,什么也不做。

    4、注意构造方法,前四个构造方法都将accessOrder设为false,说明默认是按照插入顺序排序的,而第五个构造方法可以自定义传入的accessOrder的值,因此可以指定双向循环链表中元素的排序规则,一般要用LinkedHashMap实现LRU算法,就要用该构造方法,将accessOrder置为true。

    5、LinkedHashMap并没有覆写HashMap中的put方法,而是覆写了put方法中调用的addEntry方法和recordAccess方法,我们回过头来再看下HashMap的put方法:

[java]  view plain copy Java集合源码剖析(三)【TreeMap、LinkedHashmap】 Java集合源码剖析(三)【TreeMap、LinkedHashmap】
  1. // 将“key-value”添加到HashMap中      
  2. public V put(K key, V value) {      
  3.     // 若“key为null”,则将该键值对添加到table[0]中。      
  4.     if (key == null)      
  5.         return putForNullKey(value);      
  6.     // 若“key不为null”,则计算该key的哈希值,然后将其添加到该哈希值对应的链表中。      
  7.     int hash = hash(key.hashCode());      
  8.     int i = indexFor(hash, table.length);      
  9.     for (Entry<K,V> e = table[i]; e != null; e = e.next) {      
  10.         Object k;      
  11.         // 若“该key”对应的键值对已经存在,则用新的value取代旧的value。然后退出!      
  12.         if (e.hash == hash && ((k = e.key) == key || key.equals(k))) {      
  13.             V oldValue = e.value;      
  14.             e.value = value;      
  15.             e.recordAccess(this);      
  16.             return oldValue;      
  17.         }      
  18.     }      
  19.   
  20.     // 若“该key”对应的键值对不存在,则将“key-value”添加到table中      
  21.     modCount++;    
  22.     //将key-value添加到table[i]处    
  23.     addEntry(hash, key, value, i);      
  24.     return null;      
  25. }      
    当要put进来的Entry的key在哈希表中已经在存在时,会调用recordAccess方法,当该key不存在时,则会调用addEntry方法将新的Entry插入到对应槽的单链表的头部。

    我们先来看recordAccess方法:

[java]  view plain copy Java集合源码剖析(三)【TreeMap、LinkedHashmap】 Java集合源码剖析(三)【TreeMap、LinkedHashmap】
  1. //覆写HashMap中的recordAccess方法(HashMap中该方法为空),  
  2. //当调用父类的put方法,在发现插入的key已经存在时,会调用该方法,  
  3. //调用LinkedHashmap覆写的get方法时,也会调用到该方法,  
  4. //该方法提供了LRU算法的实现,它将最近使用的Entry放到双向循环链表的尾部,  
  5. //accessOrder为true时,get方法会调用recordAccess方法  
  6. //put方法在覆盖key-value对时也会调用recordAccess方法  
  7. //它们导致Entry最近使用,因此将其移到双向链表的末尾  
  8.       void recordAccess(HashMap<K,V> m) {  
  9.           LinkedHashMap<K,V> lm = (LinkedHashMap<K,V>)m;  
  10.     //如果链表中元素按照访问顺序排序,则将当前访问的Entry移到双向循环链表的尾部,  
  11.     //如果是按照插入的先后顺序排序,则不做任何事情。  
  12.           if (lm.accessOrder) {  
  13.               lm.modCount++;  
  14.         //移除当前访问的Entry  
  15.               remove();  
  16.         //将当前访问的Entry插入到链表的尾部  
  17.               addBefore(lm.header);  
  18.           }  
  19.       }  
    该方法会判断accessOrder是否为true,如果为true,它会将当前访问的Entry(在这里指put进来的Entry)移动到双向循环链表的尾部,从而实现双向链表中的元素按照访问顺序来排序(最近访问的Entry放到链表的最后,这样多次下来,前面就是最近没有被访问的元素,在实现、LRU算法时,当双向链表中的节点数达到最大值时,将前面的元素删去即可,因为前面的元素是最近最少使用的),否则什么也不做。
    再来看addEntry方法:

[java]  view plain copy Java集合源码剖析(三)【TreeMap、LinkedHashmap】 Java集合源码剖析(三)【TreeMap、LinkedHashmap】
  1. //覆写HashMap中的addEntry方法,LinkedHashmap并没有覆写HashMap中的put方法,  
  2. //而是覆写了put方法所调用的addEntry方法和recordAccess方法,  
  3. //put方法在插入的key已存在的情况下,会调用recordAccess方法,  
  4. //在插入的key不存在的情况下,要调用addEntry插入新的Entry  
  5.    void addEntry(int hash, K key, V value, int bucketIndex) {  
  6.     //创建新的Entry,并插入到LinkedHashMap中  
  7.        createEntry(hash, key, value, bucketIndex);  
  8.   
  9.        //双向链表的第一个有效节点(header后的那个节点)为近期最少使用的节点  
  10.        Entry<K,V> eldest = header.after;  
  11.     //如果有必要,则删除掉该近期最少使用的节点,  
  12.     //这要看对removeEldestEntry的覆写,由于默认为false,因此默认是不做任何处理的。  
  13.        if (removeEldestEntry(eldest)) {  
  14.            removeEntryForKey(eldest.key);  
  15.        } else {  
  16.         //扩容到原来的2倍  
  17.            if (size >= threshold)  
  18.                resize(2 * table.length);  
  19.        }  
  20.    }  
  21.   
  22.    void createEntry(int hash, K key, V value, int bucketIndex) {  
  23.     //创建新的Entry,并将其插入到数组对应槽的单链表的头结点处,这点与HashMap中相同  
  24.        HashMap.Entry<K,V> old = table[bucketIndex];  
  25.     Entry<K,V> e = new Entry<K,V>(hash, key, value, old);  
  26.        table[bucketIndex] = e;  
  27.     //每次插入Entry时,都将其移到双向链表的尾部,  
  28.     //这便会按照Entry插入LinkedHashMap的先后顺序来迭代元素,  
  29.     //同时,新put进来的Entry是最近访问的Entry,把其放在链表末尾 ,符合LRU算法的实现  
  30.        e.addBefore(header);  
  31.        size++;  
  32.    }  
    同样是将新的Entry插入到table中对应槽所对应单链表的头结点中,但可以看出,在createEntry中,同样把新put进来的Entry插入到了双向链表的尾部,从插入顺序的层面来说,新的Entry插入到双向链表的尾部,可以实现按照插入的先后顺序来迭代Entry,而从访问顺序的层面来说,新put进来的Entry又是最近访问的Entry,也应该将其放在双向链表的尾部。

    上面还有个removeEldestEntry方法,该方法如下:

[java]  view plain copy Java集合源码剖析(三)【TreeMap、LinkedHashmap】 Java集合源码剖析(三)【TreeMap、LinkedHashmap】
  1.     //该方法是用来被覆写的,一般如果用LinkedHashmap实现LRU算法,就要覆写该方法,  
  2.     //比如可以将该方法覆写为如果设定的内存已满,则返回true,这样当再次向LinkedHashMap中put  
  3.     //Entry时,在调用的addEntry方法中便会将近期最少使用的节点删除掉(header后的那个节点)。  
  4.     protected boolean removeEldestEntry(Map.Entry<K,V> eldest) {  
  5.         return false;  
  6.     }  
  7. }  
    该方法默认返回false,我们一般在用LinkedHashMap实现LRU算法时,要覆写该方法,一般的实现是,当设定的内存(这里指节点个数)达到最大值时,返回true,这样put新的Entry(该Entry的key在哈希表中没有已经存在)时,就会调用removeEntryForKey方法,将最近最少使用的节点删除(head后面的那个节点,实际上是最近没有使用)。
    6、LinkedHashMap覆写了HashMap的get方法:

[java]  view plain copy Java集合源码剖析(三)【TreeMap、LinkedHashmap】 Java集合源码剖析(三)【TreeMap、LinkedHashmap】
  1. //覆写HashMap中的get方法,通过getEntry方法获取Entry对象。  
  2. //注意这里的recordAccess方法,  
  3. //如果链表中元素的排序规则是按照插入的先后顺序排序的话,该方法什么也不做,  
  4. //如果链表中元素的排序规则是按照访问的先后顺序排序的话,则将e移到链表的末尾处。  
  5.    public V get(Object key) {  
  6.        Entry<K,V> e = (Entry<K,V>)getEntry(key);  
  7.        if (e == null)  
  8.            return null;  
  9.        e.recordAccess(this);  
  10.        return e.value;  
  11.    }  
    先取得Entry,如果不为null,一样调用recordAccess方法,上面已经说得很清楚,这里不在多解释了。

    7、最后说说LinkedHashMap是如何实现LRU的。首先,当accessOrder为true时,才会开启按访问顺序排序的模式,才能用来实现LRU算法。我们可以看到,无论是put方法还是get方法,都会导致目标Entry成为最近访问的Entry,因此便把该Entry加入到了双向链表的末尾(get方法通过调用recordAccess方法来实现,put方法在覆盖已有key的情况下,也是通过调用recordAccess方法来实现,在插入新的Entry时,则是通过createEntry中的addBefore方法来实现),这样便把最近使用了的Entry放入到了双向链表的后面,多次操作后,双向链表前面的Entry便是最近没有使用的,这样当节点个数满的时候,删除的最前面的Entry(head后面的那个Entry)便是最近最少使用的Entry。


注:本集合源码剖析系列文章转自 http://blog.csdn.net/ns_code 感谢博主!