如何在Java中使用TreeMap进行排序

TreeMap基于红黑树实现，按键排序存储，支持自然排序和自定义Comparator排序，适用于需有序键值对的场景。

在Java中，TreeMap 是一个基于红黑树（Red-Black tree）实现的有序映射集合。它会自动根据键（key）进行排序，排序方式可以是默认的自然排序，也可以通过自定义比较器（Comparator）来指定排序规则。这使得 TreeMap 非常适合需要按键排序存储键值对的场景。

TreeMap 的默认排序（自然排序）

如果键的类型实现了 Comparable 接口（如 String、Integer 等），TreeMap 会按照键的自然顺序进行排序。

TreeMap map = new TreeMap<>();
map.put("banana", 2);
map.put("apple", 1);
map.put("orange", 3);
// 输出时会按字母顺序排列
map.forEach((k, v) -> System.out.println(k + ": " + v));

输出结果：

视野自助系统小型企业版2.0 Build 20050310

自定义设置的程度更高可以满足大部分中小型企业的建站需求，同时修正了上一版中发现的BUG，优化了核心的代码占用的服务器资源更少，执行速度比上一版更快 主要的特色功能如下： 1）特色的菜单设置功能，菜单设置分为顶部菜单和底部菜单，每一项都可以进行更名、选择是否隐 藏，排序等。 2）增加企业基本信息设置功能，输入的企业信息可以在网页底部的醒目位置看到。 3）增加了在线编辑功能，输入产品信息，企业介绍等栏

下载

立即学习“Java免费学习笔记（深入）”；

apple: 1
banana: 2
orange: 3

可以看到，字符串键已经按字典序自动排序。

使用自定义 Comparator 进行排序

如果你希望使用非默认的排序规则，比如按降序排列，或者使用自定义对象作为键，可以通过传入 Comparator 来实现。

TreeMap map = new TreeMap<>(Collections.reverseOrder());
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

使用自定义对象作为键进行排序

当使用自定义类作为键时，需要显式提供 Comparator，或让该类实现 Comparable 接口。

class Person {
    String name;
    int age;
Person(String name, int age) {
    this.name = name;
    this.age = age;
}

@Override
public String toString() {
    return name + "(" + age + ")";
}
}
// 创建 TreeMap 并按年龄升序排序
TreeMap 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 + " - " + v));
输出结果：
立即学习“Java免费学习笔记（深入）”；
Bob(25) - Designer
Alice(30) - Engineer
Charlie(35) - Manager
注意事项
使用 TreeMap 时需要注意以下几点：

键不能为 null（否则会抛出 NullPointerException），因为排序时无法比较 null 值。
插入的元素会被自动排序，查询和插入的时间复杂度为 O(log n)。
如果不需要排序功能，建议使用 HashMap 以获得更好的性能。
TreeMap 是非线程安全的，多线程环境下需手动同步或使用 Collections.synchronizedSortedMap。

基本上就这些。TreeMap 的排序能力让它在需要有序键值对的场景中非常实用。