TreeMap基于红黑树实现,按键排序,默认按自然顺序升序排列;可通过Comparator自定义排序规则,如降序或忽略大小写;使用自定义对象作键时需实现Comparable接口或提供Comparator;键不可为null,所有键必须可比较,操作时间复杂度为O(log n),遍历按排序顺序输出,适用于需有序映射的场景。

TreeMap 是 Java 中基于红黑树(Red-Black Tree)实现的 Map 接口,它最大的特点就是可以自动按键排序。默认情况下,TreeMap 会根据键的自然顺序进行升序排列;你也可以通过自定义 Comparator 来指定排序规则。
TreeMap 默认排序:按键自然顺序
当使用可比较类型(如 String、Integer 等实现了 Comparable 接口的类)作为键时,TreeMap 会自动按照键的自然顺序排序。
示例代码:
TreeMap
map.put("banana", 2);
map.put("apple", 1);
map.put("orange", 3);
// 输出结果将按字母顺序排序
map.forEach((k, v) -> System.out.println(k + " = " + v));
// 输出:
// apple = 1
// banana = 2
// orange = 3
这里 String 类型天然实现了 Comparable
自定义排序:传入 Comparator
如果你希望使用非自然顺序,比如降序、忽略大小写排序等,可以在创建 TreeMap 时传入一个 Comparator 对象。
立即学习“Java免费学习笔记(深入)”;
示例:按键降序排列
TreeMap
map.put("banana", 2);
map.put("apple", 1);
map.put("orange", 3);
map.forEach((k, v) -> System.out.println(k + " = " + v));
// 输出:
// orange = 3
// banana = 2
// apple = 1
TreeMap
map.put("Banana", 2);
map.put("apple", 1);
map.put("Apple", 4); // 与 "apple" 视为相同键
map.put("Cherry", 3);
System.out.println(map);
// 输出:{Apple=4, Banana=2, Cherry=3}
注意:如果两个键通过 Comparator 判定相等,则后插入的值会覆盖前一个。
使用自定义对象作为键
如果你想用自定义类作为 TreeMap 的键,必须确保该类实现了 Comparable 接口,或者在构造 TreeMap 时提供 Comparator。
方式一:实现 Comparable
class Person implements Comparable
String name;
int age;
Person(String name, int age) {
this.name = name;
this.age = age;
}
@Override
public int compareTo(Person other) {
return this.name.compareTo(other.name); // 按姓名排序
}
}
TreeMap
map.put(new Person("Alice", 25), "Engineer");
map.put(new Person("Bob", 30), "Manager");
TreeMap
// 按年龄排序
TreeMap 排序注意事项
使用 TreeMap 进行排序时需要注意以下几点:
- 键不能为 null(除非使用不比较 null 的特殊 Comparator),否则会抛出 NullPointerException
- 所有键必须是可比较的,且比较逻辑要一致,避免运行时 ClassCastException
- 插入、删除、查找的时间复杂度为 O(log n),适合对性能和有序性都有要求的场景
- 遍历时总是按排序后的顺序输出,无需额外排序操作
基本上就这些。TreeMap 的排序能力让它在需要有序映射的场合非常实用,比如排行榜、时间线、字典序处理等。关键是理解它的排序依赖于 Comparable 或 Comparator,只要这个逻辑清晰,使用起来就很自然。









