TreeMap基于红黑树实现,按键排序存储,默认按自然顺序排列,支持自定义Comparator排序规则,提供有序遍历与范围查询,时间复杂度为O(log n)。

在Java中,TreeMap 是一个基于红黑树(Red-Black Tree)实现的有序映射集合,它会自动根据键(key)进行排序存储。默认情况下,TreeMap 按照键的自然顺序(natural ordering)排序,也可以通过自定义 Comparator 来指定排序规则。
TreeMap 的基本特性
TreeMap 实现了 SortedMap 和 NavigableMap 接口,支持按键有序遍历、范围查询等操作。插入、删除、查找的时间复杂度为 O(log n),适合对有序性有要求的场景。
使用自然顺序排序
如果键的类型实现了 Comparable 接口(如 Integer、String),TreeMap 会自动按自然顺序排序:
TreeMapmap = new TreeMap<>(); map.put(3, "Three"); map.put(1, "One"); map.put(4, "Four"); map.put(2, "Two"); // 输出时按键升序排列 map.forEach((k, v) -> System.out.println(k + ": " + v)); // 输出: // 1: One // 2: Two // 3: Three // 4: Four
使用自定义比较器排序
可以通过构造函数传入 Comparator 来改变排序规则。例如,让 Integer 键按降序排列:
立即学习“Java免费学习笔记(深入)”;
TreeMapmap = new TreeMap<>((a, b) -> b.compareTo(a)); map.put(3, "Three"); map.put(1, "One"); map.put(4, "Four"); map.put(2, "Two"); map.forEach((k, v) -> System.out.println(k + ": " + v)); // 输出: // 4: Four // 3: Three // 2: Two // 1: One
也可以对字符串按长度排序:
TreeMapmap = new TreeMap<>( (s1, s2) -> Integer.compare(s1.length(), s2.length()) ); map.put("apple", 5); map.put("hi", 2); map.put("banana", 6); map.forEach((k, v) -> System.out.println(k + ": " + v)); // 输出(按字符串长度排序): // hi: 2 // apple: 5 // banana: 6
常用有序操作方法
TreeMap 提供了一些方便的导航方法:
- firstKey() / lastKey():获取最小和最大键
- floorKey(k):小于等于给定键的最大键
- ceilingKey(k):大于等于给定键的最小键
- subMap(k1, k2):获取键在范围 [k1, k2) 内的子映射
- descendingMap():返回逆序视图
System.out.println("最小键: " + map.firstKey());
System.out.println("最大键: " + map.lastKey());
System.out.println(">=2 的最小键: " + map.ceilingKey(2));
System.out.println("<=3 的最大键: " + map.floorKey(3));
基本上就这些。只要键类型可比较,TreeMap 就能帮你维持有序结构,不复杂但容易忽略细节,比如 null 值处理(TreeMap 不允许 null 键,除非使用自定义 comparator 显式支持)。合理利用,可以高效实现排序存储需求。










