TreeMap基于红黑树实现,按键自然顺序或自定义比较器排序,支持高效范围查询与有序遍历,适用于排行榜、区间统计等场景,使用时需注意非线程安全及不支持null键。

在Java中,TreeMap 是一个基于红黑树(Red-Black Tree)实现的有序映射集合,它不仅具备Map的基本功能,还能自动根据键(key)进行排序。如果你需要一个按关键字自然顺序或自定义顺序排列的键值对集合,TreeMap是一个非常合适的选择。
TreeMap的特点与适用场景
TreeMap实现了SortedMap和NavigableMap接口,因此除了支持基本的put、get、remove操作外,还提供丰富的有序访问方法。它的主要特点包括:
- 键值对按键的自然顺序(natural ordering)或自定义比较器(Comparator)排序
- 不支持null键(若使用自然排序时插入null会抛出NullPointerException)
- 非线程安全,高并发环境下需手动同步或使用ConcurrentSkipListMap替代
- 时间复杂度为O(log n)的插入、删除和查找操作
适用于需要按键有序遍历、范围查询或查找最接近键值的场景,例如排行榜、区间统计、字典序处理等。
使用自然排序构建有序映射
当键类型实现了Comparable接口(如String、Integer等),TreeMap会自动按照键的自然顺序排列。
立即学习“Java免费学习笔记(深入)”;
TreeMaptreeMap = new TreeMap<>(); treeMap.put("banana", 3); treeMap.put("apple", 1); treeMap.put("orange", 2); // 遍历时按键的字典序输出 for (String key : treeMap.keySet()) { System.out.println(key + ": " + treeMap.get(key)); } // 输出顺序:apple, banana, orange
这种写法简洁直观,适合大多数默认排序需求。
使用自定义比较器控制排序规则
如果需要逆序或其他特殊排序方式,可以通过传入Comparator来定制排序逻辑。
TreeMapreverseMap = new TreeMap<>((a, b) -> b.compareTo(a)); reverseMap.put("banana", 3); reverseMap.put("apple", 1); reverseMap.put("orange", 2); // 输出顺序将为:orange, banana, apple for (String key : reverseMap.keySet()) { System.out.println(key + ": " + reverseMap.get(key)); }
这个例子实现了字符串的降序排列。你也可以根据实际业务需求编写更复杂的比较逻辑,比如忽略大小写排序、按长度排序等。
利用TreeMap的导航方法进行高效查询
由于TreeMap实现了NavigableMap,提供了多种便捷的查找方法:
- floorKey(k):返回小于等于k的最大键
- ceilingKey(k):返回大于等于k的最小键
- higherKey(k):返回大于k的最小键
- lowerKey(k):返回小于k的最大键
- firstEntry()/lastEntry():获取最小和最大键值对
这些方法在处理范围匹配、最近匹配等问题时非常高效。
TreeMapscoreRank = new TreeMap<>(); scoreRank.put(85, "Alice"); scoreRank.put(90, "Bob"); scoreRank.put(95, "Charlie"); // 查找得分不低于92的第一个学生 Map.Entry entry = scoreRank.ceilingEntry(92); if (entry != null) { System.out.println("Score: " + entry.getKey() + ", Name: " + entry.getValue()); }
基本上就这些。TreeMap在需要有序键存储时非常实用,掌握其排序机制和导航方法能显著提升代码效率和可读性。注意避免在多线程环境中直接共享实例,并合理选择自然排序或自定义比较器即可。










