在Java中如何使用TreeSet实现排序

TreeSet基于红黑树实现，自动排序且去重，默认按自然顺序升序排列，如Integer、String等实现Comparable的类型可直接使用；通过传入Comparator可自定义排序规则，如降序或按对象字段排序（如Person按年龄）；不允许插入null值，否则抛出NullPointerException；元素必须可比较，否则运行时报错；插入和查找时间复杂度为O(log n)，非线程安全。示例中TreeSet添加5、2、8、1后输出[1, 2, 5, 8]，而使用逆序Comparator则输出[8, 5, 2, 1]；对Person对象按年龄排序后输出Bob(25), Alice(30), Charlie(35)。

在Java中，TreeSet 是一个基于红黑树（Red-Black Tree）的 SortedSet 实现，它能够自动对其中的元素进行排序。默认情况下，TreeSet 会按照元素的自然顺序（natural ordering）进行升序排列。如果需要自定义排序规则，可以通过传入 Comparator 来实现。

使用自然排序

当元素类型实现了 Comparable 接口时，TreeSet 可以直接使用其自然排序规则。例如：Integer、String 等类都实现了 Comparable。

示例代码：

TreeSet numbers = new TreeSet<>();
numbers.add(5);
numbers.add(2);
numbers.add(8);
numbers.add(1);

System.out.println(numbers); // 输出: [1, 2, 5, 8]

使用自定义比较器（Comparator）

如果想改变排序方式，比如降序，或对自定义对象排序，可以传入一个 Comparator 实现。

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

情感家园企业站5.0 多语言多风格版

一套面向小企业用户的企业网站程序！功能简单，操作简单。实现了小企业网站的很多实用的功能，如文章新闻模块、图片展示、产品列表以及小型的下载功能，还同时增加了邮件订阅等相应模块。公告，友情链接等这些通用功能本程序也同样都集成了！同时本程序引入了模块功能，只要在系统默认模板上创建模块，可以在任何一个语言环境（或任意风格）的适当位置进行使用！

下载

例如，让整数按降序排列：

TreeSet descending = new TreeSet<>((a, b) -> b.compareTo(a));
descending.add(5);
descending.add(2);
descending.add(8);
descending.add(1);

System.out.println(descending); // 输出: [8, 5, 2, 1]

对自定义对象排序

假设有一个 Person 类，希望按年龄排序。

class Person {
    String name;
    int age;

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

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

使用 Comparator 按年龄排序：

TreeSet people = new TreeSet<>((p1, p2) -> Integer.compare(p1.age, p2.age));
people.add(new Person("Alice", 30));
people.add(new Person("Bob", 25));
people.add(new Person("Charlie", 35));

System.out.println(people); // 按年龄升序输出

注意事项