java TreeSet 源码分析

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

小世界-_-

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

阅读量212

点赞数

分类专栏： java

本文链接：https://blog.csdn.net/qq_38572383/article/details/83624911

版权

java 专栏收录该内容

41 篇文章 0 订阅

订阅专栏

本文探讨了Java中的TreeSet，揭示其在Map(NavigableMap)基础上实现有序Set的秘密。文章通过源码分析，解答了TreeMap、TreeSet及Collections.sort()方法的比较原理，并对比了HashSet与TreeSet的区别，强调了它们在数据结构、有序性及时间空间复杂度上的差异，为读者提供了选择合适集合类的依据。

摘要由CSDN通过智能技术生成

底层数据结构：

TreeMap

哈哈，是不是很惊喜。

没错，treeSet在map(NavigableMap)的基础上，将存储内容作为键存储在map当中，实现了有序的set。

TreeMap和TreeSet在排序时如何比较元素？Collections工具类中的sort()方法如何比较元素？

源码：

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);
            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;
        do {
            parent = t;
            cmp = k.compareTo(t.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);
    if (cmp < 0)
        parent.left = e;
    else
        parent.right = e;
    fixAfterInsertion(e);
    size++;
    modCount++;
    return null;
}

所以，需要实现比较器Comparator或者存放对象实现Comparable

Collections工具类中的sort()方法如何比较元素：（源码）

public static <T> void sort(List<T> list, Comparator<? super T> c) {
    list.sort(c);
}

list.sort():
default void sort(Comparator<? super E> c) {
    Object[] a = this.toArray();
    Arrays.sort(a, (Comparator) c);
    ListIterator<E> i = this.listIterator();
    for (Object e : a) {
        i.next();
        i.set((E) e);
    }
}

public static void sort(Object[] a, int fromIndex, int toIndex) {
    rangeCheck(a.length, fromIndex, toIndex);
    if (LegacyMergeSort.userRequested)
        legacyMergeSort(a, fromIndex, toIndex);
    else
        ComparableTimSort.sort(a, fromIndex, toIndex, null, 0, 0);
}//Arrays 对Object的排序操作是归并排序（优化后的）

HashSet和TreeSet有什么区别？

hashSet 的底层数据结构是HashMap，所以他俩的区别相当于TreeMap和HashMap的区别.

总结下：

数据结构

hashmap：桶加链表

treemap: 红黑树

是否有序

treemap有序

hashmap无序

是否安全

都不安全

时间、空间复杂度

TreeMap基于红黑树（一种自平衡二叉查找树）实现的，时间复杂度平均能达到O(log n)。
HashMap是基于散列表实现的，时间复杂度平均能达到O(1)。

适用场景