TreeMap源码分析——深入分析（基于JDK1.6）

TreeMap有Values、EntrySet、KeySet、PrivateEntryIterator、EntryIterator、ValueIterator、KeyIterator、DescendingKeyIterator、NavigableSubMap、AscendingSubMap、DescendingSubMap、SubMap、Entry共十三个内部类。Entry是在TreeMap中用于表示树的节点的内部类，已经在《TreeMap源码分析——基础分析》中分析过。下面逐一介绍上面的内部类以及TreeMap中提供的和内部类相关的方法。

先看Values。

 1 // 从类的定义可以看出，Values是一个集合类
 2 class Values extends AbstractCollection<V> {
 3     // 提供集合类Values的迭代器
 4     public Iterator<V> iterator() {
 5         return new ValueIterator(getFirstEntry());
 6     }
 7     // 返回TreeMap中保存的节点数
 8     public int size() {
 9         return TreeMap.this.size();
10     }
11     // 判断TreeMap中是否存在Value为o的节点
12     public boolean contains(Object o) {
13         return TreeMap.this.containsValue(o);
14     }
15     // 删除一个对象
16     public boolean remove(Object o) {
17         // 遍历TreeMap
18         for (Entry<K,V> e = getFirstEntry(); e != null; e = successor(e)) {
19             // 寻找值相等的节点
20             if (valEquals(e.getValue(), o)) {
21                 // 删除找到的节点
22                 deleteEntry(e);
23                 return true;
24             }
25         }
26         return false;
27     }
28     // 清空TreeMap
29     public void clear() {
30         TreeMap.this.clear();
31     }
32 }

Values类实际上是一个代理，多数方法都是调用TreeMap的方法。在Values的iterator()方法中返回了一个ValuesIterator对象，下面来看和迭代器相关的内部类。PrivateEntryIterator是TreeMap中和迭代器相关的类的基础，以下是PrivateEntryIterator的内容。

 1 abstract class PrivateEntryIterator<T> implements Iterator<T> {
 2     // 指向next的引用
 3 Entry<K,V> next;
 4 // 保留对上一次返回节点的引用
 5     Entry<K,V> lastReturned;
 6     int expectedModCount;
 7     // 构造方法，lastReturned置空，next指向传入的节点
 8     PrivateEntryIterator(Entry<K,V> first) {
 9         expectedModCount = modCount;
10         lastReturned = null;
11         next = first;
12     }
13     // 判断是否还有下一个节点
14     public final boolean hasNext() {
15         return next != null;
16     }
17     // 返回下一个节点
18     final Entry<K,V> nextEntry() {
19         Entry<K,V> e = next;
20         if (e == null)
21             throw new NoSuchElementException();
22         if (modCount != expectedModCount)
23             throw new ConcurrentModificationException();
24         // next移动到它的继承者
25         next = successor(e);
26         // 记录被返回的节点
27         lastReturned = e;
28         // 返回原先记录的next节点
29         return e;
30     }
31     // 前一个节点
32     final Entry<K,V> prevEntry() {
33         Entry<K,V> e = next;
34         if (e == null)
35             throw new NoSuchElementException();
36         if (modCount != expectedModCount)
37             throw new ConcurrentModificationException();
38         // 获取指定节点的“前任”（按遍历次序的前一个节点）
39         next = predecessor(e);
40         // 记录被返回的节点
41         lastReturned = e;
42         return e;
43     }
44     // 移除最近一次被返回的节点
45     public void remove() {
46         if (lastReturned == null)
47             throw new IllegalStateException();
48         if (modCount != expectedModCount)
49             throw new ConcurrentModificationException();
50         // deleted entries are replaced by their successors
51         if (lastReturned.left != null && lastReturned.right != null)
52             /* 如果被删除节点有两个孩子，删除节点e的时候e的引用会被修改为指向原节点的继承者，所以这里先保留next对lastReturned的引用，这样在删除节点后就能获取到继承者的引用，继而继续遍历树 */
53             next = lastReturned;
54         // 删除节点
55         deleteEntry(lastReturned);
56         expectedModCount = modCount;
57         lastReturned = null;
58     }
59 }

PrivateEntryIterator类的prevEntry()方法用到了predecessor(Entry<K,V> t)方法，下面对这个方法进行介绍。

predecessor(Entry<K,V> t)方法返回传入节点的“前一个”节点，至于前一个节点是哪个节点，这和树的遍历次序相关。根据successor(Entry<K,V> t)和predecessor(Entry<K,V> t)方法可以推出TreeMap中树的遍历次序是中根遍历（左孩子-根-右孩子）。

 1 static <K,V> Entry<K,V> predecessor(Entry<K,V> t) {
 2     if (t == null)
 3         return null;
 4 else if (t.left != null) {
 5     // 获得左孩子
 6         Entry<K,V> p = t.left;
 7         // 对左孩子进行遍历，获取左孩子最右的子孙
 8         while (p.right != null)
 9             p = p.right;
10         return p;
11 } else {
12     // 获取t的父节点
13         Entry<K,V> p = t.parent;
14         Entry<K,V> ch = t; 
15 // 沿着右孩子向上查找继承者，直到根节点或找到节点ch是其父节点的右孩子的节点
16         while (p != null && ch == p.left) {
17             ch = p;
18             p = p.parent;
19         }
20         return p;
21     }
22 }

下面是TreeMap中其它和迭代器相关的内部类。

 1 // EntryIterator就是树节点的迭代器，和PrivateEntryIterator完全一样，因为提供的方法都直接的调用而来父类的方法。
 2 final class EntryIterator extends PrivateEntryIterator<Map.Entry<K,V>> {
 3     EntryIterator(Entry<K,V> first) {
 4         super(first);
 5     }
 6     public Map.Entry<K,V> next() {
 7         return nextEntry();
 8     }
 9 }
10 /** Value的迭代器 */
11 final class ValueIterator extends PrivateEntryIterator<V> {
12     ValueIterator(Entry<K,V> first) {
13         super(first);
14 }
15 // next()方法返回的是节点的value值
16     public V next() {
17         return nextEntry().value;
18     }
19 }
20 /** Key迭代器 */
21 final class KeyIterator extends PrivateEntryIterator<K> {
22     KeyIterator(Entry<K,V> first) {
23         super(first);
24 }
25 // next()方法返回的是节点的key
26     public K next() {
27         return nextEntry().key;
28     }
29 }
30 /** 逆序的Key迭代器 */
31 final class DescendingKeyIterator extends PrivateEntryIterator<K> {
32     DescendingKeyIterator(Entry<K,V> first) {
33         super(first);
34 }
35 // next()方法返回的是节点的“前任”（按照遍历次序的前一个节点）的key
36     public K next() {
37         return prevEntry().key;
38     }
39 }

除了迭代器相关的内部类，TreeMap还有两个和Set相关的内部类，分别是EntrySet和KeySet。两个类分别表示节点的集合和键的集合。下面具体看这两个类的实现。

 1 // 继承自AbstractSet说明是一个Set
 2 class EntrySet extends AbstractSet<Map.Entry<K,V>> {
 3     // iterator()方法返回的是上面介绍过的EntryIterator对象
 4     public Iterator<Map.Entry<K,V>> iterator() {
 5         return new EntryIterator(getFirstEntry());
 6     }
 7     // 判断是否包含某个节点的方法
 8     public boolean contains(Object o) {
 9         if (!(o instanceof Map.Entry))
10             return false;
11         Map.Entry<K,V> entry = (Map.Entry<K,V>) o;
12         V value = entry.getValue();
13         Entry<K,V> p = getEntry(entry.getKey());
14         // 判断是否包含某个对象的标准是存在节点的key的与传入对象的key值，且该节点的value也与存入对象的value值相等
15         return p != null && valEquals(p.getValue(), value);
16     }
17     // 删除一个对象
18     public boolean remove(Object o) {
19         if (!(o instanceof Map.Entry))
20             return false;
21         Map.Entry<K,V> entry = (Map.Entry<K,V>) o;
22         V value = entry.getValue();
23         Entry<K,V> p = getEntry(entry.getKey());
24         // 如果存在该对象，则进行删除操作并返回true
25         if (p != null && valEquals(p.getValue(), value)) {
26             deleteEntry(p);
27             return true;
28         }
29         // 不存在直接返回false
30         return false;
31     }
32     // size()返回的是TreeMap中包含的节点的数量
33     public int size() {
34         return TreeMap.this.size();
35     }
36     // clear()方法实际调用了TreeMap的clear()方法，和size()方法都是代理方法
37     public void clear() {
38         TreeMap.this.clear();
39     }
40 }

 1 // KeySet同样继承自AbstractSet。KeySet实现了NavigableSet接口，意味着是“可导航”的Set，包含更多的获取指定节点的方法
 2 static final class KeySet<E> extends AbstractSet<E> implements NavigableSet<E> {
 3 private final NavigableMap<E, Object> m;
 4 // 构造方法
 5     KeySet(NavigableMap<E,Object> map) { m = map; }
 6     // 
 7     public Iterator<E> iterator() {
 8         if (m instanceof TreeMap)
 9             return ((TreeMap<E,Object>)m).keyIterator();
10         else
11             // 这里涉及到的NavigableSubMap将在后面介绍
12             return (Iterator<E>)(((TreeMap.NavigableSubMap)m).keyIterator());
13     }
14 
15     public Iterator<E> descendingIterator() {
16         if (m instanceof TreeMap)
17             return ((TreeMap<E,Object>)m).descendingKeyIterator();
18         else
19             return (Iterator<E>)(((TreeMap.NavigableSubMap)m).descendingKeyIterator());
20     }
21     // size()方法返回的是通过构造方法传入的map的大小
22 public int size() { return m.size(); }
23 // isEmpty()判断是否为空也是判断的传入的map是否为空
24 public boolean isEmpty() { return m.isEmpty(); }
25 // contains(Object o)方法判断传入map中是否包含这个key
26     public boolean contains(Object o) { return m.containsKey(o); }
27 public void clear() { m.clear(); }
28 // 因为传入的map是NavigableMap，所以下面这几个方法都是代理方法，调用map中相应的方法
29     public E lower(E e) { return m.lowerKey(e); }
30     public E floor(E e) { return m.floorKey(e); }
31     public E ceiling(E e) { return m.ceilingKey(e); }
32     public E higher(E e) { return m.higherKey(e); }
33     public E first() { return m.firstKey(); }
34 public E last() { return m.lastKey(); }
35 // 获取传入map的比较器
36 public Comparator<? super E> comparator() { return m.comparator(); }
37 // 获取map中第一个节点的key
38     public E pollFirst() {
39         Map.Entry<E,Object> e = m.pollFirstEntry();
40         return e == null? null : e.getKey();
41 }
42 // 获取map中最后一个节点的key
43     public E pollLast() {
44         Map.Entry<E,Object> e = m.pollLastEntry();
45         return e == null? null : e.getKey();
46 }
47 // 删除一个对象，实际上是删除map中以这个对象为key的一个节点
48     public boolean remove(Object o) {
49         int oldSize = size();
50         m.remove(o);
51         return size() != oldSize;
52 }
53 // 下面的方法都是通过NavigableMap和TreeSet实现的，NavigableMap将在下文介绍，TreeSet将另开博文介绍
54     public NavigableSet<E> subSet(E fromElement, boolean fromInclusive,
55                                       E toElement,   boolean toInclusive) {
56         return new TreeSet<E>(m.subMap(fromElement, fromInclusive,
57                                            toElement,   toInclusive));
58     }
59     public NavigableSet<E> headSet(E toElement, boolean inclusive) {
60         return new TreeSet<E>(m.headMap(toElement, inclusive));
61     }
62     public NavigableSet<E> tailSet(E fromElement, boolean inclusive) {
63         return new TreeSet<E>(m.tailMap(fromElement, inclusive));
64     }
65     public SortedSet<E> subSet(E fromElement, E toElement) {
66         return subSet(fromElement, true, toElement, false);
67     }
68     public SortedSet<E> headSet(E toElement) {
69         return headSet(toElement, false);
70     }
71     public SortedSet<E> tailSet(E fromElement) {
72         return tailSet(fromElement, true);
73     }
74     public NavigableSet<E> descendingSet() {
75         return new TreeSet(m.descendingMap());
76     }
77 }

介绍完了两个和Set相关的内部类，现在还剩下四个和SubMap相关的内部类：NavigableSubMap、AscendingSubMap、DescendingSubMap、SubMap。

首先看NavigableSubMap，它足足有400多行代码，相当的多，需要耐心啊。

// NavigableSubMap是一个抽象类，继承了AbstractMap，实现了NavigableMap接口
static abstract class NavigableSubMap<K,V> extends AbstractMap<K,V>
       implements NavigableMap<K,V>, java.io.Serializable {
   // 存储内容的Map
   final TreeMap<K,V> m;
   // lowKey、highKey
   final K lo, hi;
   // 标识map的边界是否是map的第一个节点和最后一个节点
   final boolean fromStart, toEnd;
   // 是否包含最低lo、最高位置hi
final boolean loInclusive, hiInclusive;
// 通过上面的三组变量可以组成两个三元组表示一个集合的两个端点
   // 构造方法
   NavigableSubMap(TreeMap<K,V> m,
                       boolean fromStart, K lo, boolean loInclusive,
                       boolean toEnd,     K hi, boolean hiInclusive) {
       if (!fromStart && !toEnd) {
           // lo>hi抛出异常
           if (m.compare(lo, hi) > 0)
               throw new IllegalArgumentException("fromKey > toKey");
       } else {
           if (!fromStart) // type check
               m.compare(lo, lo);
           if (!toEnd)
               m.compare(hi, hi);
       }

       this.m = m;
       this.fromStart = fromStart;
       this.lo = lo;
       this.loInclusive = loInclusive;
       this.toEnd = toEnd;
       this.hi = hi;
       this.hiInclusive = hiInclusive;
   }

   // tooLow 判断传入的key是否太小
   final boolean tooLow(Object key) {
       // 如果fromStart为false，需要判断最低边界 
       if (!fromStart) {
           int c = m.compare(key, lo);
           // 如果key<lo或者相等但是map的边界不包含lo，那么key越界了，即小于最小值
           if (c < 0 || (c == 0 && !loInclusive))
               return true;
       }
       // 默认返回false
       return false;
   }
   // 与上面的tooLow类似
   final boolean tooHigh(Object key) {
       if (!toEnd) {
           int c = m.compare(key, hi);
           if (c > 0 || (c == 0 && !hiInclusive))
               return true;
       }
       return false;
   }
   // 判断是否在范围内，即满足最低最高限制，结合tooLow和tooHigh即可
   final boolean inRange(Object key) {
       return !tooLow(key) && !tooHigh(key);
   }
   // 是否在封闭的区间内
   final boolean inClosedRange(Object key) {
       return (fromStart || m.compare(key, lo) >= 0)
               && (toEnd || m.compare(hi, key) >= 0);
   }
   // 判断是否是在一个区间内
   final boolean inRange(Object key, boolean inclusive) {
       return inclusive ? inRange(key) : inClosedRange(key);
   }
   // 获取绝对的最低的节点
final TreeMap.Entry<K,V> absLowest() {
/* 如果fromStart为true，获取第一个节点，否则根据loInclusive是否为true，即是否包含lo来决定获取Ceiling节点或Higher节点；getCeilingEntry意为获取指定key的节点或者比指定key大的最小节点，如果不存在则返回null；getHigherEntry意为获取比指定key大的最小节点，如果不存在，返回null */
   TreeMap.Entry<K,V> e =
               (fromStart ?  m.getFirstEntry() :
                (loInclusive ? m.getCeilingEntry(lo) :
                               m.getHigherEntry(lo)));
       // 判断得到的节点是否为空或者key过大
       return (e == null || tooHigh(e.key)) ? null : e;
   }
   // 与absLowest类似
   final TreeMap.Entry<K,V> absHighest() {
   TreeMap.Entry<K,V> e =
               (toEnd ?  m.getLastEntry() :
                (hiInclusive ?  m.getFloorEntry(hi) :
                                m.getLowerEntry(hi)));
       return (e == null || tooLow(e.key)) ? null : e;
   }
   // 寻找大于等于key的最小的节点
final TreeMap.Entry<K,V> absCeiling(K key) {
   // 如果key太小，返回绝对的最小的节点
       if (tooLow(key))
           return absLowest();
       // 获取允许的key的极限节点（满足要求的最小的节点）
       TreeMap.Entry<K,V> e = m.getCeilingEntry(key);
       return (e == null || tooHigh(e.key)) ? null : e;
   }
   // 和absCeiling类似，只是获取的不包含相等的情况，而是寻找大于key的最小节点
   final TreeMap.Entry<K,V> absHigher(K key) {
       if (tooLow(key))
           return absLowest();
       TreeMap.Entry<K,V> e = m.getHigherEntry(key);
       return (e == null || tooHigh(e.key)) ? null : e;
   }
   // 获取绝对的小于等于key的节点
final TreeMap.Entry<K,V> absFloor(K key) {
   // 指定的key超出了hi，直接返回绝对的允许的最大的节点
       if (tooHigh(key))
           return absHighest();
       /* getFloorEntry 获取的是指定key的节点，如果不存在这样的节点，就去获取比指定key小的最大的节点，如果这样的节点也不存在，返回null*/
       TreeMap.Entry<K,V> e = m.getFloorEntry(key);
       return (e == null || tooLow(e.key)) ? null : e;
   }
   // 与上面的absFloor类似，只是不包含等于的情况
   final TreeMap.Entry<K,V> absLower(K key) {
       if (tooHigh(key))
           return absHighest();
       TreeMap.Entry<K,V> e = m.getLowerEntry(key);
       return (e == null || tooLow(e.key)) ? null : e;
   }

   // 返回比最大的节点“还要大”的节点（Fence是栅栏、围栏的意思）
final TreeMap.Entry<K,V> absHighFence() {
   /* 如果toEnd是true，那么“围在”它外面的是null，如果是false，根据hi是否被包含返回getHigherEntry或getCeilingEntry，这两个方法意思在上面的方法中说明过了 */
       return (toEnd ? null : (hiInclusive ?
                                   m.getHigherEntry(hi) :
                                   m.getCeilingEntry(hi)));
   }

   // 与absHighFence类似
   final TreeMap.Entry<K,V> absLowFence() {
           return (fromStart ? null : (loInclusive ?
                                       m.getLowerEntry(lo) :
                                       m.getFloorEntry(lo)));
   }

   abstract TreeMap.Entry<K,V> subLowest();
   abstract TreeMap.Entry<K,V> subHighest();
   abstract TreeMap.Entry<K,V> subCeiling(K key);
   abstract TreeMap.Entry<K,V> subHigher(K key);
   abstract TreeMap.Entry<K,V> subFloor(K key);
   abstract TreeMap.Entry<K,V> subLower(K key);
   abstract Iterator<K> keyIterator();
   abstract Iterator<K> descendingKeyIterator();

   // 如果fromStart、toEnd都是true，那么判断空、获取大小都是直接通过m，不然就必须使用entrySet()先获取节点集
   public boolean isEmpty() {
       return (fromStart && toEnd) ? m.isEmpty() : entrySet().isEmpty();
   }

   public int size() {
       return (fromStart && toEnd) ? m.size() : entrySet().size();
   }
   // 判断是否存在key先判断范围，在通过TreeMap的containKey方法来判断
   public final boolean containsKey(Object key) {
       return inRange(key) && m.containsKey(key);
   }
   // 添加节点
public final V put(K key, V value) {
   // 判断要添加的key是否在范围内
       if (!inRange(key))
           throw new IllegalArgumentException("key out of range");
       return m.put(key, value);
   }
   public final V get(Object key) {
       return !inRange(key)? null :  m.get(key);
   }
   public final V remove(Object key) {
       return !inRange(key)? null  : m.remove(key);
}
public final Map.Entry<K,V> ceilingEntry(K key) {
       /* exportEntry(TreeMap.Entry<K,V> e)方法返回的是Map.Entry<K,V>对象，它的Key 和Value和传入的节点的Key 和Value相同 */
       return exportEntry(subCeiling(key));
   }
public final K ceilingKey(K key) {
       // keyOrNull根据传入的节点是否为null返回null或节点的key（相当于提供了一个null安全的获取key的方法）
       return keyOrNull(subCeiling(key));
   }
   public final Map.Entry<K,V> higherEntry(K key) {
       return exportEntry(subHigher(key));
   }
   public final K higherKey(K key) {
      return keyOrNull(subHigher(key));
   }
   public final Map.Entry<K,V> floorEntry(K key) {
       return exportEntry(subFloor(key));
   }
   public final K floorKey(K key) {
       return keyOrNull(subFloor(key));
   }
   public final Map.Entry<K,V> lowerEntry(K key) {
       return exportEntry(subLower(key));
   }
   public final K lowerKey(K key) {
       return keyOrNull(subLower(key));
   }
   public final K firstKey() {
       return key(subLowest());
   }
   public final K lastKey() {
       return key(subHighest());
   }
   public final Map.Entry<K,V> firstEntry() {
       return exportEntry(subLowest());
   }
   public final Map.Entry<K,V> lastEntry() {
       return exportEntry(subHighest());
   }
   // 返回并删除第一个节点
   public final Map.Entry<K,V> pollFirstEntry() {
   TreeMap.Entry<K,V> e = subLowest();
       Map.Entry<K,V> result = exportEntry(e);
       if (e != null)
           m.deleteEntry(e);
       return result;
   }
   // 返回并删除最后一个节点
   public final Map.Entry<K,V> pollLastEntry() {
   TreeMap.Entry<K,V> e = subHighest();
       Map.Entry<K,V> result = exportEntry(e);
       if (e != null)
           m.deleteEntry(e);
       return result;
   }

   // 这些都是视图
   transient NavigableMap<K,V> descendingMapView = null;
   transient EntrySetView entrySetView = null;
   transient KeySet<K> navigableKeySetView = null;
   // 返回TreeMap的KeySet
   public final NavigableSet<K> navigableKeySet() {
       KeySet<K> nksv = navigableKeySetView;
       return (nksv != null) ? nksv :
           (navigableKeySetView = new TreeMap.KeySet(this));
   }
   public final Set<K> keySet() {
       return navigableKeySet();
   }
   // 逆序的KeySet
   public NavigableSet<K> descendingKeySet() {
       return descendingMap().navigableKeySet();
   }
   // 返回一个子Map
   public final SortedMap<K,V> subMap(K fromKey, K toKey) {
       return subMap(fromKey, true, toKey, false);
   }
   // 下面这几个方法会在后面给出分析
   public final SortedMap<K,V> headMap(K toKey) {
       return headMap(toKey, false);
   }
   public final SortedMap<K,V> tailMap(K fromKey) {
       return tailMap(fromKey, true);
   }

   // 视图类
   abstract class EntrySetView extends AbstractSet<Map.Entry<K,V>> {
       private transient int size = -1, sizeModCount;
       // 返回子Map的大小
       public int size() {
           // 如果fromStart和toEnd都是true，返回的是m的size
           if (fromStart && toEnd)
               return m.size();
           // size=-1或标记size不同，重新计算一次size
           if (size == -1 || sizeModCount != m.modCount) {
               sizeModCount = m.modCount;
               size = 0;
               Iterator i = iterator();
               while (i.hasNext()) {
                   size++;
                   i.next();
               }
           }
           return size;
       }
       // 判断EntrySet是否为空
       public boolean isEmpty() {
           TreeMap.Entry<K,V> n = absLowest();
           return n == null || tooHigh(n.key);
       }
       // 判断是否包含某个对象
       public boolean contains(Object o) {
           if (!(o instanceof Map.Entry))
               return false;
           Map.Entry<K,V> entry = (Map.Entry<K,V>) o;
           K key = entry.getKey();
           // key不在范围内，返回false
           if (!inRange(key))
               return false;
           // 判断是否有键和值如传入节点的键和值相等的节点
           TreeMap.Entry node = m.getEntry(key);
           return node != null &&
               valEquals(node.getValue(), entry.getValue());
       }
       // 移除一个节点
       public boolean remove(Object o) {
           if (!(o instanceof Map.Entry))
               return false;
           Map.Entry<K,V> entry = (Map.Entry<K,V>) o;
           K key = entry.getKey();
           if (!inRange(key))
               return false;
           TreeMap.Entry<K,V> node = m.getEntry(key);
           // 找到节点并移除，返回true
           if (node!=null && valEquals(node.getValue(),entry.getValue())){
               m.deleteEntry(node);
               return true;
           }
           return false;
       }
   }

   //子类迭代器
   abstract class SubMapIterator<T> implements Iterator<T> {
       // 上一次被返回的节点
       TreeMap.Entry<K,V> lastReturned;
       // 下一个节点
       TreeMap.Entry<K,V> next;
       // “栅栏”key（如果是向大的方向遍历，不能访问key大于等于fenceKey的节点；如果是向小的方向遍历，不能访问key小于等于fenceKey的节点）
       final K fenceKey;
       int expectedModCount;
       // 构造方法
       SubMapIterator(TreeMap.Entry<K,V> first,
                      TreeMap.Entry<K,V> fence) {
           expectedModCount = m.modCount;
           lastReturned = null;
           next = first;
           fenceKey = fence == null ? null : fence.key;
       }
       // 判断是否还有下一个节点
       public final boolean hasNext() {
           // 与普通的hasNext的判断不同的是这里必须判断next的key是否超出了fenceKey
           return next != null && next.key != fenceKey;
       }
       // 获得下一个节点的方法，比较容易理解
       final TreeMap.Entry<K,V> nextEntry() {
           TreeMap.Entry<K,V> e = next;
           if (e == null || e.key == fenceKey)
               throw new NoSuchElementException();
           if (m.modCount != expectedModCount)
               throw new ConcurrentModificationException();
           next = successor(e);
lastReturned = e;
           return e;
       }
       // 另一种遍历方法，向前遍历
       final TreeMap.Entry<K,V> prevEntry() {
           TreeMap.Entry<K,V> e = next;
           if (e == null || e.key == fenceKey)
               throw new NoSuchElementException();
           if (m.modCount != expectedModCount)
               throw new ConcurrentModificationException();
           next = predecessor(e);
lastReturned = e;
           return e;
       }
       // 删除节点后可以继续遍历剩余的节点，因为删除前用next保留了lastReturned节点，而这个节点在删除操作的过程中被替换成了它的继承者
       final void removeAscending() {
           if (lastReturned == null)
               throw new IllegalStateException();
           if (m.modCount != expectedModCount)
               throw new ConcurrentModificationException();
           if (lastReturned.left != null && lastReturned.right != null)
               // next指向lastReturned所指向的节点，这个节点的内容在删除lastReturned的时候被改变了
               next = lastReturned;
           m.deleteEntry(lastReturned);
           lastReturned = null;
           expectedModCount = m.modCount;
       }
       // 删除之后next指向的节点其实被删除了，不能继续迭代访问
       final void removeDescending() {
           if (lastReturned == null)
               throw new IllegalStateException();
           if (m.modCount != expectedModCount)
               throw new ConcurrentModificationException();
           m.deleteEntry(lastReturned);
           lastReturned = null;
           expectedModCount = m.modCount;
       }

   }
   //下面的几个内部类很简单，都是对SubMapIterator的调用或间接调用，不再解释
   final class SubMapEntryIterator extends SubMapIterator<Map.Entry<K,V>> {
       SubMapEntryIterator(TreeMap.Entry<K,V> first,
                               TreeMap.Entry<K,V> fence) {
           super(first, fence);
       }
       public Map.Entry<K,V> next() {
           return nextEntry();
       }
       public void remove() {
           removeAscending();
       }
   }

   final class SubMapKeyIterator extends SubMapIterator<K> {
       SubMapKeyIterator(TreeMap.Entry<K,V> first,
                             TreeMap.Entry<K,V> fence) {
           super(first, fence);
       }
       public K next() {
           return nextEntry().key;
       }
       public void remove() {
           removeAscending();
       }
   }

   final class DescendingSubMapEntryIterator extends SubMapIterator<Map.Entry<K,V>> {
       DescendingSubMapEntryIterator(TreeMap.Entry<K,V> last,
                                         TreeMap.Entry<K,V> fence) {
           super(last, fence);
       }

       public Map.Entry<K,V> next() {
           return prevEntry();
       }
       public void remove() {
           removeDescending();
       }
   }

   final class DescendingSubMapKeyIterator extends SubMapIterator<K> {
       DescendingSubMapKeyIterator(TreeMap.Entry<K,V> last,
                                       TreeMap.Entry<K,V> fence) {
           super(last, fence);
       }
       public K next() {
           return prevEntry().key;
       }
       public void remove() {
           removeDescending();
       }
   }
}

NavigableSubMap类算是看了一遍，很复杂，自身是个内部类，它里面还包含了好几个类。理解它的代码需要部分TreeMap中的其他代码的深入理解，如涉及到的deleteEntry等方法（见《TreeMap源码分析——基础分析》）。

下面看TreeMap的其他内部类，它们是NavigableSubMap的子类。

AscendingSubMap

 1 // AscendingSubMap继承自NavigableSubMap
 2 static final class AscendingSubMap<K,V> extends NavigableSubMap<K,V> {
 3     private static final long serialVersionUID = 912986545866124060L;
 4     // 构造方法，直接调用父类构造方法
 5     AscendingSubMap(TreeMap<K,V> m,
 6                         boolean fromStart, K lo, boolean loInclusive,
 7                         boolean toEnd,     K hi, boolean hiInclusive) {
 8         super(m, fromStart, lo, loInclusive, toEnd, hi, hiInclusive);
 9     }
10     // 获得比较器
11     public Comparator<? super K> comparator() {
12         return m.comparator();
13     }
14     // “截取”子Map
15     public NavigableMap<K,V> subMap(K fromKey, boolean fromInclusive,
16                                         K toKey,   boolean toInclusive) {
17         // 截取之前判断是否超出范围
18         if (!inRange(fromKey, fromInclusive))
19             throw new IllegalArgumentException("fromKey out of range");
20         if (!inRange(toKey, toInclusive))
21             throw new IllegalArgumentException("toKey out of range");
22         return new AscendingSubMap(m,
23                                        false, fromKey, fromInclusive,
24                                        false, toKey,   toInclusive);
25     }
26     // “截取”子Map，headMap通过构造方法便可以实现
27     public NavigableMap<K,V> headMap(K toKey, boolean inclusive) {
28         if (!inRange(toKey, inclusive))
29             throw new IllegalArgumentException("toKey out of range");
30         return new AscendingSubMap(m,
31                                        fromStart, lo,    loInclusive,
32                                        false,     toKey, inclusive);
33     }
34     // 和headMap类似
35     public NavigableMap<K,V> tailMap(K fromKey, boolean inclusive){
36         if (!inRange(fromKey, inclusive))
37             throw new IllegalArgumentException("fromKey out of range");
38         return new AscendingSubMap(m,
39                                        false, fromKey, inclusive,
40                                        toEnd, hi,      hiInclusive);
41     }
42     // 这个方法涉及到DescendingSubMap类的构造方法，在下面会介绍到
43     public NavigableMap<K,V> descendingMap() {
44         NavigableMap<K,V> mv = descendingMapView;
45         return (mv != null) ? mv :
46                 (descendingMapView =
47                  new DescendingSubMap(m,
48                                       fromStart, lo, loInclusive,
49                                       toEnd,     hi, hiInclusive));
50     }
51     // 下面两个方法都是对上面提到过的构造方法的调用
52     Iterator<K> keyIterator() {
53         return new SubMapKeyIterator(absLowest(), absHighFence());
54     }
55     Iterator<K> descendingKeyIterator() {
56         return new DescendingSubMapKeyIterator(absHighest(), absLowFence());
57 }
58     // AscendingEntrySetView是一个视图类，重写了父类的iterator()方法，调用SubMapEntryIterator构造迭代器
59     final class AscendingEntrySetView extends EntrySetView {
60         public Iterator<Map.Entry<K,V>> iterator() {
61             return new SubMapEntryIterator(absLowest(), absHighFence());
62         }
63     }
64     // 获取节点集合的方法
65     public Set<Map.Entry<K,V>> entrySet() {
66         EntrySetView es = entrySetView;
67         return (es != null) ? es : new AscendingEntrySetView();
68     }
69     // 父类中抽象方法的实现，都很简单
70     TreeMap.Entry<K,V> subLowest()       { return absLowest(); }
71     TreeMap.Entry<K,V> subHighest()      { return absHighest(); }
72     TreeMap.Entry<K,V> subCeiling(K key) { return absCeiling(key); }
73     TreeMap.Entry<K,V> subHigher(K key)  { return absHigher(key); }
74     TreeMap.Entry<K,V> subFloor(K key)   { return absFloor(key); }
75     TreeMap.Entry<K,V> subLower(K key)   { return absLower(key); }
76 }

DescendingSubMap

 1 // DescendingSubMap也继承自NavigableSubMap，和上面的AscendingSubMap对应
 2 static final class DescendingSubMap<K,V>  extends NavigableSubMap<K,V> {
 3     private static final long serialVersionUID = 912986545866120460L;
 4     DescendingSubMap(TreeMap<K,V> m,
 5                         boolean fromStart, K lo, boolean loInclusive,
 6                         boolean toEnd,     K hi, boolean hiInclusive) {
 7         super(m, fromStart, lo, loInclusive, toEnd, hi, hiInclusive);
 8     }
 9     // 构造一个“相反”的比较器
10     private final Comparator<? super K> reverseComparator =
11         Collections.reverseOrder(m.comparator);
12     // 获取的比较器是“相反”的比较器，比较结果会对调
13     public Comparator<? super K> comparator() {
14         return reverseComparator;
15     }
16     // subMap方法和AscendingSubMap类中类似
17     public NavigableMap<K,V> subMap(K fromKey, boolean fromInclusive,
18                                         K toKey,   boolean toInclusive) {
19         if (!inRange(fromKey, fromInclusive))
20             throw new IllegalArgumentException("fromKey out of range");
21         if (!inRange(toKey, toInclusive))
22             throw new IllegalArgumentException("toKey out of range");
23         return new DescendingSubMap(m,
24                                         false, toKey,   toInclusive,
25                                         false, fromKey, fromInclusive);
26     }
27     // 与AscendingSubMap中其实是相反的
28     public NavigableMap<K,V> headMap(K toKey, boolean inclusive) {
29         if (!inRange(toKey, inclusive))
30             throw new IllegalArgumentException("toKey out of range");
31         // 因为DescendingSubMap表示的是逆序的map，所以其实是通过获取原序的尾部
32         return new DescendingSubMap(m,
33                                         false, toKey, inclusive,
34                                         toEnd, hi,    hiInclusive);
35     }
36     // 与headMap对应，tailMap其实获取的是原序中的头部
37     public NavigableMap<K,V> tailMap(K fromKey, boolean inclusive){
38         if (!inRange(fromKey, inclusive))
39             throw new IllegalArgumentException("fromKey out of range");
40         return new DescendingSubMap(m,
41                                         fromStart, lo, loInclusive,
42                                         false, fromKey, inclusive);
43     }
44     // 逆序的逆序其实是正序
45     public NavigableMap<K,V> descendingMap() {
46         NavigableMap<K,V> mv = descendingMapView;
47         return (mv != null) ? mv :
48                 (descendingMapView =
49                  new AscendingSubMap(m,
50                                      fromStart, lo, loInclusive,
51                                      toEnd,     hi, hiInclusive));
52     }
53     // 剩余内容和AscendingSubMap很类似，就不说了
54     Iterator<K> keyIterator() {
55         return new DescendingSubMapKeyIterator(absHighest(), absLowFence());
56     }
57     Iterator<K> descendingKeyIterator() {
58         return new SubMapKeyIterator(absLowest(), absHighFence());
59     }
60     final class DescendingEntrySetView extends EntrySetView {
61         public Iterator<Map.Entry<K,V>> iterator() {
62             return new DescendingSubMapEntryIterator(absHighest(), absLowFence());
63         }
64     }
65     public Set<Map.Entry<K,V>> entrySet() {
66         EntrySetView es = entrySetView;
67         return (es != null) ? es : new DescendingEntrySetView();
68     }
69     TreeMap.Entry<K,V> subLowest()       { return absHighest(); }
70     TreeMap.Entry<K,V> subHighest()      { return absLowest(); }
71     TreeMap.Entry<K,V> subCeiling(K key) { return absFloor(key); }
72     TreeMap.Entry<K,V> subHigher(K key)  { return absLower(key); }
73     TreeMap.Entry<K,V> subFloor(K key)   { return absCeiling(key); }
74     TreeMap.Entry<K,V> subLower(K key)   { return absHigher(key); }
75 }

最后一个内部类是SubMap，它比较特别。这个类存在仅仅为了序列化兼容之前的版本不支持NavigableMap TreeMap。它被翻译成一个旧版本AscendingSubMap子映射到一个新版本。这个类是从来没有以其他方式使用。

 1 // SubMap 继承自AbstractMap；这个类存在仅仅为了序列化兼容之前的版本不支持NavigableMap TreeMap。它被翻译成一个旧版本AscendingSubMap子映射到一个新版本。这个类是从来没有以其他方式使用。
 2 private class SubMap extends AbstractMap<K,V>
 3     implements SortedMap<K,V>, java.io.Serializable {
 4     private static final long serialVersionUID = -6520786458950516097L;
 5     // 标识是否从map的开始到结尾都属于子map
 6     private boolean fromStart = false, toEnd = false;
 7     // 开始位置和结束位置的key
 8     private K fromKey, toKey;
 9     private Object readResolve() {
10         return new AscendingSubMap(TreeMap.this,
11                                        fromStart, fromKey, true,
12                                        toEnd, toKey, false);
13     }
14     // 结合类定义和类的说明就明白为什么提供了这么多方法但是都不能用了
15     public Set<Map.Entry<K,V>> entrySet() { throw new InternalError(); }
16     public K lastKey() { throw new InternalError(); }
17     public K firstKey() { throw new InternalError(); }
18     public SortedMap<K,V> subMap(K fromKey, K toKey) { throw new InternalError(); }
19     public SortedMap<K,V> headMap(K toKey) { throw new InternalError(); }
20     public SortedMap<K,V> tailMap(K fromKey) { throw new InternalError(); }
21     public Comparator<? super K> comparator() { throw new InternalError(); }
22 }

结合上面的内部类分析和《TreeMap源码分析——基础分析》，对TreeMap的实现应该有个大致轮廓。不过TreeMap的代码很长很复杂，不自己看一遍分析一边，很难想明白，很难理解进去。

自己也理解的不是很好，如果有牛人有对TreeMap的看法，望多指点。

posted on 2013-08-19 19:33 Java码界探秘阅读(117) 评论(0) 编辑收藏举报

刷新页面返回顶部

Java码界探秘

TreeMap源码分析——深入分析（基于JDK1.6）

公告

导航