TreeSet源码解析

TreeSet 的实现方式与HashSet 的实现方式非常类似，一如既往的先来一波源码，有助于理解：

public class TreeSet<E> extends AbstractSet<E> implements NavigableSet<E>, Cloneable, java.io.Serializable {
    /**
     * The backing map.
     */
    private transient NavigableMap<E,Object> m;

    // 虚拟对象
    private static final Object PRESENT = new Object();

    /**
     * Constructs a set backed by the specified navigable map.
     */
    TreeSet(NavigableMap<E,Object> m) {
        this.m = m;
    }

    // 无参构造器
    public TreeSet() {
        this(new TreeMap<E,Object>());
    }

    // 自定义排序构造器
    public TreeSet(Comparator<? super E> comparator) {
        this(new TreeMap<>(comparator));
    }

    // collection 构造器
    public TreeSet(Collection<? extends E> c) {
        this();
        addAll(c);
    }

    // 有序Set构造器
    public TreeSet(SortedSet<E> s) {
        this(s.comparator());
        addAll(s);
    }
}

从源码中我们能够很明显的看到，TreeSet 底层维护了 TreeMap （就像HashSet 底层维护了 HashMap 一样），拥有TreeMap的所有特性，下面我们从更具体的方法中来看一下 TreeSet 的实现。

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

在 add 方法中，直接调用 TreeMap 中的 put() 方法; 以元素为key 以虚拟对象为 value ，具体实现可以参考TreeMap的解析。在 remove 函数中也是相同的实现思路：

    public boolean remove(Object o) {
        return m.remove(o)==PRESENT;
    }

看过 TreeMap 源码之后在看 TreeSet 就会发现还是十分简明的，TreeSet 解析到此结束。