Java中TreeSet排序和去重技巧

TreeSet依赖比较逻辑实现排序和去重，需确保compareTo与equals一致，优先使用不可变对象，避免修改关键字段，并通过Comparator处理null值。

在Java中，TreeSet 是一个基于红黑树（Red-Black Tree）实现的有序集合，它不仅自动排序元素，还能保证元素的唯一性。这使得 TreeSet 成为处理需要去重并按序存储数据场景的理想选择。但要正确使用它的排序和去重能力，需掌握一些关键技巧。

1. 自然排序与自定义排序

TreeSet 默认按照元素的自然顺序进行排序，前提是元素实现了 red">Comparable 接口。例如，Integer、String 等内置类型都支持自然排序：

若想对自定义对象排序，比如 Person 类按年龄或姓名排序，则需要实现 Comparable 接口，重写 compareTo() 方法：

<font size="-1">
class Person implements Comparable<Person> {
    String name;
    int age;

    public Person(String name, int age) {
        this.name = name;
        this.age = age;
    }

    @Override
    public int compareTo(Person p) {
        return Integer.compare(this.age, p.age); // 按年龄升序
    }
}
</font>

或者，在创建 TreeSet 时传入 Comparator 实现自定义排序逻辑：

立即学习“Java免费学习笔记（深入）”；

<font size="-1">
TreeSet<Person> set = new TreeSet<>((a, b) -> a.name.compareTo(b.name));
</font>

2. 去重机制依赖比较逻辑

TreeSet 判断重复的依据不是 equals()，而是通过 compareTo() 或 Comparator.compare() 是否返回 0。这意味着即使两个对象 equals 不相等，只要比较结果为 0，就会被视为重复。

常见陷阱：如果只重写 equals/hashCode 而忽略 compareTo，可能导致去重失效或出现“逻辑重复”却未被去除的情况。

建议：当使用 TreeSet 时，确保 compareTo() 的一致性——即若 a.compareTo(b) == 0，则 a.equals(b) 最好也为 true，避免歧义。

阿里妈妈·创意中心

阿里妈妈营销创意中心

下载

3. 处理可变对象的风险

若将可变对象存入 TreeSet 后修改其参与排序的字段，会导致集合内部结构混乱，可能破坏排序规则或使去重失效。

例如：

<font size="-1">
Person p = new Person("Alice", 25);
set.add(p);
p.age = 30; // 修改已添加对象的 age —— 危险！
</font>

此时该对象在树中的位置已不匹配新值，后续操作可能出现异常或错误结果。解决方案是使用不可变对象，或避免修改已加入 TreeSet 的关键字段。

4. null 值限制

TreeSet 不允许插入 null 元素（除非构造时指定特殊 Comparator 支持 null）。否则会抛出 NullPointerException。特别是在自定义 Comparator 中，要注意对 null 的安全处理：

<font size="-1">
TreeSet<String> set = new TreeSet<>(Comparator.nullsFirst(String::compareTo));
set.add(null); // 现在允许 null
</font>

这种写法可用于允许 null 并将其排在最前的场景。

基本上就这些。用好 TreeSet 的核心在于理解它依赖比较逻辑来实现排序和去重，而不是靠 equals 和 hashCode。只要合理设计 compareTo 或提供合适的 Comparator，并注意对象的可变性问题，就能高效利用 TreeSet 的特性。