TreeMap

最新推荐文章于 2024-09-26 05:30:00 发布

最新推荐文章于 2024-09-26 05:30:00 发布 · 108 阅读

文章标签：

#TreeMap #Comparator

java容器专栏收录该内容

6 篇文章

订阅专栏

TreeMap定义

与LinkedHashMap不同，TreeMap不是继承于HashMap，而是继承于AbstractMap，归根到底，TreeMap实现的不是hash算法，而是二叉树算法；底层的数据存储也不是数组，而是链表实现的二叉树。二叉树数据结构决定了该map是一个按key进行排序排序的map。定义如下：

......

private final Comparator<? super K> comparator;

private transient Entry<K,V> root = null;

private transient int modCount = 0;

......

属性comparator

比较器，在二叉树中进行查找、插入指定的key节点时，就难免与left、parent和right节点的key进行比较，比较算法由comparator提供，若comparator不为空，则执行该comparator的比较算法：

int cmp = cpr.compare(key, t.key);

若comparator为空，则采用key自身的比较算法：

cmp = k.compareTo(t.key);

从代码可以看出，当没有提供comparator 时，默认就会将key强制转换为Comparable，然后调用其 compareTo（）方法，这就要求key必须实现 Comparable接口，否则将发生ClassCastException。

属性root

root是一个用链表实现的二叉树，left.key<parent.key<right.key,具体实现代码如下：

static final class Entry<K,V> implements Map.Entry<K,V> {
      K key;
        V value;
        Entry<K,V> left = null;
        Entry<K,V> right = null;
        Entry<K,V> parent;
        boolean color = BLACK;

        Entry(K key, V value, Entry<K,V> parent) {
            this.key = key;
            this.value = value;
            this.parent = parent;
        }

        public K getKey() {
            return key;
        }

        public V getValue() {
            return value;
        }

        public V setValue(V value) {
            V oldValue = this.value;
            this.value = value;
            return oldValue;
        }

        public boolean equals(Object o) {
            if (!(o instanceof Map.Entry))
                return false;
            Map.Entry<?,?> e = (Map.Entry<?,?>)o;

            return valEquals(key,e.getKey()) && valEquals(value,e.getValue());
        }

        public int hashCode() {
            int keyHash = (key==null ? 0 : key.hashCode());
            int valueHash = (value==null ? 0 : value.hashCode());
            return keyHash ^ valueHash;
        }

        public String toString() {
            return key + "=" + value;
        }
    }

从代码中的” boolean color = BLACK;“可以看出，不仅是二叉树，还是一棵红黑树。红黑树近似平衡，降低了对旋转的要求，提高了性能。

属性modCount

同HashMap一样，该属性用来记录map上的修改（put、remove、clear）次数，以便在对key进行迭代时判断是否发生了并发冲突。

遍历KeySet

KetSet的Iterator按照前序遍历的方式遍历二叉树中的Key，即从left到parent再到right，即第一个节点是最左边的节点，最后一个节点是最右边的一个节点，代码如下：

//获取第一个节点

final Entry<K,V> getFirstEntry() {
        Entry<K,V> p = root;
        if (p != null)
            while (p.left != null)
                p = p.left;
        return p;
    }

//获取最后一个节点

final Entry<K,V> getLastEntry() {
        Entry<K,V> p = root;
        if (p != null)
            while (p.right != null)
                p = p.right;
        return p;
    }