JavaTreeSet有序存储实现原理

Java中的TreeSet是一个基于红黑树实现的有序集合，能自动按自然顺序或自定义比较器对元素进行升序或降序排列，同时严格保证元素唯一性且不支持null值；其增删查操作均保持O(log n)高效性能，特别适用于需要实时去重与动态排序的业务场景——无论是处理基础类型如Integer、String，还是通过Lambda表达式或Comparator灵活定制复杂对象（如Student）的排序逻辑，TreeSet都能以简洁代码提供稳定可靠的有序集合能力。

在Java中，TreeSet 是一个基于红黑树（Red-Black Tree）实现的有序集合，它能自动对插入的元素进行排序，保证集合中的元素始终处于有序状态。如果你需要一个不重复且有序的集合结构，TreeSet 是非常合适的选择。

TreeSet的基本特性

TreeSet 实现了 SortedSet 接口，内部使用 TreeMap 存储数据。它的主要特点包括：

• 元素按自然顺序（natural ordering）或自定义比较器（Comparator）排序
• 不允许插入 null 值（否则会抛出 NullPointerException）
• 不支持重复元素（Set 的通用特性）
• 添加、删除、查找操作的时间复杂度为 O(log n)

使用自然排序保持有序

当存储的元素是实现了 Comparable 接口的类时（如 Integer、String 等），TreeSet 会自动按照自然顺序排列。

TreeSet<Integer> numbers = new TreeSet<>();
numbers.add(5);
numbers.add(1);
numbers.add(3);
numbers.add(9);

System.out.println(numbers); // 输出：[1, 3, 5, 9]

上面代码中，Integer 类本身实现了 Comparable，因此插入后会自动升序排列。

使用自定义比较器实现排序规则

如果希望使用非默认的排序方式，比如降序排列，可以通过传入 Comparator 来实现。

TreeSet<Integer> descendingSet = 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 时指定比较器：

TreeSet<Student> students = 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，就能稳定工作。

终于介绍完啦！小伙伴们，这篇关于《JavaTreeSet有序存储实现原理》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布文章相关知识，快来关注吧！