TreeSet底层是通过TreeMap来实现的所有功能。
private transient NavigableMap<E,Object> m;//NavigableMap是TreeMap的父类
//定义一个静态常量,TreeSet使用TreeMap实现增加的时候,(key,value)中的value都是这个对象。
private static final Object PRESENT = new Object();
下面来看构造函数
TreeSet(NavigableMap<E,Object> m) {
this.m = m;
}
//无参构造函数,新建一个TreeMap对象传给m
public TreeSet() {
this(new TreeMap<E,Object>());
}
//通过传入比较器来创建一个TreeMap对象传给m,我们知道TreeMap可以使用比较器或者自然顺序实现比较排序。
public TreeSet(Comparator<? super E> comparator) {
this(new TreeMap<>(comparator));
}
//通过传入一个集合,调用无参构造函数,然后将集合所有元素有序插入到m
public TreeSet(Collection<? extends E> c) {
this();
addAll(c);
}
//传入一个实现了SortedSet接口的类,而SortedSet继承了Iterable接口,需要实现一个返回迭代器的函数。
public TreeSet(SortedSet<E> s) {
this(s.comparator());
addAll(s);
}
TreeSet元素插入
public boolean add(E e) {
//通过TreeMap实现插入,key是e,值是上面提到的静态常量PRESENT
return m.put(e, PRESENT)==null;
}
这里说一下,TreeMap是通过自然顺序,或者比较器进行比较排序,当构造器没有传入比较器时候,就需要key实现Comparable接口,实现自然排序,否则会报ClassCastException.这里TreeSet底层就是通过TreeMap实现的,所以需要TreeSet类型参数实现了 Comparable接口,或者新增的元素实现了Comparable接口。
TreeSet元素的删除
public boolean remove(Object o) {
return m.remove(o)==PRESENT;
}
TreeSet元素的遍历
public static void main(String[] args) {
TreeSet<Integer> t=new TreeSet<Integer>();
t.add(1);
t.add(2);
t.add(3);
//正序遍历
for(int i:t){
System.out.println(i);
}
//倒叙遍历
for(int i:t.descendingSet()){
System.out.println(i);
}
}
///运行结果如下
1
2
3
3
2
1
TreeSet实现了AbstractSet类也就间接继承了Iterable接口,所以可以直接用foreach正序遍历,descendingSet()是反向遍历。