关于TreeMap

什么是 TreeMap

TreeMap 是 Java 集合框架中的一个类，它实现了 NavigableMap 接口，并且基于红黑树的实现。它存储的键值对是有序的，默认情况下按键的自然顺序（如数字的从小到大，字符串的字母顺序）排序。

TreeMap 的特点

1. 排序：

   • TreeMap 是按键的自然顺序排序的。如果你希望按其他方式排序，可以提供一个 Comparator 来定制排序顺序。

2. 时间复杂度：

   • 插入、删除和查找操作的时间复杂度为 O(log n)，因为底层使用红黑树结构。

3. 键的唯一性：

   • TreeMap 中的键是唯一的，即一个键只能对应一个值。

4. 线程安全性：

   • 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 可以大大提高处理有序数据的效率。