掌握 Java TreeSet（Java TreeSet 使用详解与实战指南）

如何创建和使用 TreeSet？

下面是一个简单的示例，展示如何创建一个 TreeSet 并添加元素：

import java.util.TreeSet;public class TreeSetExample {    public static void main(String[] args) {        // 创建一个 TreeSet        TreeSet<Integer> numbers = new TreeSet<>();                // 添加元素        numbers.add(10);        numbers.add(5);        numbers.add(20);        numbers.add(15);                // 打印 TreeSet（自动排序）        System.out.println("TreeSet 内容: " + numbers);        // 输出: [5, 10, 15, 20]    }}

如你所见，即使我们以乱序添加数字，TreeSet 也会自动按升序排列它们。这就是 TreeSet 排序 的强大之处！

自定义排序：使用 Comparator

如果你想按降序或其他规则排序，可以传入一个 Comparator：

import java.util.Comparator;import java.util.TreeSet;public class TreeSetCustomSort {    public static void main(String[] args) {        // 创建一个按降序排列的 TreeSet        TreeSet<String> words = new TreeSet<>(Comparator.reverseOrder());                words.add("apple");        words.add("banana");        words.add("cherry");                System.out.println("降序 TreeSet: " + words);        // 输出: [cherry, banana, apple]    }}

常用方法一览

以下是一些 TreeSet 的常用方法：

• add(E e)：添加元素
• remove(Object o)：移除指定元素
• first() / last()：获取最小/最大元素
• headSet(E toElement)：返回小于指定元素的子集
• tailSet(E fromElement)：返回大于等于指定元素的子集
• subSet(E from, E to)：返回指定范围的子集

注意事项

- TreeSet 不允许 null 值（Java 7 及以后版本）
- 如果你存储的是自定义对象，必须实现 Comparable 接口或提供 Comparator，否则会抛出 ClassCastException
- 由于基于红黑树，TreeSet 在频繁排序场景下比 HashSet 更高效

总结

通过本篇 Java TreeSet 教程，你应该已经掌握了 TreeSet 的基本用法、排序机制以及常见操作。无论你是初学者还是有一定经验的开发者，理解 Java 集合框架 中的 TreeSet 都能帮助你写出更高效、更清晰的代码。

记住：当你需要一个自动排序、无重复的集合时，TreeSet 是你的最佳伙伴！