17370845950

TreeMap基于红黑树实现，默认按键自然升序排序，支持通过Comparator自定义排序规则，遍历有序，键不可为null，操作时间复杂度O(log n)，适用于需排序的映射场景。

在Java中，TreeMap 是一个基于红黑树（Red-Black tree）实现的有序映射集合。它会自动根据键（key）进行排序，默认是按照键的自然升序排列。如果需要自定义排序规则，也可以传入一个比较器（Comparator）。这使得 TreeMap 非常适合用于需要按键排序的场景。

默认自然排序

当使用实现了 Comparable 接口的类型作为键时（如 String、Integer 等），TreeMap 会自动按自然顺序排序。

TreeMap map = new TreeMap<>();
map.put("banana", 2);
map.put("apple", 1);
map.put("orange", 3);
// 输出顺序为：apple=1, banana=2, orange=3
map.forEach((k, v) -> System.out.println(k + "=" + v));

上面的例子中，字符串键会按字母顺序自动排序。

自定义排序规则

如果你希望使用非自然顺序（比如降序），可以传入一个自定义的 Comparator。

TreeMap map = new TreeMap<>(Collections.reverseOrder());
map.put("banana", 2);
map.put("apple", 1);
map.put("orange", 3);
// 输出顺序为：orange=3, banana=2, apple=1
map.forEach((k, v) -> System.out.println(k + "=" + v));

这里使用 Collections.reverseOrder() 实现了键的降序排列。

也可以针对复杂对象或特定逻辑编写自己的比较器：

TreeMap map = new TreeMap<>((a, b) -> b.length() - a.length());
map.put("hi", 1);
map.put("hello", 2);
map.put("hey", 3);
// 按字符串长度降序排列：hello=2, hey=3, hi=1
map.forEach((k, v) -> System.out.println(k + "=" + v));

注意事项

使用 TreeMap 时需要注意以下几点：

• 键不能为 null（除非使用自定义 Comparator 支持 null 值处理）
• 插入、删除、查找操作的时间复杂度为 O(log n)
• 遍历时元素是有序的，顺序由键决定
• 如果键是自定义对象，必须实现 Comparable 接口或提供 Comparator

基本上就这些。TreeMap 提供了简单而强大的排序能力，只要合理利用其构造函数中的比较器参数，就能灵活控制排序方式。实际开发中常用于统计后按关键字排序输出等场景。