在Java中如何使用TreeSet实现有序集合_TreeSet使用技巧解析-java教程-PHP中文网

在Java中如何使用TreeSet实现有序集合_TreeSet使用技巧解析

P粉602998670

发布： 2025-11-09 20:02:02

原创

106人浏览过

TreeSet基于红黑树实现有序唯一集合，支持自然排序与自定义Comparator排序，提供first、last、lower、higher及subSet等高效查询方法，适用于需排序场景，但不支持null值，增删性能为O(log n)，使用时应权衡需求选择合适集合类型。

在java中如何使用treeset实现有序集合_treeset使用技巧解析

在Java中，TreeSet 是一个基于红黑树（Red-Black Tree）实现的有序集合类，属于 java.util 包。它不仅保证元素唯一性（继承自 Set 接口），还能自动对元素进行排序。这种特性使得 TreeSet 在需要有序且不重复数据结构的场景下非常实用。

TreeSet 的基本使用方法

创建一个 TreeSet 实例非常简单，可以直接使用无参构造函数：

TreeSet set = new TreeSet();
set.add(5);
set.add(1);
set.add(3);
System.out.println(set); // 输出：[1, 3, 5]

从输出可以看出，元素按升序自动排列。TreeSet 支持任何实现了 Comparable 接口的对象，如 Integer、String、Double 等。

自定义排序规则：使用 Comparator

如果想改变默认排序方式，比如降序排列，可以通过传入 Comparator 来实现：

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

TreeSet descendingSet = new TreeSet((a, b) -> b.compareTo(a));
descendingSet.add(5);
descendingSet.add(1);
descendingSet.add(3);
System.out.println(descendingSet); // 输出：[5, 3, 1]

这个技巧适用于所有可比较类型。对于自定义对象，例如 Person 类，可以按姓名或年龄排序：

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

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

注意：若未提供 Comparator，自定义类必须实现 Comparable 接口，否则添加元素时会抛出 ClassCastException。