TreeMap基于红黑树实现,自动按键排序,默认按自然顺序排列,支持自定义Comparator;键不可为null,非线程安全,增删查时间复杂度O(log n),适用于需有序映射的场景。

在Java中,TreeMap 是一个基于红黑树(Red-Black tree)的 SortedMap 实现,它能自动对键(key)进行排序。如果你需要一个映射结构,并且希望其中的元素按照键的自然顺序或自定义顺序排列,TreeMap 是理想选择。
TreeMap 的基本特性
TreeMap 会根据键的值自动排序,默认使用键的自然排序(即实现 Comparable 接口),也可以通过构造函数传入自定义的 Comparator 来控制排序规则。
- 键不允许为 null(除非使用自定义 Comparator 处理 null 值)
- 线程不安全,如需并发环境应手动同步或使用其他结构
- 插入、删除、查找操作的时间复杂度为 O(log n)
使用自然排序
当键是 String、Integer 等实现了 Comparable 接口的类型时,TreeMap 会自动按升序排列。
TreeMapmap = new TreeMap<>(); map.put("banana", 3); map.put("apple", 1); map.put("orange", 2); // 输出时按键的字母顺序排列 map.forEach((k, v) -> System.out.println(k + ": " + v)); // 输出: // apple: 1 // banana: 3 // orange: 2
使用自定义比较器
如果想改变排序规则,比如按降序排列,可以在创建 TreeMap 时传入 Comparator。
立即学习“Java免费学习笔记(深入)”;
TreeMapmap = new TreeMap<>(Comparator.reverseOrder()); map.put("banana", 3); map.put("apple", 1); map.put("orange", 2); map.forEach((k, v) -> System.out.println(k + ": " + v)); // 输出(降序): // orange: 2 // banana: 3 // apple: 1
你也可以针对自定义对象实现比较逻辑:
class Person {
String name;
int age;
Person(String name, int age) {
this.name = name;
this.age = age;
}
}
TreeMap personMap = new TreeMap<>(
Comparator.comparing(p -> p.age)
);
personMap.put(new Person("Alice", 25), "Engineer");
personMap.put(new Person("Bob", 20), "Intern");
personMap.put(new Person("Charlie", 30), "Manager");
// 按年龄升序输出
personMap.keySet().forEach(p -> System.out.println(p.name + ", " + p.age));
// 输出:
// Bob, 20
// Alice, 25
// Charlie, 30
常用方法与遍历
TreeMap 提供了一些有用的有序访问方法:
- firstKey() / lastKey():获取最小和最大键
- headMap(toKey):返回小于某键的所有条目
- tailMap(fromKey):返回大于等于某键的所有条目
- subMap(fromKey, toKey):返回范围内的条目
- descendingMap():返回逆序视图
遍历时,元素默认按键升序输出:
for (Map.Entry基本上就这些。TreeMap 在需要有序映射时非常实用,掌握它的排序机制和常用方法,能有效提升数据处理效率。entry : map.entrySet()) { System.out.println(entry.getKey() + " = " + entry.getValue()); }










