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

P粉602998670 2025-11-08 00:00:00 次阅读

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 后，添加元素会自动按升序排列
重复元素不会被添加（Set 特性）

示例代码：


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 实现。

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


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); // 按年龄升序输出