TreeMap基于红黑树实现键的有序存储，支持自然排序或自定义Comparator排序，适用于需快速查找且键有序的场景。默认按键的自然顺序排列，如Integer升序、String字典序；可通过构造函数传入比较器改变排序规则，如按字符串长度或倒序排列；使用自定义对象作键时须实现Comparable接口或提供Comparator，否则抛ClassCastException；TreeMap不允许null键（除非比较器处理null），插入、删除、查找时间复杂度均为O(log n)，遍历时元素已排序，适合中等规模数据；若无需排序应选用HashMap以提升性能；关键注意比较器逻辑的一致性与偏序关系正确性，避免排序混乱。
在Java中，TreeMap 是一个基于红黑树（Red-Black Tree）实现的有序映射集合，它会自动根据键（key）的自然顺序或自定义比较器对元素进行排序。这使得 TreeMap 成为需要键有序存储和快速查找场景下的理想选择。
TreeMap 的默认自然排序
如果键是实现了 Comparable 接口的类型（如 String、Integer 等），TreeMap 会按照键的自然顺序自动排序。
- 例如，使用 Integer 作为键时，数字从小到大排列
- 使用 String 作为键时，按字典序排序
示例代码：
TreeMap<Integer, String> map = new TreeMap<>(); map.put(3, "Three"); map.put(1, "One"); map.put(4, "Four"); map.put(2, "Two"); // 输出顺序为：1=One, 2=Two, 3=Three, 4=Four map.forEach((k, v) -> System.out.println(k + "=" + v));
使用自定义比较器排序
若想改变默认排序规则，可以在创建 TreeMap 时传入一个 Comparator 实现。
立即学习“Java免费学习笔记（深入）”；
- 比如让字符串按键的长度排序，而不是字典序
- 或者实现倒序排列
示例：按键字符串长度排序
TreeMap<String, Integer> map = new TreeMap<>(
(s1, s2) -> Integer.compare(s1.length(), s2.length())
);
map.put("apple", 5);
map.put("hi", 2);
map.put("dog", 3);
map.put("a", 1);
// 输出顺序按长度：a=1, hi=2, dog=3, apple=5
map.forEach((k, v) -> System.out.println(k + "=" + v));
处理自定义对象作为键
当使用自定义类作为 TreeMap 的键时，必须确保该类实现 Comparable 接口，或提供外部 Comparator，否则运行时会抛出 ClassCastException。
示例：Person 类按年龄排序
class Person {
String name;
int age;
// 构造方法、getter等省略
}
TreeMap<Person, String> map = new TreeMap<>(
(p1, p2) -> Integer.compare(p1.age, p2.age)
);
map.put(new Person("Alice", 30), "Engineer");
map.put(new Person("Bob", 25), "Designer");
map.put(new Person("Charlie", 35), "Manager");
// 按年龄升序输出
map.forEach((k, v) -> System.out.println(k.name + ": " + v));
注意事项与技巧
使用 TreeMap 进行键排序时，有几个关键点需要注意：
- TreeMap 不允许 null 键（除非使用定制比较器并显式处理 null）
- 插入、删除、查找时间复杂度为 O(log n)，适合中等规模数据
- 遍历时元素已按键排序，无需额外排序操作
- 若不需要排序功能，应优先考虑 HashMap 以获得更高性能
基本上就这些。TreeMap 的键排序能力非常实用，只要理解其依赖自然顺序或比较器的机制，就能灵活应用于各种有序映射场景。不复杂但容易忽略的是比较器的一致性——确保比较逻辑稳定且符合数学上的偏序关系，避免出现排序混乱或运行异常。
