Skip to content

TreeSet 是 Java 中 Set 接口的另一个实现类,它基于红黑树(Red-Black Tree)数据结构实现。与 HashSet 不同,TreeSet 保持元素的有序性。下面是一个简单的示例代码,展示如何使用 TreeSet:

import java.util.TreeSet;

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

        // 添加元素到 TreeSet
        set.add(5);
        set.add(2);
        set.add(8);

        // 打印 TreeSet 的内容
        System.out.println("TreeSet: " + set);

        // 检查元素是否存在于 TreeSet
        System.out.println("Contains 2: " + set.contains(2));

        // 删除元素从 TreeSet
        set.remove(5);

        // 打印更新后的 TreeSet 的内容
        System.out.println("Updated TreeSet: " + set);
    }
}

输出结果应该为:

TreeSet: [2, 5, 8]
Contains 2: true
Updated TreeSet: [2, 8]

TreeSet 会对元素进行排序,并且在添加、删除和查找元素方面具有较高的性能。这使得 TreeSet 在需要有序集合的场景中非常有用。

如果要存储自定义对象,则需要确保对象实现了Comparable接口或者提供Comparator来指定排序规则。