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
来指定排序规则。