TreeSet的去重原理

最新推荐文章于 2025-06-26 00:04:33 发布

原创最新推荐文章于 2025-06-26 00:04:33 发布 · 2k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#java #开发语言 #后端

本文探讨了Java中TreeSet的去重机制。在未重写Comparator时，TreeSet依赖对象的Comparable接口进行排序和去重。通过源码分析，当key为String类型时，TreeSet利用Comparable比较字母顺序，相同字母顺序的元素将不会被添加，从而实现去重功能。

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

TreeSet去重分两种，第一种是我们重写了Comparator匿名类，第二种则是没有重写。

一，我们先分析一下没有重写的情况，当我们没有重写Comparator匿名类时

首先可以看看TreeSet的主要去重源代码

public V merge(K key, V value, BiFunction<? super V, ? super V, ? extends V> remappingFunction) {
    Objects.requireNonNull(remappingFunction);
    Objects.requireNonNull(value);
    Entry<K,V> t = root;
    if (t == null) {
        addEntryToEmptyMap(key, value);
        return value;
    }
    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);
            if (cmp < 0)
                t = t.left;
            else if (cmp > 0)
                t = t.right;
            else return mergeValue(t, value, remappingFunction);
        } while (t != null);
    } else {
        Objects.requireNonNull(key);
        @SuppressWarnings("unchecked")
        Comparable<? super K> k = (Comparable<? super K>) key；
        do {
            parent = t;
            cmp = k.compareTo(t.key);
            if (cmp < 0)
                t = t.left;
            else if (cmp > 0)
                t = t.right;
            else return mergeValue(t, value, remappingFunction);
        } while (t != null);
    }
    addEntry(key, value, parent, cmp < 0);
    return value;
}

可以看到上方的源代码，当我们没有重写Comparator匿名类时，TreeMap自身的Comparator属性就是null，赋给cpr。cpr为null，故不会执行第一个if条件语句，转而执行第二个else条件语句，也就是紫色部分，进去之后，执行到深绿色的代码时，此部分会将key向上转型成Comparable类型，因为key这个时候已经是String类型，而String类型实现了Comparable接口，所以可以用多态的方式结构实现接口的方法，往后就进入do，while循环调用String的compare方法比较字母大小，按字母的大小进行排序，如果字母大小一致，则返回0赋予cmp，cmp=0则执行红色代码块的语句。