码迷,mamicode.com
首页 > 其他好文 > 详细

TreeMap 的简单解释

时间:2019-12-22 20:29:20      阅读:113      评论:0      收藏:0      [点我收藏+]

标签:mui   实现   possibly   键值对   父节点   public   except   class   cal   

TreeMap的构造函数
技术图片技术图片
 
可以传入 自定义的比较器、Map、SortedMap。
 
put方法:
public V put(K key, V value) {
    Entry<K,V> t = root; //得到根节点
    if (t == 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) {
        do {
            parent = t;
            cmp = cpr.compare(key, t.key); //根据比较器 比较当前节点与插入节点的key
            if (cmp < 0) //如果当前节点较小 
                t = t.left;
            else if (cmp > 0) //如果当前节点较大
                t = t.right;
            else //如果相同
                return t.setValue(value);
        } while (t != null);
    }
    else { //如果没有传入一个比较器
        if (key == null) 
            throw new NullPointerException();
        @SuppressWarnings("unchecked")
            Comparable<? super K> k = (Comparable<? super K>) key; //尝试将key转为Comparable<? super K>类型,就是说 如果没有传入比较器
                                                    //,key所在的类需要实现Comparable接口
        do {
            parent = t;
            cmp = k.compareTo(t.key); //尝试用key自己实现的comparteTo方法比较父节点的key
            if (cmp < 0)
                t = t.left;
            else if (cmp > 0)
                t = t.right;
            else
                return t.setValue(value);
        } while (t != null);
    }
    Entry<K,V> e = new Entry<>(key, value, parent); //创建新节点,设置key value 父节点
    if (cmp < 0)
        parent.left = e;
    else
        parent.right = e;
    fixAfterInsertion(e); //插入新节点后 对 红黑树继续宁
    size++;
    modCount++;
    return null;
}
public V put(K key, V value) {
    Entry<K,V> t = root; //得到根节点
    if (t == 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) {
        do {
            parent = t;
            cmp = cpr.compare(key, t.key); //根据比较器 比较当前节点与插入节点的key
            if (cmp < 0) //如果当前节点较小 
                t = t.left;
            else if (cmp > 0) //如果当前节点较大
                t = t.right;
            else //如果相同
                return t.setValue(value);
        } while (t != null);
    }
    else { //如果没有传入一个比较器
       if (key == null)
        throw new NullPointerException();
    @SuppressWarnings("unchecked")
    //尝试将key转为Comparable ,如果没有实现此接口,会报错
        Comparable<? super K> k = (Comparable<? super K>) key;
    do {
        parent = t;
        cmp = k.compareTo(t.key);//尝试用key自身的compareTo方法比较
        if (cmp < 0)
            t = t.left;
        else if (cmp > 0)
            t = t.right;
        else
            return t.setValue(value);
    } while (t != null);
}
Entry<K,V> e = new Entry<>(key, value, parent); //此时,找到插入的位置,创建新的节点,传入参数,以及对父节点的引用
if (cmp < 0)
    parent.left = e;
else
    parent.right = e;
fixAfterInsertion(e); //插入节点后,对红黑树进行相应的变化
size++;
modCount++;
return null;

 

TreeMap 的简单解释

标签:mui   实现   possibly   键值对   父节点   public   except   class   cal   

原文地址:https://www.cnblogs.com/zwb1/p/12080633.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!