TreeSet用于存储有序唯一元素,适合去重排序场景;TreeMap存储有序键值对,适用于按键查找与范围查询,二者均基于红黑树实现,增删查改时间复杂度为O(log n)。
TreeSet 和 TreeMap 是 Java 集合框架中基于红黑树(一种自平衡二叉查找树)实现的重要数据结构,它们都保证元素的有序性,但用途和使用方式有明显区别。理解它们的差异有助于在实际开发中做出合理选择。
TreeSet 是一个集合(Set),用于存储唯一的、有序的元素。它不允许重复值,内部通过 TreeMap 来实现,只使用 key 存储数据,value 是一个统一的占位对象。
TreeMap 是一个映射(Map),用来存储键值对(key-value),其中 key 有序且唯一,value 可重复。它根据 key 的自然顺序或自定义比较器排序。
简单来说:
由于结构不同,它们适用于不同的业务场景。
TreeSet 常用于去重并排序的场景,比如:
TreeSetset = new TreeSet<>(); set.add(5); set.add(1); set.add(3); System.out.println(set); // 输出 [1, 3, 5]
TreeMap 更适合需要按键排序查找的映射关系,比如:
TreeMapscores = new TreeMap<>(Collections.reverseOrder()); scores.put(95, "Alice"); scores.put(87, "Bob"); scores.put(92, "Charlie"); System.out.println(scores); // 按分数降序输出
两者都基于红黑树,因此增删改查的时间复杂度均为 O(log n),性能相近。
TreeSet 提供 Set 相关方法:
TreeMap 除了 Map 基本操作外,也支持类似范围操作:
这些特性让 TreeMap 在区间查询上非常高效,例如查找某段时间内的订单。
两者都可以利用元素的自然排序(实现 Comparable 接口),或传入 Comparator 比较器来自定义顺序。
TreeSet 定义排序方式:
TreeSetts = new TreeSet<>(Comparator.comparing(String::length)); ts.add("hi"); ts.add("hello"); ts.add("ok"); System.out.println(ts); // 按长度排序:[hi, ok, hello]
TreeMap 同样可以:
TreeMaptm = new TreeMap<>((a, b) -> b.length() - a.length()); tm.put("short", 1); tm.put("verylongword", 2); System.out.println(tm.keySet()); // 按长度降序
基本上就这些。TreeSet 适合需要有序无重复元素的场景,TreeMap 适合需要有序键值对映射的情况。选哪个,取决于你要存的是“值”还是“键值对”。结构相似,用途不同,合理使用能提升代码清晰度和执行效率。