集合源码学习(八):HashSet和TreeSet

何为Set？

Set定义如下：

public interface Set<E> extends Collection<E>

如上代码，在Set只是简单的继承自Collection，但是在jdk代码解释中表明Set集合有以下特征。

只有一个元素

不允许重复

在Java中，Set集合的典型实现有HashSet和TreeSet，他们分别是基于HashMap和TreeMap实现的，关于这两个的分析，可以看我写过的分析博客：

HashMap

TreeMap

其中的主题代码都是基于上两个Map的实现。

HashSet

在HashSet中，有如下定义头：

public class HashSet<E>
extends AbstractSet<E>
implements Set<E>, Cloneable, java.io.Serializable

首先实现了基本的Set接口，并且是可克隆以及可以被序列化的。

而在内部实现代码里面，定义了一个私有的HashMap变量：

private transient HashMap<E,Object> map;

这样以来，其实在HashSet里面的所有关于set操作，都是使用map来实现的，

而关于value部分（因为HashMap里面有key和value），则是用一个全局常量PRESENT。

这里举两个例子：

private static final Object PRESENT = new Object();
//判断是否为null
public boolean isEmpty() {
return map.isEmpty();
}
//添加元素
public boolean add(E e) {
return map.put(e, PRESENT)==null;
}
//删除元素
public boolean remove(Object o) {
return map.remove(o)==PRESENT;
}

TreeSet

在TreeSet中定义头：

public class TreeSet<E> extends AbstractSet<E>
implements NavigableSet<E>, Cloneable, java.io.Serializable

其中NavigableSet则是继承子SortedSet，所以TreeSet内部也是有序的。而另一个重要的点则为，

TreeSet是基于TreeMap实现的，类似于上面的HashSet的操作，都是里面隐式用TreeMap来进行操作。

所以TreeSet实际上包含了TreeMap的key的所有特征，并且同样是用一个PRESENT的常量表示value。

// Dummy value to associate with an Object in the backing Map
private static final Object PRESENT = new Object();

这里关于TreeSet的调用代码就不多讲了，有兴趣可以先看集合源码学习：TreeMap，TreeMap的key集合就是TreeSet，可以先研究完TreeMap，过一遍TreeSet就可以了。