源码解析java集合框架，TreeSet源码

一、TreeSet剖析

TreeSet与TreeMap数据结构底层都是红黑树，区别就是每个数据的结构不一样，TreeMap存储的是key-value键值对，TreeSet值存储一个值，且值是存放在TreeMap的键之上，TreeSet底层实现就是TreeMap，底层都是红黑树。

TreeSet，红黑树

TreeSet是通过TreeMap实现，TreeSet直接把值放在TreeMap的key上，value放置了PRESENT空对象。

 

二、源码解读

2.1 构造方法

public TreeSet() {
        this(new TreeMap());
    }

 TreeSet构造方法中直接调用了TreeMap的构造方法，传入空值。

 

2.2 TreeSet添加元素，add(E e)方法

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

private transient NavigableMap m;

TreeSet的add(E e)方法调用了Map接口中的put(k,v)方法，构造方法中使用的是TreeMap对象，所以调用的是Map接口实现类TreeMap的put方法。

 

2.3 TreeSet移除元素，remove(Object o) 方法

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

调用了TreeMap的remove方法。

TreeSet底层没有自己的实现，底层是使用TreeMap，它们的关系类似于HashSet与HashMap的关系，HashSet底层是HashMap实现。