TreeMap 自定义排序的几种使用方法

1.使用键的自然排序

如果键实现了 Comparable 接口，那么 TreeMap 会使用键的自然排序。例如，如果你的键是 Integer 或 String 类型，那么不需要额外的配置。

// 使用 Integer 的自然排序
TreeMap<Integer, String> map = new TreeMap<>();
map.put(3, "three");
map.put(1, "one");
map.put(2, "two");

System.out.println(map);  // 输出 {1=one, 2=two, 3=three}

2. 通过实现 Comparable 排序

如果键是自定义类型，你可以让该类型实现 Comparable 接口，并重写 compareTo 方法来定义键之间的比较逻辑。

class Person implements Comparable<Person> {
    private String name;
    private int age;

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

    @Override
    public int compareTo(Person other) {
        return Integer.compare(this.age, other.age);  // 按年龄排序
    }

    @Override
    public String toString() {
        return name + ":" + age;
    }
}

// 使用 Person 的自然排序
TreeMap<Person, String> personMap = new TreeMap<>();
personMap.put(new Person("Alice", 25), "A");
personMap.put(new Person("Bob", 20), "B");
personMap.put(new Person("Charlie", 30), "C");

System.out.println(personMap);  // 输出 {Bob:20=A, Alice:25=B, Charlie:30=C}

3. 使用自定义 Comparator 排序

你还可以通过提供一个 Comparator 实例来定义键之间的比较逻辑。

// 使用自定义 Comparator 按年龄排序
TreeMap<Person, String> personMap2 = new TreeMap<>(new Comparator<Person>() {
    @Override
    public int compare(Person p1, Person p2) {
        return Integer.compare(p1.age, p2.age);
    }
});

personMap2.put(new Person("Alice", 25), "A");
personMap2.put(new Person("Bob", 20), "B");
personMap2.put(new Person("Charlie", 30), "C");

System.out.println(personMap2);  // 输出 {Bob:20=A, Alice:25=B, Charlie:30=C}

4. 使用 Lambda 表达式简化 Comparator

从 Java 8 开始，可以使用 Lambda 表达式来简化 Comparator 的定义。

// 使用 Lambda 表达式
TreeMap<Person, String> personMap3 = new TreeMap<>(Comparator.comparingInt(Person::getAge));

personMap3.put(new Person("Alice", 25), "A");
personMap3.put(new Person("Bob", 20), "B");
personMap3.put(new Person("Charlie", 30), "C");

System.out.println(personMap3);  // 输出 {Bob:20=A, Alice:25=B, Charlie:30=C}

5. 使用方法引用来进一步简化

如果你使用的 Comparator 只是比较对象的一个属性，可以使用方法引用来进一步简化代码。

// 使用方法引用
TreeMap<Person, String> personMap4 = new TreeMap<>(Comparator.comparing(Person::getAge));

personMap4.put(new Person("Alice", 25), "A");
personMap4.put(new Person("Bob", 20), "B");
personMap4.put(new Person("Charlie", 30), "C");

System.out.println(personMap4);  // 输出 {Bob:20=A, Alice:25=B, Charlie:30=C}