Java TreeSet 比较器返回 0 时为什么会丢掉元素

如果把对象放进 TreeSet 后发现输入有两个，集合却只剩一个，先检查比较器，而不是先改 equals()。TreeSet 的插入、查找和删除都沿着 compareTo() 或 Comparator.compare() 判断位置；当比较结果为 0 时，它就把两个对象当成同一个集合元素。

官方资料：https://docs.oracle.com/en/java/javase/26/docs/api/java.base/java/util/TreeSet.html

比较器契约：https://docs.oracle.com/en/java/javase/26/docs/api/java.base/java/util/Comparator.html

所以，标题中的“丢掉”通常不是 TreeSet 随机删除了对象，而是比较器定义了一个比业务更粗的“相等”规则。只按业务排序字段比较时，必须再补一个能区分对象的稳定字段。

先用最小案例复现第二次 add 为什么失败

假设订单列表里有两个不同的人，他们年龄相同。业务希望按年龄排序，但年龄只是排序主字段，不是对象身份。下面的比较器只比较年龄，第二个对象加入时就会得到 0。

import java.util.Comparator;
import java.util.TreeSet;

record Person(long id, String name, int age) {}

public class TreeSetCompareDemo {
    public static void main(String[] args) {
        // 这里只比较 age，id 和 name 不参与判断
        Comparator byAge = Comparator.comparingInt(Person::age);
        TreeSet people = new TreeSet(byAge);

        // 两个对象的 age 相同，但它们的 id、name 和 equals 都不同
        Person first = new Person(101, "林舟", 30);
        Person second = new Person(102, "周宁", 30);

        System.out.println(people.add(first));
        System.out.println(people.add(second));
        System.out.println(people.size());
    }
}

第二次 add 返回 false，集合大小仍是 1。从普通对象语义看，first.equals(second) 是 false；从这个 TreeSet 的排序语义看，byAge.compare(first, second) 是 0，因此它们落在同一个比较等价类中。

TreeSet 判断重复看比较结果，不直接调用 equals

Oracle 文档把这个边界说得很清楚：TreeSet 的元素使用自然顺序，或者使用创建集合时传入的 Comparator；集合执行元素比较时使用 compareTo 或 compare。因此，只要比较结果为零，TreeSet 就认为两个元素从集合角度等价。

这里有两个容易混淆的概念：

这不是说 equals() 没用，而是有序集合必须先找到树中的比较位置。若比较器与 equals 不一致，集合仍能运行，但它不再满足 Set 以 equals 定义成员关系的直觉契约。contains、remove 以及范围视图也会继续沿用这套比较规则。

真正的风险是比较器把业务身份截断了

只比较年龄并非技术错误：如果业务本来就只需要“每个年龄保留一个对象”，它甚至可能是正确设计。问题在于，代码表面声明的是 TreeSet，读者通常会理解为“保存不重复的 Person”；实际比较器却把“同年龄”定义成“同一个元素”。

迁移旧代码或替换 HashSet 时，重点检查这三个地方：

1. 比较器使用的字段是不是业务身份，而只是展示、排序或分组字段。
2. 主字段相同时，是否存在稳定且唯一的后备字段。
3. 对象字段变化后，比较结果是否仍然满足反对称、传递和可重复判断。

如果业务需要“同年龄的人仍然是不同元素”，就不能让年龄成为比较器的终点。

用 thenComparing 补上稳定的第二关键字

最直接的修复是保留年龄排序，再用不可变的唯一标识打破平局。Comparator.thenComparing 的含义是：前一个比较器认为两个对象相等时，继续使用后一个比较器。

import java.util.Comparator;
import java.util.TreeSet;

record Person(long id, String name, int age) {}

public class TreeSetTieBreakerDemo {
    public static void main(String[] args) {
        Comparator stableOrder = Comparator
                // 先满足页面展示的年龄排序需求
                .comparingInt(Person::age)
                // 年龄相同再比较不可变 id，避免不同对象折叠
                .thenComparingLong(Person::id);

        TreeSet people = new TreeSet(stableOrder);
        people.add(new Person(101, "林舟", 30));
        people.add(new Person(102, "周宁", 30));

        // 两个对象的最终比较结果不同，因此两个元素都会保留
        System.out.println(people.size());
    }
}

这次相同年龄只会触发第二关键字比较：101 与 102 不同，最终比较结果不为零，两个对象就能占据两个有序位置。唯一标识应当稳定；如果比较过程中 id 会变化，元素可能在树的逻辑位置与新值之间失配。

比较器改完后要回归哪些集合行为

不要只断言 size() 变成了预期值。比较器定义的是整个有序集合的坐标系，修复后应一起检查插入、查找、删除和遍历顺序。

static void checkSetContract(TreeSet people) {
    Person probe = new Person(102, "周宁", 30);

    // contains 使用同一个比较器定位，不要用另一套排序规则构造探针
    if (!people.contains(probe)) {
        throw new IllegalStateException("目标元素无法按比较器找到");
    }

    // remove 同样依赖比较器，确认删除的是预期等价类中的元素
    if (!people.remove(probe)) {
        throw new IllegalStateException("目标元素无法按比较器删除");
    }
}

还要留意 null、不同类型对象和可变字段。比较器如果不支持空值，应保持行为一致；如果字段来自可变对象，最好在放入 TreeSet 前冻结排序字段。否则即使没有出现“少一个元素”，也可能出现 contains 找不到现有对象、遍历顺序异常等更隐蔽的问题。

从旧比较器迁移时的速查清单

常见问题

比较器返回 0 就一定说明两个对象 equals 相等吗？

不一定。Comparator 允许建立与 equals 不一致的排序，但放进 TreeSet 时，返回 0 的对象会被当成同一集合元素。是否允许这样做，取决于业务想要的去重边界。

把 id 作为第二关键字后，equals 也必须改吗？

不必为了让 TreeSet 工作而机械修改 equals。应先明确对象身份和业务相等规则；如果两套规则确实代表同一概念，再同步设计 equals、hashCode 和比较器。

用 name 作为 tie-breaker 可以吗？

只有当 name 唯一、稳定且大小写与空白规则明确时才适合。展示名称通常会重复或被修改，优先使用不可变的业务 id。

为什么改成 HashSet 就不丢元素了？

HashSet 主要依据 equals 和 hashCode 判断成员关系，确实可能保留这些对象，但它不提供 TreeSet 的排序能力。应先确认业务需要的是哪种去重语义，不要用换集合掩盖比较器定义错误。

排查 TreeSet 元素变少时，第一步就是把 compare(a, b) 的结果与 a.equals(b) 并排打印或断言。若对象并不相等却返回了 0，就给排序主字段补上稳定的第二关键字，或者明确把“比较等价”作为业务去重规则。这个判断比单纯修改 equals 更接近问题根因。