TreeSet 是基于 TreeMap 实现的。TreeSet 中含有一个 NavigableMap 类型的成员变量 m,而 m 实际上是 TreeMap 的实例。TreeSet 是 SortedSet 接口的实现类,TreeSet 可以保证元素处于排序状态,它采用红黑树的数据结构来存储集合元素。TreeSet 支持两种排序方法:自然排序和定制排序,默认采用自然排序。
自然排序
TreeSet 会调用集合元素的 compareTo(Object obj) 方法来比较元素的大小关系,然后将元素按照升序排列,这就是自然排序。如果试图将一个对象添加到 TreeSet 集合中,则该对象必须实现 Comparable 接口,否则会抛出异常。当一个对象调用方法与另一个对象比较时,例如 obj1.compareTo(obj2),如果该方法返回 0,则两个对象相等;如果返回一个正数,则 obj1 大于 obj2;如果返回一个负数,则 obj1 小于 obj2 。
Java 常用类中已经实现了 Comparable 接口的类有以下几个:
-
BigInteger、BigDecimal 以及所有数值型对应的包装类:按照它们对应的数值大小进行比较。
-
Character:按照字符的 unicode 值进行比较。
-
Boolean:true 对应的包装类实例大于 false 对应的包装类实例。
-
String:按照字符串中的字符的 unicode 值进行比较。
-
Date、Time:后面的时间、日期比前面的时间、日期大。
对于TreeSet集合而言,它判断两个对象是否相等的标准是:两个对象通过compareTo(Object obj)方法比较是否返回 0,如果返回 0 则相等。
定制排序
想要实现定制排序,需要在创建 TreeSet 集合对象时,提供一个 Comparator 对象与该 TreeSet 集合关联,由 Comparator 对象负责集合元素的排序逻辑。
综上:自然排序实现的是 Comparable
接口,定制排序实现的是 Comparator
接口。
常用方法
// 将指定的元素添加到此 set(如果该元素尚未存在于 set 中)。
boolean add(E object)
// 将指定 collection 中的所有元素添加到此 set 中。
boolean addAll(Collection<? extends E> collection)
// 移除此 set 中的所有元素。
void clear()
// 如果此 set 不包含任何元素,则返回 true。
boolean isEmpty()
// 如果此 set 包含指定的元素,则返回 true。
boolean contains(Object object)
// 将指定的元素从 set 中移除(如果该元素存在于此 set 中)。
boolean remove(Object o)
// 返回 set 中的元素数(set 的容量)。
int size()
// 返回此 set 中当前第一个(最低)元素。
E first()
// 返回此 set 中当前最后一个(最高)元素。
E last()
// 获取并移除第一个(最低)元素;如果此 set 为空,则返回 null。
E pollFirst()
// 获取并移除最后一个(最高)元素;如果此 set 为空,则返回 null。
E pollLast()
// 返回此 set 中严格小于给定元素的最大元素;如果不存在这样的元素,则返回 null。
E lower(E e)
// 返回此 set 中严格大于给定元素的最小元素;如果不存在这样的元素,则返回 null。
E higher(E e)
// 返回此 set 中小于等于给定元素的最大元素;如果不存在这样的元素,则返回 null。
E floor(E e)
// 返回此 set 中大于等于给定元素的最小元素;如果不存在这样的元素,则返回 null。
E ceiling(E e)
// 返回对此 set 中的元素进行排序的比较器;如果此 set 使用其元素的自然顺序,则返回 null。
Comparator<? super E> comparator()
// 返回在此 set 中的元素上按升序进行迭代的迭代器。
Iterator<E> iterator()
// 返回在此 set 中的元素上按降序进行迭代的迭代器。
Iterator<E> descendingIterator()
// 返回此 set 中所包含元素的逆序视图。
NavigableSet<E> descendingSet()
// 返回此 set 的部分视图,其元素严格小于 toElement。
SortedSet<E> headSet(E toElement)
// 返回此 set 的部分视图,其元素小于(或等于,如果 inclusive 为 true)toElement。
NavigableSet<E> headSet(E toElement, boolean inclusive)
// 返回此 set 的部分视图,其元素从 fromElement(包括)到 toElement(不包括)。
SortedSet<E> subSet(E fromElement, E toElement)
// 返回此 set 的部分视图,其元素范围从 fromElement 到 toElement。
NavigableSet<E> subSet(
E fromElement, boolean fromInclusive, E toElement, boolean toInclusive)
// 返回此 set 的部分视图,其元素大于等于 fromElement。
SortedSet<E> tailSet(E fromElement)
// 返回此 set 的部分视图,其元素大于(或等于,如果 inclusive 为 true)fromElement。
NavigableSet<E> tailSet(E fromElement, boolean inclusive)