TreeMap是Java集合框架中基于红黑树实现的有序映射，它的核心特性就是会按照Key的顺序来存储键值对，默认情况下会使用Key的 natural order 也就是自然顺序进行排序，如果Key没有实现Comparable接口，或者我们需要按照非默认规则排序，就可以通过自定义Comparator比较器来实现按Key的自定义排序。
TreeMap默认按Key排序的实现
如果Key类型实现了Comparable接口，比如Integer、String等常见类型，TreeMap会直接使用它们的compareTo方法来进行排序。下面是一个使用默认排序的示例：
import java.util.TreeMap;
public class TreeMapDefaultSort {
public static void main(String[] args) {
// 创建默认排序的TreeMap，Key为String类型，默认按字典序排序
TreeMap<String, Integer> defaultSortMap = new TreeMap<>();
// 添加键值对
defaultSortMap.put("apple", 10);
defaultSortMap.put("banana", 20);
defaultSortMap.put("cherry", 15);
defaultSortMap.put("date", 8);
// 遍历输出，会按照Key的字典序排列
System.out.println("默认排序结果：");
for (String key : defaultSortMap.keySet()) {
System.out.println(key + " : " + defaultSortMap.get(key));
}
}
}
运行上述代码，输出结果会按照apple、banana、cherry、date的字典序排列，这就是TreeMap的默认排序效果。
自定义Comparator实现按Key排序
当我们需要按照非自然顺序排序时，就需要在创建TreeMap的时候传入自定义的Comparator比较器，Comparator是一个函数式接口，我们只需要实现它的compare方法即可，compare方法接收两个参数o1和o2，返回负整数、零或正整数，分别表示o1小于、等于或大于o2。
场景一：字符串Key按长度倒序排序
假设我们需要让TreeMap的Key按照字符串长度从长到短排序，如果长度相同再按字典序排序，就可以自定义如下Comparator：
import java.util.Comparator;
import java.util.TreeMap;
public class TreeMapCustomSort {
public static void main(String[] args) {
// 创建自定义Comparator，按字符串长度倒序，长度相同按字典序升序
Comparator<String> lengthComparator = new Comparator<String>() {
@Override
public int compare(String o1, String o2) {
// 先比较长度，长度长的排在前面
if (o1.length() != o2.length()) {
return o2.length() - o1.length();
}
// 长度相同则按字典序升序排列
return o1.compareTo(o2);
}
};
// 创建TreeMap时传入自定义Comparator
TreeMap<String, Integer> customSortMap = new TreeMap<>(lengthComparator);
// 添加键值对
customSortMap.put("apple", 10);
customSortMap.put("banana", 20);
customSortMap.put("cherry", 15);
customSortMap.put("date", 8);
customSortMap.put("fig", 5);
// 遍历输出
System.out.println("自定义排序结果（按长度倒序）：");
for (String key : customSortMap.keySet()) {
System.out.println(key + " : " + customSortMap.get(key));
}
}
}
运行上述代码，输出结果会先按照字符串长度从长到短排列，长度相同的按照字典序排列，结果依次是banana、cherry、apple、date、fig。
场景二：自定义对象作为Key按属性排序
如果我们的Key是自定义的对象，比如一个User对象，包含id和name两个属性，我们需要按照id升序排序，就需要自定义Comparator，同时要注意自定义对象作为TreeMap的Key时，要保证排序规则的一致性，避免后续出现查询异常。
首先定义User类：
public class User {
private Integer id;
private String name;
public User(Integer id, String name) {
this.id = id;
this.name = name;
}
// getter和setter方法
public Integer getId() {
return id;
}
public void setId(Integer id) {
this.id = id;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
@Override
public String toString() {
return "User{id=" + id + ", name='" + name + "'}";
}
}
然后使用自定义Comparator创建TreeMap：
import java.util.Comparator;
import java.util.TreeMap;
public class TreeMapCustomObjectKey {
public static void main(String[] args) {
// 自定义Comparator，按User的id升序排序
Comparator<User> userComparator = new Comparator<User>() {
@Override
public int compare(User o1, User o2) {
// 按id升序排列，如果id相同可以再按name排序，这里只按id排序
return o1.getId() - o2.getId();
}
};
// 创建TreeMap时传入自定义Comparator
TreeMap<User, String> userMap = new TreeMap<>(userComparator);
// 添加键值对
userMap.put(new User(3, "张三"), "开发部");
userMap.put(new User(1, "李四"), "测试部");
userMap.put(new User(2, "王五"), "产品部");
// 遍历输出
System.out.println("自定义对象Key排序结果（按id升序）：");
for (User key : userMap.keySet()) {
System.out.println(key + " : " + userMap.get(key));
}
}
}
运行上述代码，输出结果会按照User的id从小到大排列，依次是id为1、2、3的User对象对应的键值对。
自定义Comparator的注意事项
- Comparator的compare方法必须满足自反性、对称性和传递性，否则TreeMap的排序会出现异常，甚至导致数据查询错误。
- 当使用自定义对象作为Key时，如果同时重写了equals方法和compare方法，要保证两者的排序逻辑一致，避免出现compare返回0但equals返回false的情况，这种情况会导致TreeMap中出现多个相同的Key（逻辑上相同但对象不同）。
- 如果TreeMap已经创建完成，后续无法再修改它的Comparator，所以需要在创建实例的时候就确定好排序规则。
两种排序方式的对比
下面是TreeMap默认排序和自定义Comparator排序的对比：
| 排序方式 | 适用场景 | 实现方式 |
|---|---|---|
| 默认排序 | Key类型实现了Comparable接口，且需要按自然顺序排序 | 创建TreeMap时不传入Comparator，使用Key的compareTo方法 |
| 自定义Comparator排序 | 需要非自然顺序排序，或者Key没有实现Comparable接口 | 创建TreeMap时传入自定义的Comparator实例，实现compare方法 |
通过上述示例可以看出，TreeMap按Key排序的核心就是通过Comparator比较器来自定义排序规则，只要掌握了Comparator的compare方法的实现逻辑，就可以应对各种复杂的Key排序需求。
TreeMapComparator按Key排序自定义比较器修改时间：2026-06-22 05:36:46