TreeSet.add()方法一细节

最新推荐文章于 2021-12-29 20:45:20 发布

云原生之家

最新推荐文章于 2021-12-29 20:45:20 发布

阅读量5.3k

点赞数 5

CC 4.0 BY-SA版权

分类专栏： java相关文章标签： TreeSet add

本文链接：https://blog.youkuaiyun.com/xtayfjpk/article/details/24806887

java相关专栏收录该内容

39 篇文章

订阅专栏

本文探讨了在使用TreeSet的add()方法时遇到的问题，特别是当使用自定义比较器时，compare()方法返回0的影响。揭示了此行为背后的源码逻辑。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

TreeSet.add()方法一细节

最近在一次使用TreeSet.add()方法中遇到了一个很迷惑的问题：往一个TreeSet集合中添加几个对象的时候发现有些没有添加进去，一开始是怀疑hashcode与equals的问题，因为Set集合是不允许添加两个相同对象的，但经查检确认没有问题，几经测试发现是传入的Comporator对象引起的(即使用的是TreeSet(Comparator<? super E> comparator)构造方法)，当其compare()方法返回0时，只能添加进一个对象，只要当前添加的对象与集合中的任意一个对象比较时compare()方法返回0，那么该对象就添加不进去。以前一直以为compare()方法返回0时只是两个对象的排列顺序不确定，后经查看源码才明白其中的原因。
下面是TreeSet的add()方法源码：

public boolean add(E e) {
	return m.put(e, PRESENT)==null;
}

其中m是一个TreeMap对象，其为TreeSet的实现是使用TreeMap的key部分来实现的，PRESENT是一个Object对象，纯粹是为了补全方法参数而已，下面是TreeMap.put()方法源码：

public V put(K key, V value) {
    Entry<K,V> t = root;//根元素
    if (t == null) {
    // TBD:
    // 5045147: (coll) Adding null to an empty TreeSet should
    // throw NullPointerException
    //
    // compare(key, key); // type check
        root = new Entry<K,V>(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值，也就是Set中元素的值
            if (cmp < 0) //如果小于0则放到当前元素的左边
                t = t.left;
            else if (cmp > 0) //如果大于0则放到当前元素的右边
                t = t.right;
            else
                return t.setValue(value);  //否则，即返回0时调用java.util.TreeMap.Entry.setValue(value)方法
        } while (t != null);
    }
    //后面代码省略......
}

下面是java.util.TreeMap.Entry.setValue(value)方法源码：

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

可以看到更新了值后把以前的值返回了。所以当compare方法的返回值为了时，TreeMap.put()方法只是返回的是被替换掉的值，根本就没有加入新的元素，当然m.put(e, PRESENT)==null 返回的也为false了。