首页 > 基础教程 > 集合框架 > TreeSet类

Java TreeSet类

TreeSet是SortedSet接口的实现,元素不论以什么元素插入,在遍历的时候,都会以天然顺序遍历。

TreeSet基于TreeMap的NavigableSet实现，使用元素的自然顺序对元素进行排序，或者根据创建 set 时提供的 Comparator 进行排序，具体取决于使用的构造方法。

构造方法

TreeSet()

构造一个新的空 set，该 set 根据其元素的自然顺序进行排序。

TreeSet(Collection extends E> c)

构造一个包含指定 collection 元素的新 TreeSet，它按照其元素的自然顺序进行排序。

TreeSet(Comparator super E> comparator)

构造一个新的空 TreeSet，它根据指定比较器进行排序。

TreeSet(SortedSet s)

构造一个与指定有序 set 具有相同映射关系和相同排序的新 TreeSet。

TreeSet是带排序的,所以想要为TreeSet增加自定义类型,必须指定排序规则

方法

E ceiling(E e)

返回此 set 中大于等于给定元素的最小元素；如果不存在这样的元素，则返回 null。

Comparator super E> comparator()

返回对此 set 中的元素进行排序的比较器；如果此 set 使用其元素的自然顺序，则返回 null。

Iterator descendingIterator()

返回在此 set 元素上按降序进行迭代的迭代器。

NavigableSet descendingSet()

返回此 set 中所包含元素的逆序视图。

E first()

返回此 set 中当前第一个(最低)元素。

E floor(E e)

返回此 set 中小于等于给定元素的最大元素；如果不存在这样的元素，则返回 null。

SortedSet headSet(E toElement)

返回此 set 的部分视图，其元素严格小于 toElement。

NavigableSet headSet(E toElement, boolean inclusive)

返回此 set 的部分视图，其元素小于(或等于，如果 inclusive 为 true)toElement。

E higher(E e)

返回此 set 中严格大于给定元素的最小元素；如果不存在这样的元素，则返回 null。

Iterator iterator()

返回在此 set 中的元素上按升序进行迭代的迭代器。

E last()

返回此 set 中当前最后一个(最高)元素。

E lower(E e)

返回此 set 中严格小于给定元素的最大元素；如果不存在这样的元素，则返回 null。

E pollFirst()

获取并移除第一个(最低)元素；如果此 set 为空，则返回 null。

E pollLast()

获取并移除最后一个(最高)元素；如果此 set 为空，则返回 null。

NavigableSet subSet(E fromElement, boolean fromInclusive, E toElement, boolean toInclusive)

返回此 set 的部分视图，其元素范围从 fromElement 到 toElement。

SortedSet subSet(E fromElement, E toElement)

返回此 set 的部分视图，其元素从 fromElement(包括)到 toElement(不包括)。

SortedSet tailSet(E fromElement)

返回此 set 的部分视图，其元素大于等于 fromElement。

NavigableSet tailSet(E fromElement, boolean inclusive)

返回此 set 的部分视图，其元素大于(或等于，如果 inclusive 为 true)fromElement。

Comparator排序规则

代码如下：

public class Test {

public static void main(String[] args) {

TreeSet set = new TreeSet(new PersonComparator());

set.add(new Person("lwc", 80));

set.add(new Person("nxj", 70));

set.add(new Person("lp", 60));

set.add(new Person("fy", 75));

Iterator ite = set.iterator();

while (ite.hasNext()) {

Person p = (Person)ite.next();

System.out.println(p.name);

}

}

}

class Person {

String name;

int score;

public Person(String name, int score) {

this.name = name;

this.score = score;

}

}

class PersonComparator implements Comparator {

public int compare(Object o1, Object o2) {

Person p1 = (Person) o1;

Person p2 = (Person) o2;

return p1.score - p2.score;

}

}

例子

public static void main(String[] args) {

//      test1();

test2();

}

/**

* 自然排序

*/

private static void test1() {

TreeSet treeSet=new TreeSet();

treeSet.add("a");

treeSet.add("d");

treeSet.add("b");

treeSet.add("e");

treeSet.add("c");

for (String string : treeSet) {

System.out.println(string);

}

}

/**

*

* TreeSet添加自定义对象时满足其一 ①构造时传入比较器 ②实现Comparable接口并重写compare方法,否则抛出异常

*  java.lang.ClassCastException:

*  com.yu.bean.Info cannot be cast to java.lang.Comparable

*/

private static void test2() {

TreeSet infosSet =new TreeSet(new Comparator() { // 传入比较器，保证顺序

@Override

public int compare(Info o1, Info o2) {

int num=o1.getId()-o2.getId();

num = num==0?o1.hashCode()-o2.hashCode():num;

return num;

}

}) ;

infosSet.add(new Info(0, "qqq"));

infosSet.add(new Info(1, "wew"));

infosSet.add(new Info(3, "ete"));

infosSet.add(new Info(2, "bjk"));

infosSet.add(new Info(1, "wew")); // 重复元素，compare方法返回0，不会被插入到TreeSet

for (Info info : infosSet) {

System.out.println(info);

}

//      output:

//      Info [id=0, adress=qqq]

//      Info [id=1, adress=wew]

//      Info [id=2, adress=bjk]

//      Info [id=3, adress=ete]

//  可以看出，Comparator保证了元素的有序性，重复元素将不会被添加到TreeSet

}

总结

1. 基于TreeMap实现的SortedSet

2. 排序后按升序排列元素

3. 非线程安全

版权声明：本文为JAVASCHOOL原创文章，未经本站允许不得转载。