You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
android-notes/blogs/集合/TreeMap 和 LinkedHashMap.md

3.6 KiB

TreeMap 和 LinkedHashMap

TreeMap

TreeMap 底层的数据结构就是红黑树,和 HashMap 的红黑树结构一样。

不同的是,TreeMap 利用了红黑树左节点小,右节点大的性质,根据 key 进行排序,使每个元素能够插入到红黑树大小适当的位置,维护了 key 的大小关系,适用于 key 需要排序的场景。

因为底层使用的是平衡红黑树的结构,所以 containsKey、get、put、remove 等方法的时间复杂度都是 log(n)。

TreeMap#put 方法源码:

    public V put(K key, V value) {
        Entry<K,V> t = root;
        // 根节点为空,直接新建
        if (t == null) {
            // compare 方法限制了 key 不能为 null
            compare(key, key); // type (and possibly null) check

            root = new Entry<>(key, value, null);
            size = 1;
            modCount++;
            return null;
        }
        int cmp;
        Entry<K,V> parent;
        // split comparator and comparable paths
        Comparator<? super K> cpr = comparator;
        if (cpr != null) {
            // 自旋找到 key 应该新增的位置
            do {
                parent = t;
                cmp = cpr.compare(key, t.key);
                if (cmp < 0)
                    t = t.left;
                else if (cmp > 0)
                    t = t.right;
                else
                    return t.setValue(value);
            } while (t != null);
        }
		//...
        return null;
    }

从源码中,可以得到:

  1. 新增节点时,利用了红黑树左小右大的特性,从根节点不断往下查找,直到找到节点为 null 为止,节点为 null 说明到达了叶子节点
  2. 查找过程中,发现 key 值已经存在,直接覆盖
  3. TreeMap 是禁止 key 为 null 的

LinkedHashMap

HashMap 是无序的,TreeMap 可以按照 key 进行排序,那有木有 Map 是可以维护插入顺序的呢?那就是 LinkedHashMap。

LinkedHashMap 本身是继承 HashMap 的,所以它拥有 HashMap 的所有特性,在此基础上还提供了两大特性:

  1. 按照插入顺序进行访问
  2. 实现了最近最少使用策略
public class LinkedHashMap<K,V>
    extends HashMap<K,V>
    implements Map<K,V>
{
    // 链表头
    transient LinkedHashMap.Entry<K,V> head;
    // 链表尾
    transient LinkedHashMap.Entry<K,V> tail;
    static class Entry<K,V> extends HashMap.Node<K,V>{
        Entry<K,V> before, after;
        Entry(int hash, K key, V value, Node<K,V> next){
            super(hash,ket,value,next);
        }
    }
    // 控制两种访问模式的字段,默认是 false
    // true 按照访问顺序,会把经常访问的 key 放到队尾
    // false 按照插入顺序提供访问
    final boolean accessOrder;
}

LinkedHashMap 只提供了单向访问,即按照插入的顺序从头到尾进行访问,不能像 LinkedList 那样可以双向访问。

看源码会发现,我们用 LinkedHashMap 的 get/put 方法,其实都是调用到 HashMap 的 get/put 方法,那么 LinkedHashMap 是如何实现自己独特的特性呢?

其实答案在 HashMap 的三个空方法:

    // Callbacks to allow LinkedHashMap post-actions
    void afterNodeAccess(Node<K,V> p) { }
    void afterNodeInsertion(boolean evict) { }
    void afterNodeRemoval(Node<K,V> p) { }

这三个方法是在 HashMap get/put/remove 时会调用的方法,LinkedHashMap 重写了这些方法,然后会在这些方法里面进行排序。

所以,LinkedHashMap 实现的重点有两个:

  1. 扩展 HashMap.Entry 使其拥有链表结构
  2. 重写 HashMap 里面三个方法