Java中TreeSet与TreeMap的区别与应用

TreeSet用于存储有序唯一元素，适合去重排序场景；TreeMap存储有序键值对，适用于按键查找与范围查询，二者均基于红黑树实现，增删查改时间复杂度为O(log n)。

TreeSet 和 TreeMap 是 Java 集合框架中基于红黑树（一种自平衡二叉查找树）实现的重要数据结构，它们都保证元素的有序性，但用途和使用方式有明显区别。理解它们的差异有助于在实际开发中做出合理选择。

1. 基本概念与存储内容不同

TreeSet 是一个集合（Set），用于存储唯一的、有序的元素。它不允许重复值，内部通过 TreeMap 来实现，只使用 key 存储数据，value 是一个统一的占位对象。

TreeMap 是一个映射（Map），用来存储键值对（key-value），其中 key 有序且唯一，value 可重复。它根据 key 的自然顺序或自定义比较器排序。

简单来说：

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

• TreeSet 存的是单个元素，关注“有哪些值”；
• TreeMap 存的是键值对，关注“某个键对应什么值”。

2. 主要应用场景不同

由于结构不同，它们适用于不同的业务场景。

TreeSet 常用于去重并排序的场景，比如：

• 统计一篇文章中出现的所有不重复单词，并按字母顺序输出；
• 维护一个已处理的任务 ID 列表，要求自动排序且不能重复。

TreeSet<Integer> set = new TreeSet<>();
set.add(5); set.add(1); set.add(3);
System.out.println(set); // 输出 [1, 3, 5]

TreeMap 更适合需要按键排序查找的映射关系，比如：

• 按用户 ID 排序的用户信息缓存；
• 时间戳作为 key 的日志记录，需按时间范围查询。

TreeMap<Integer, String> scores = new TreeMap<>(Collections.reverseOrder());
scores.put(95, "Alice");
scores.put(87, "Bob");
scores.put(92, "Charlie");
System.out.println(scores); // 按分数降序输出

3. 支持的操作与性能特点

两者都基于红黑树，因此增删改查的时间复杂度均为 O(log n)，性能相近。

AppMall应用商店

AI应用商店，提供即时交付、按需付费的人工智能应用服务

56

查看详情

TreeSet 提供 Set 相关方法：

• add(e), remove(e), contains(e)
• first(), last() —— 获取最小/最大元素
• higher(e), lower(e) —— 获取比 e 大/小的最小或最大元素
• headSet(e), tailSet(e), subSet(from, to) —— 范围查询

TreeMap 除了 Map 基本操作外，也支持类似范围操作：

• firstKey(), lastKey()
• ceilingKey(k), floorKey(k)
• subMap(fromKey, toKey) 等视图操作

这些特性让 TreeMap 在区间查询上非常高效，例如查找某段时间内的订单。

4. 排序机制一致但使用方式略有差异

两者都可以利用元素的自然排序（实现 Comparable 接口），或传入 Comparator 比较器来自定义顺序。

TreeSet 定义排序方式：

TreeSet<String> ts = new TreeSet<>(Comparator.comparing(String::length));
ts.add("hi"); ts.add("hello"); ts.add("ok");
System.out.println(ts); // 按长度排序：[hi, ok, hello]

TreeMap 同样可以：

TreeMap<String, Integer> tm = new TreeMap<>((a, b) -> b.length() - a.length());
tm.put("short", 1);
tm.put("verylongword", 2);
System.out.println(tm.keySet()); // 按长度降序

基本上就这些。TreeSet 适合需要有序无重复元素的场景，TreeMap 适合需要有序键值对映射的情况。选哪个，取决于你要存的是“值”还是“键值对”。结构相似，用途不同，合理使用能提升代码清晰度和执行效率。

以上就是Java中TreeSet与TreeMap的区别与应用的详细内容，更多请关注php中文网其它相关文章！