什么是 TreeMap
TreeMap
是 Java 集合框架中的一个类,它实现了 NavigableMap
接口,并且基于红黑树的实现。它存储的键值对是有序的,默认情况下按键的自然顺序(如数字的从小到大,字符串的字母顺序)排序。
TreeMap
的特点
-
排序:
TreeMap
是按键的自然顺序排序的。如果你希望按其他方式排序,可以提供一个Comparator
来定制排序顺序。
-
时间复杂度:
- 插入、删除和查找操作的时间复杂度为
O(log n)
,因为底层使用红黑树结构。
- 插入、删除和查找操作的时间复杂度为
-
键的唯一性:
TreeMap
中的键是唯一的,即一个键只能对应一个值。
-
线程安全性:
TreeMap
不是线程安全的。如果需要在多线程环境下使用,建议使用Collections.synchronizedSortedMap
方法来包装TreeMap
。
TreeMap
的主要方法
以下是 TreeMap
中一些常用的方法:
-
构造方法:
TreeMap<K, V> treeMap = new TreeMap<>(); TreeMap<K, V> treeMap = new TreeMap<>(Comparator<? super K> comparator);
-
插入元素:
V put(K key, V value);
-
获取元素:
V get(Object key);
-
删除元素:
V remove(Object key);
-
检查键是否存在:
boolean containsKey(Object key);
-
检查值是否存在:
boolean containsValue(Object value);
-
获取第一个和最后一个键:
K firstKey(); K lastKey();
-
获取子集视图:
SortedMap<K, V> subMap(K fromKey, K toKey); SortedMap<K, V> headMap(K toKey); SortedMap<K, V> tailMap(K fromKey);
-
获取键值对的集合视图:
Set<Map.Entry<K, V>> entrySet();
示例代码
下面是一些使用 TreeMap
的简单示例代码:
import java.util.*;
public class TreeMapExample {
public static void main(String[] args) {
// 创建 TreeMap 实例
TreeMap<String, Integer> treeMap = new TreeMap<>();
// 插入元素
treeMap.put("apple", 10);
treeMap.put("banana", 20);
treeMap.put("cherry", 15);
// 获取元素
int value = treeMap.get("banana");
System.out.println("Value for banana: " + value);
// 删除元素
treeMap.remove("cherry");
// 遍历元素
for (Map.Entry<String, Integer> entry : treeMap.entrySet()) {
System.out.println(entry.getKey() + ": " + entry.getValue());
}
// 获取第一个和最后一个键
System.out.println("First key: " + treeMap.firstKey());
System.out.println("Last key: " + treeMap.lastKey());
// 获取子集视图
SortedMap<String, Integer> subMap = treeMap.subMap("apple", "banana");
for (Map.Entry<String, Integer> entry : subMap.entrySet()) {
System.out.println("SubMap - " + entry.getKey() + ": " + entry.getValue());
}
}
}
TreeMap
的高级用法
-
自定义排序:
TreeMap<String, Integer> treeMap = new TreeMap<>(Comparator.reverseOrder()); treeMap.put("apple", 10); treeMap.put("banana", 20); treeMap.put("cherry", 15);
-
获取键的范围视图:
NavigableMap<String, Integer> descendingMap = treeMap.descendingMap();
-
获取近似匹配:
Map.Entry<String, Integer> floorEntry = treeMap.floorEntry("banana"); Map.Entry<String, Integer> ceilingEntry = treeMap.ceilingEntry("banana");
总结
TreeMap
是一个强大的集合类,提供了按键排序存储数据的能力。它适用于需要按自然顺序或自定义顺序访问键值对的场景。理解和熟练使用 TreeMap
可以大大提高处理有序数据的效率。