TreeMap基于红黑树实现,按键的自然顺序或自定义比较器排序,支持高效范围查询和有序遍历,适用于需顺序访问键的场景。

在Java中,TreeMap 是一个基于红黑树(Red-Black Tree)实现的有序映射集合。它自动根据键(key)的自然顺序或自定义比较器对键值对进行排序,非常适合需要按顺序访问键的场景。
TreeMap的排序机制
TreeMap默认按照键的自然顺序排序,前提是键类型实现了 Comparable 接口,例如 String、Integer 等基本包装类型都支持自然排序。
如果使用自定义对象作为键,则需要提供一个 Comparator,否则运行时会抛出 ClassCastException。
示例:使用String作为键,自动按字母顺序排序
TreeMap
map.put("banana", 2);
map.put("apple", 1);
map.put("orange", 3);
// 遍历时输出顺序为:apple, banana, orange
自定义排序规则
通过传入 Comparator 可以改变排序逻辑。比如让字符串按键的长度排序,或者反向排序。
立即学习“Java免费学习笔记(深入)”;
示例:按键长度升序排列
TreeMap
map.put("hi", 1);
map.put("hello", 2);
map.put("ok", 3);
// 输出顺序:hi, ok, hello(按长度)
高效检索与范围查询
TreeMap 提供了多个方法用于高效检索和范围操作,这是它相比 HashMap 的优势之一。
- firstKey() 和 lastKey():获取最小和最大键
- floorKey(key):小于或等于给定键的最大键
- ceilingKey(key):大于或等于给定键的最小键
- subMap(fromKey, toKey):获取指定范围内的子映射(左闭右开)
- headMap(toKey) 和 tailMap(fromKey):获取小于某键或大于等于某键的所有条目
TreeMap
scores.put(85, "Bob");
scores.put(90, "Alice");
scores.put(78, "Tom");
System.out.println(scores.floorKey(88)); // 输出:85
System.out.println(scores.ceilingKey(88)); // 输出:90
实际应用场景建议
TreeMap适用于以下情况:
- 需要按键有序遍历数据,如排行榜、时间线事件存储
- 频繁进行区间查询,如查找某个分数段的学生
- 需要快速定位“前一个”或“后一个”元素
注意:由于底层是红黑树,插入和查找时间复杂度为 O(log n),比 HashMap 稍慢,但提供了顺序保障。
基本上就这些。只要理解了TreeMap的排序特性和提供的检索方法,就能在实际开发中灵活运用。关键是选择合适的键类型或比较器,确保排序逻辑符合业务需求。










