TreeSet是基于红黑树的有序集合,元素按自然顺序或自定义比较器排序,不支持null值和重复元素,增删查时间复杂度为O(log n),适用于需要自动排序且去重的场景。

在Java中,TreeSet 是一个基于红黑树(Red-Black Tree)实现的有序集合,它能自动对插入的元素进行排序,保证集合中的元素始终处于有序状态。如果你需要一个不重复且有序的集合结构,TreeSet 是非常合适的选择。
TreeSet的基本特性
TreeSet 实现了 SortedSet 接口,内部使用 TreeMap 存储数据。它的主要特点包括:
- 元素按自然顺序(natural ordering)或自定义比较器(Comparator)排序
- 不允许插入 null 值(否则会抛出 NullPointerException)
- 不支持重复元素(Set 的通用特性)
- 添加、删除、查找操作的时间复杂度为 O(log n)
使用自然排序保持有序
当存储的元素是实现了 Comparable 接口的类时(如 Integer、String 等),TreeSet 会自动按照自然顺序排列。
TreeSetnumbers = new TreeSet<>(); numbers.add(5); numbers.add(1); numbers.add(3); numbers.add(9); System.out.println(numbers); // 输出:[1, 3, 5, 9]
上面代码中,Integer 类本身实现了 Comparable,因此插入后会自动升序排列。
立即学习“Java免费学习笔记(深入)”;
使用自定义比较器实现排序规则
如果希望使用非默认的排序方式,比如降序排列,可以通过传入 Comparator 来实现。
TreeSetdescendingSet = new TreeSet<>((a, b) -> b.compareTo(a)); descendingSet.add(5); descendingSet.add(1); descendingSet.add(8); System.out.println(descendingSet); // 输出:[8, 5, 1]
这里通过 Lambda 表达式反转了比较逻辑,实现了从大到小排序。你也可以用于自定义对象,比如按学生年龄或姓名排序。
处理自定义对象的排序
假设有一个 Student 类,我们希望按名字排序:
class Student {
String name;
int age;
public Student(String name, int age) {
this.name = name;
this.age = age;
}
@Override
public String toString() {
return name + "(" + age + ")";
}
}
创建 TreeSet 时指定比较器:
TreeSetstudents = new TreeSet<>((s1, s2) -> s1.name.compareTo(s2.name)); students.add(new Student("Alice", 20)); students.add(new Student("Bob", 22)); students.add(new Student("Charlie", 19)); System.out.println(students); // 输出:[Alice(20), Bob(22), Charlie(19)]
注意:如果不提供 Comparator,而 Student 没有实现 Comparable,运行时会抛出 ClassCastException。
常用方法与遍历
TreeSet 提供了一些方便的方法来访问有序数据:
- first():获取最小元素
- last():获取最大元素
- headSet(e):返回小于 e 的所有元素视图
- tailSet(e):返回大于等于 e 的所有元素视图
- subSet(e1, e2):返回区间 [e1, e2) 的元素
遍历时,元素按顺序输出:
for (Integer num : numbers) {
System.out.print(num + " "); // 1 3 5 9
}
基本上就这些。TreeSet 在需要自动排序且去重的场景下非常实用,只要注意元素可比较、避免插入 null,就能稳定工作。










