Java——TreeSet用法

Java——TreeSet

TreeSet 是 Java 中的一个有序集合类，它基于红黑树（Red-Black Tree）实现。

下面详细介绍 TreeSet 的用法和特点：

有序性：TreeSet 中的元素按照自然顺序或者通过自定义的比较器进行排序。它保证了元素的有序性，可以用于按照升序或降序处理数据。
去重性：TreeSet 中不允许存在重复的元素。当添加重复元素时，新元素将被忽略。
基于红黑树：TreeSet 内部使用红黑树数据结构来存储元素。红黑树是一种自平衡二叉查找树，可以保持较好的插入、删除和查找性能。
元素比较：当向 TreeSet 中添加元素时，会根据元素的自然顺序或自定义的比较器来确定元素的位置。对于自定义的对象类型，需要实现 Comparable 接口或提供自定义的 Comparator 来定义元素之间的比较规则。
常用方法：
- add(E element)：向集合中添加元素。
- remove(Object element)：从集合中移除指定元素。
- contains(Object element)：判断集合中是否包含指定元素。
- size()：返回集合中的元素个数。
- iterator()：返回一个迭代器，用于遍历集合中的元素。

下面是一个示例代码展示了如何使用 TreeSet：

import java.util.TreeSet;

public class TreeSetExample {
    public static void main(String[] args) {
        // 创建一个TreeSet对象
        TreeSet<Integer> treeSet = new TreeSet<>();

        // 添加元素
        treeSet.add(5);
        treeSet.add(2);
        treeSet.add(8);
        treeSet.add(3);
        treeSet.add(1);

        // 遍历输出元素（按照升序）
        for (Integer num : treeSet) {
            System.out.println(num);
        }

        // 输出集合大小
        System.out.println("Size: " + treeSet.size());

        // 判断集合中是否包含指定元素
        System.out.println("Contains 2: " + treeSet.contains(2));

        // 移除元素
        treeSet.remove(3);

        // 输出集合大小
        System.out.println("Size: " + treeSet.size());
    }
}