17370845950

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

TreeMap 是 Java 中基于红黑树（Red-Black Tree）实现的 Map 接口，它最大的特点就是可以自动按键排序。默认情况下，TreeMap 会根据键的自然顺序进行升序排列；你也可以通过自定义 Comparator 来指定排序规则。

TreeMap 默认排序：按键自然顺序

当使用可比较类型（如 String、Integer 等实现了 Comparable 接口的类）作为键时，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));
// 输出：
// apple = 1
// banana = 2
// orange = 3

这里 String 类型天然实现了 Comparable，所以 TreeMap 能够自动排序。

自定义排序：传入 Comparator

如果你希望使用非自然顺序，比如降序、忽略大小写排序等，可以在创建 TreeMap 时传入一个 Comparator 对象。

TreeMap map = new TreeMap(Comparator.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

TreeMap map = new TreeMap(String.CASE_INSENSITIVE_ORDER);
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。

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 = new TreeMap();
map.put(new Person("Alice", 25), "Engineer");
map.put(new Person("Bob", 30), "Manager");

TreeMap map = new TreeMap((p1, p2) -> Integer.compare(p1.age, p2.age));
// 按年龄排序

TreeMap 排序注意事项

使用 TreeMap 进行排序时需要注意以下几点：

• 键不能为 null（除非使用不比较 null 的特殊 Comparator），否则会抛出 NullPointerException
• 所有键必须是可比较的，且比较逻辑要一致，避免运行时 ClassCastException
• 插入、删除、查找的时间复杂度为 O(log n)，适合对性能和有序性都有要求的场景
• 遍历时总是按排序后的顺序输出，无需额外排序操作

基本上就这些。TreeMap 的排序能力让它在需要有序映射的场合非常实用，比如排行榜、时间线、字典序处理等。关键是理解它的排序依赖于 Comparable 或 Comparator，只要这个逻辑清晰，使用起来就很自然。