Java TreeMap 源码解析

这篇文章开始介绍Map系列另一个比较重要的类TreeMap。 大家也许能感觉到，网络上介绍HashMap的文章比较多，但是介绍TreeMap反而不那么多，这里面是有原因：一方面HashMap的使用场景比较多；二是相对于HashMap来说，TreeMap所用到的数据结构更为复杂。 废话不多说，进入正题。

签名（signature）

可以看到，相比HashMap来说，TreeMap多继承了一个接口NavigableMap，也就是这个接口，决定了TreeMap与HashMap的不同：

接口NavigableMap

首先看下NavigableMap的签名

发现NavigableMap继承了SortedMap，再看SortedMap的签名

SortedMap

SortedMap

就像其名字那样，说明这个Map是有序的。这个顺序一般是指由Comparable接口提供的keys的自然序（natural ordering），或者也可以在创建SortedMap实例时，指定一个Comparator来决定。 当我们在用集合视角（collection views，与HashMap一样，也是由entrySet、keySet与values方法提供）来迭代（iterate）一个SortedMap实例时会体现出key的顺序。 这里引申下关于Comparable与Comparator的区别（参考这里）：

Comparable一般表示类的自然序，比如定义一个Student类，学号为默认排序

Comparator一般表示类在某种场合下的特殊分类，需要定制化排序。比如现在想按照Student类的age来排序

插入SortedMap中的key的类类都必须继承Comparable类（或指定一个comparator），这样才能确定如何比较（通过

k1.compareTo(k2)

或

comparator.compare(k1, k2)

）两个key，否则，在插入时，会报

ClassCastException

的异常。 此为，SortedMap中key的顺序性应该与

equals

方法保持一致。也就是说

k1.compareTo(k2)

或

comparator.compare(k1, k2)

为true时，

k1.equals(k2)

也应该为true。 介绍完了SortedMap，再来回到我们的NavigableMap上面来。 NavigableMap是JDK1.6新增的，在SortedMap的基础上，增加了一些“导航方法”（navigation methods）来返回与搜索目标最近的元素。例如下面这些方法：

lowerEntry，返回所有比给定Map.Entry小的元素

floorEntry，返回所有比给定Map.Entry小或相等的元素

ceilingEntry，返回所有比给定Map.Entry大或相等的元素

higherEntry，返回所有比给定Map.Entry大的元素

设计理念（design concept）

红黑树（Red–black tree）

TreeMap是用红黑树作为基础实现的，红黑树是一种二叉搜索树，让我们在一起回忆下二叉搜索树的一些性质

二叉搜索树

先看看二叉搜索树（binary search tree，BST）长什么样呢？


二叉搜索树
相信大家对这个图都不陌生，关键点是：


二叉搜索树的优势在于每进行一次判断就是能将问题的规模减少一半，所以如果二叉搜索树是平衡的话，查找元素的时间复杂度为

log(n)

，也就是树的高度。 我这里想到一个比较严肃的问题，如果说二叉搜索树将问题规模减少了一半，那么三叉搜索树不就将问题规模减少了三分之二，这不是更好嘛，以此类推，我们还可以有四叉搜索树，五叉搜索树……对于更一般的情况：

K 叉搜索树

如果大家按照我上面分析，很可能也陷入一个误区，就是


我们不能把这两次给忽略了，对于更一般的情况：


对于极端情况k＝n时，K叉树就转化为了线性表了，复杂度也就是

O(n)

了，如果用数学角度来解这个问题，相当于：

k*log(n/k)

根据对数的运算规则可以转化为

ln(n)*k/ln(k)

，

ln(n)

为常数，所以相当于取

k/ln(k)

的极小值。这个问题对于大一刚学高数的人来说再简单不过了，我们这里直接看结果


自然数e的取值大约为2.718左右，可以看到二叉树基本上就是这样最优解了。在Nodejs的REPL中进行下面的操作



貌似k＝3时比k＝2时得到的结果还要小，那也就是说三叉搜索树应该比二叉搜索树更好些呀，但是为什么二叉树更流行呢？后来在万能的stackoverflow上找到了答案，主旨如下：


这样也就大概能理解为什么二叉树这么流行了，就是因为进行一次比较操作，我们最多可以将问题规模减少一半。 好了这里扯的有点远了，我们再回到红黑树上来。

红黑树性质

先看看红黑树的样子：


红黑树示例
上图是从wiki截来的，需要说明的一点是：


红黑树通过下面5条规则，保证了树是平衡的：

树的节点只有红与黑两种颜色

根节点为黑色的

叶子节点为黑色的

红色节点的字节点必定是黑色的

从任意一节点出发，到其后继的叶子节点的路径中，黑色节点的数目相同

满足了上面5个条件后，就能够保证：

根节点到叶子节点的最长路径不会大于根节点到叶子最短路径的2倍

。 其实这个很好理解，主要是用了性质4与5，这里简单说下：

红黑树操作

红黑树旋转示例（没有画出NIL节点）
关于红黑树的插入、删除、左旋、右旋这些操作，我觉得最好可以做到可视化，文字表达比较繁琐，我这里就不在献丑了，网上能找到的也比较多，像v_July_v的《教你透彻了解红黑树》。我这里推荐个swf教学视频（视频为英文，大家不要害怕，重点是看图??），7分钟左右，大家可以参考。 这里还有个交互式红黑树的可视化网页，大家可以上去自己操作操作，插入几个节点，删除几个节点玩玩，看看左旋右旋是怎么玩的。

源码剖析

由于红黑树的操作我这里不说了，所以这里基本上也就没什么源码可以讲了，因为这里面重要的算法都是

From CLR

，这里的CLR是指Cormen, Leiserson, Rivest，他们是算法导论的作者，也就是说TreeMap里面算法都是参照算法导论的伪代码。 因为红黑树是平衡的二叉搜索树，所以其put（包含update操作）、get、remove的时间复杂度都为

log(n)

。

总结

到目前为止，TreeMap与HashMap的的实现算是都介绍完了，可以看到它们实现的不同，决定了它们应用场景的不同：

TreeMap的key是有序的，增删改查操作的时间复杂度为

O(log(n))

，为了保证红黑树平衡，在必要时会进行旋转

HashMap的key是无序的，增删改查操作的时间复杂度为

O(1)

，为了做到动态扩容，在必要时会进行resize。

另外，我这里没有解释具体代码，难免有些标题党了，请大家见谅，后面理解的更深刻了再来填坑。