【JAVA】TreeSet 详解

弗瑞德学JAVA

于 2024-08-13 22:44:02 发布

阅读量348

点赞数 17

分类专栏： JAVA复习文章标签： java 开发语言

本文链接：https://blog.csdn.net/weixin_37559642/article/details/141175839

版权

JAVA复习专栏收录该内容

22 篇文章 0 订阅

订阅专栏

TreeSet

TreeSet 是 Java 集合框架中的一个类，它基于红黑树（Red-Black Tree）实现，并且提供了排序和集合操作的功能。与 HashSet 相比，TreeSet 是有序的，能够自动对元素进行排序。

1. 基本特点:

有序性: TreeSet 保证集合中的元素按自然顺序（或者根据提供的比较器）排序。默认情况下，TreeSet 中的元素按升序排列。
不重复性: 和 HashSet 一样，TreeSet 也不允许存储重复的元素。它通过 compareTo() 或 compare() 方法来判断元素是否相同。
基于红黑树实现: TreeSet 内部使用红黑树数据结构来存储元素。红黑树是一种自平衡的二叉搜索树，能够确保基本操作的时间复杂度为 O(log n)。

2. 实现原理:

TreeSet 基于 TreeMap 实现，TreeMap 本质上是一个红黑树。TreeSet 的每个元素都作为 TreeMap 的键存储，而值则是一个固定对象（通常是 PRESENT）。
当元素被插入到 TreeSet 中时，红黑树会根据元素的顺序属性来自动调整树的结构，以保持树的平衡，并确保元素的顺序。

3. 常用方法:

add(E e): 向 TreeSet 中添加元素。如果元素已经存在，返回 false；否则返回 true。
remove(Object o): 从 TreeSet 中移除指定的元素。如果元素存在，返回 true；否则返回 false。
contains(Object o): 检查 TreeSet 中是否包含指定的元素。如果存在，返回 true；否则返回 false。
size(): 返回 TreeSet 中的元素数量。
clear(): 清空 TreeSet，移除所有元素。
isEmpty(): 检查 TreeSet 是否为空。如果没有元素，返回 true；否则返回 false。
first(): 返回 TreeSet 中的第一个（最小的）元素。
last(): 返回 TreeSet 中的最后一个（最大的）元素。
subSet(E fromElement, E toElement): 返回 TreeSet 中从 fromElement（包括）到 toElement（不包括）的子集。
headSet(E toElement): 返回 TreeSet 中所有小于 toElement 的元素。
tailSet(E fromElement): 返回 TreeSet 中所有大于等于 fromElement 的元素。

4. 排序机制:

自然顺序: TreeSet 中的元素默认按自然顺序排序。对于基本数据类型（如整数、浮点数等），自然顺序是从小到大。对于字符串，自然顺序是按字母顺序。
自定义顺序: 你可以通过实现 Comparator 接口并将其传递给 TreeSet 构造函数来定义自定义排序规则。例如，可以按逆序、长度等来排序字符串。

TreeSet<String> set = new TreeSet<>(Comparator.reverseOrder());
set.add("Apple");
set.add("Banana");
set.add("Orange");

5. 性能:

TreeSet 的大部分操作（如插入、删除、查找）的时间复杂度为 O(log n)，因为它们基于红黑树这一自平衡二叉搜索树。
相较于 HashSet，TreeSet 的性能稍逊，因为红黑树的平衡操作会增加开销，但它提供了有序性，这是 HashSet 所不具备的。

6. 使用示例:

import java.util.TreeSet;

public class TreeSetExample {
    public static void main(String[] args) {
        TreeSet<Integer> set = new TreeSet<>();

        // 添加元素
        set.add(10);
        set.add(20);
        set.add(30);
        set.add(40);

        // 尝试添加重复元素
        boolean isAdded = set.add(20); // 返回 false

        // 获取第一个和最后一个元素
        int first = set.first(); // 返回 10
        int last = set.last(); // 返回 40

        // 获取小于 30 的子集
        TreeSet<Integer> headSet = (TreeSet<Integer>) set.headSet(30);

        // 遍历 TreeSet
        for (Integer num : set) {
            System.out.println(num);
        }
    }
}

7. 注意事项:

TreeSet 中的元素必须实现 Comparable 接口，或者在创建 TreeSet 时提供自定义的 Comparator，否则会抛出 ClassCastException。
TreeSet 是线程不安全的。如果多个线程同时访问同一个 TreeSet 实例，并且至少有一个线程修改了该集合，则必须在外部进行同步。
TreeSet 不允许存储 null 元素，因为无法对 null 进行比较，存储 null 会导致 NullPointerException。