TreeSet是基于红黑树实现的有序集合,自动按自然顺序或自定义Comparator排序,增删查均为O(log n);不允null(除非Comparator显式处理)和重复元素,本质是TreeMap的key集合。

TreeSet 是 Java 中基于红黑树实现的有序集合,它能自动按元素自然顺序或自定义比较器排序,插入、删除、查找时间复杂度均为 O(log n)。关键在于:它不接受 null 元素(除非使用自定义比较器且明确处理 null),也不允许重复元素——因为底层是 TreeMap 的 key 集合,而 Map 的 key 天然唯一。
TreeSet 的排序本质是 TreeMap 的 Key Set
TreeSet 内部持有一个 TreeMap 实例,你调用 add(e) 时,实际执行的是 map.put(e, PRESENT),其中 PRESENT 是一个空的静态对象(new Object())。所以 TreeSet 的“有序”完全由 TreeMap 的 key 排序机制决定:
- 若元素实现了 Comparable 接口(如 String、Integer),TreeSet 默认用其 compareTo() 方法排序
- 若未实现 Comparable,或你想改变排序逻辑,必须在构造时传入 Comparator
- TreeSet 自身不重写 equals/hashCode,它的去重和比较完全依赖 compare/compareTo 结果是否为 0
自然排序:要求元素实现 Comparable
例如用 Integer 或 String 构建 TreeSet,无需额外配置就能自动升序:
Setset.add(5); set.add(1); set.add(3); // 插入后内部已按 1,3,5 排好
但如果放入自定义类(比如 Person),必须让 Person 实现 Comparable 并定义 compareTo 规则,否则运行时报 ClassCastException。
立即学习“Java免费学习笔记(深入)”;
自定义排序:通过 Comparator 灵活控制顺序
适用于无法修改类源码,或同一类型需多种排序方式的场景。例如按字符串长度降序:
Setset.add("hi"); set.add("hello"); set.add("a"); // 结果:hello, hi, a
注意:Comparator 中返回正数表示 s1 > s2(即 s1 排在 s2 后),负数则相反;返回 0 表示相等,此时后添加的元素不会被存入。
常见误区与注意事项
- TreeSet 不是线程安全的,多线程写入需手动同步(如 Collections.synchronizedSortedSet)
- 遍历 TreeSet 得到的 Iterator 是 fail-fast 的,遍历时修改集合会抛 ConcurrentModificationException
- contains()、remove() 等操作也依赖比较逻辑,若 Comparator 使用了可能为 null 的字段,务必判空,否则 NPE
- 不能靠索引访问元素(无 get(int index)),只能用迭代器或转成数组后取下标
基本上就这些。TreeSet 的有序性不是“事后排序”,而是“边插边排”,靠红黑树结构动态维持平衡与顺序,理解这点,用起来就清晰了。










