treeset基于红黑树实现,依赖比较逻辑(comparable或comparator)排序和去重,不存重复元素且天然有序;比较结果为0即视为相等,决定插入与查找行为。

TreeSet 通过红黑树实现,它本身不存储重复元素,且天然有序——这个“序”不是插入顺序,而是由元素自身的自然顺序(Comparable)或外部定义的比较规则(Comparator)决定的。关键在于:TreeSet 在添加、查找、删除时,**全程依赖比较逻辑来定位节点位置**,而不是像 ArrayList 那样靠索引。
用 Comparable 定义自然顺序
当元素类型实现了 Comparable 接口(比如 String、Integer),TreeSet 就能直接调用其 compareTo() 方法做比较。你只需确保该方法满足自反性、对称性、传递性和一致性要求。
例如自定义类:
class Person implements Comparable<person> {
String name;
int age;
Person(String name, int age) {
this.name = name;
this.age = age;
}
public int compareTo(Person other) {
int nameCmp = this.name.compareTo(other.name);
return nameCmp != 0 ? nameCmp : Integer.compare(this.age, other.age);
}
}</person>
这样 new TreeSet
用 Comparator 覆盖或补充比较逻辑
如果类没实现 Comparable,或者你想用不同于自然顺序的规则(比如倒序、按某字段排序),就传入一个 Comparator 实例:
- 匿名内部类、Lambda 或方法引用均可,只要实现
compare(a, b) - Comparator 的逻辑必须与 equals 保持一致(即
compare(a,b)==0应尽量等价于a.equals(b)),否则可能破坏 Set 去重语义 - TreeSet 构造时传入 Comparator 后,所有操作都以此为准,不再看元素是否实现 Comparable
例如按年龄降序:
TreeSet<person> set = new TreeSet((p1, p2) -> Integer.compare(p2.age, p1.age));</person>
比较逻辑直接影响去重和排序结果
TreeSet 判断两个元素是否相等,不是靠 equals(),而是看 compare(a,b) == 0。这意味着:
- 如果 Comparator 认为两个对象“相等”(返回 0),TreeSet 就认为它们是同一个元素,后者不会被加入
- 即使两个对象
equals()返回 false,只要比较结果为 0,TreeSet 也会拒绝插入 - 反过来,若
compareTo()总返回非零值,但equals()返回 true,可能导致逻辑矛盾(比如 set.contains() 返回 false)
注意 null 和不可比较元素的处理
默认情况下,TreeSet 不允许 null 元素(add 会抛 NullPointerException)。如果要用 null,必须显式提供能处理 null 的 Comparator:
TreeSet<string> set = new TreeSet(Comparator.nullsFirst(String::compareTo));</string>
另外,若元素类型之间无法比较(比如混入 String 和 Integer),运行时会抛 ClassCastException —— TreeSet 在比较前不做类型检查,只在实际调用 compareTo() 或 compare() 时失败。
Java免费学习笔记:立即使用
解锁 Java 大师之旅:从入门到精通的终极指南











