深入Java集合学习系列:TreeSet 详解

时间:2022-10-24 17:57:45

http://www.cnblogs.com/meng72ndsc/archive/2010/12/23/1914477.html

TreeSet是依靠TreeMap来实现的。
TreeSet是一个有序集合,TreeSet中的元素将按照升序排列,缺省是按照自然排序进行排列,意味着TreeSet中的元素要实现Comparable接口。或者有一个自定义的比较器。
我们可以在构造TreeSet对象时,传递实现Comparator接口的比较器对象。

 

深入Java集合学习系列:TreeSet 详解
深入Java集合学习系列:TreeSet 详解
   
   
   
import java.util.Iterator;
import java.util. * ;

public class TreeSetTest {
public static void main(String[] args) {
Set ts
= new TreeSet();
ts.add(
" abc " );
ts.add(
" xyz " );
ts.add(
" rst " );
Iterator it
= ts.iterator();
while (it.hasNext()) {
System.out.println(it.next());
}
}
}
深入Java集合学习系列:TreeSet 详解

 

输出:

 

abc

rst

xyz

打印结果不是和先前加入的顺序一样,它是按照一个字母的排序法进行排序的。这是因为String 类实现了Comparable接口。

如果我们自己定义的一个类的对象要加入到TreeSet当中,那么这个类必须要实现Comparable接口。

 

深入Java集合学习系列:TreeSet 详解
深入Java集合学习系列:TreeSet 详解
   
   
   
import java.util.Iterator;
import java.util. * ;

public class TreeSetTest {
public static void main(String[] args) {
Set ts
= new TreeSet();
ts.add(
new Teacher( " zhangsan " , 1 ));
ts.add(
new Teacher( " lisi " , 2 ));
ts.add(
new Teacher( " wangmazi " , 3 ));
ts.add(
new Teacher( " mazi " , 3 ));
Iterator it
= ts.iterator();
while (it.hasNext()) {
System.out.println(it.next());
}
}
}


class Teacher implements Comparable {
int num;
String name;

Teacher(String name,
int num) {
this .num = num;
this .name = name;
}

public String toString() {
return " 学好 " + num + " 姓名 " + name;
}

public int compareTo(Object o) {
Teacher ss
= (Teacher) o;
int result = num > ss.num ? 1 : (num == ss.num ? 0 : - 1 );
if (result == 0 ) {
result
= name.compareTo(ss.name);
}
return result;
}
}
深入Java集合学习系列:TreeSet 详解

 

深入Java集合学习系列:TreeSet 详解

 

注意:如果将int result = num > ss.num ? 1 : (num == ss.num ? 0 : -1);写成了int result = ss.num > num ? 1 : (ss.num == num ? 0 : -1);那么得到的结果就是倒序排列的,并不是升序的。

 

在使用Arrays对数组中的元素进行排序的时候,可以传递一个比较器。

在使用Collections对集合中的元素进行排序的时候,可以传递一个比较器。

 

那么在使用TreeSet对加入到其中的元素进行排序的时候可以传入一个比较器吗?

 

  
  
  
public TreeSet(Comparator <? super E > comparator) {
this ( new TreeMap < E,Object > (comparator));
}

通过查看它的构造方法就知道可以传入一个比较器。

 

构造一个新的空TreeSet,它根据指定比较器进行排序。插入到该 set 的所有元素都必须能够由指定比较器进行相互比较:对于 set 中的任意两个元素 e1 和e2,执行 comparator.compare(e1, e2) 都不得抛出 ClassCastException。如果用户试图将违反此约束的元素添加到 set 中,则 add 调用将抛出 ClassCastException。

深入Java集合学习系列:TreeSet 详解
深入Java集合学习系列:TreeSet 详解
   
   
   
import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;

public class TreeSetTest {
public static void main(String[] args) {
TreeSet ts
= new TreeSet( new Teacher.TeacherCompare());
ts.add(
new Teacher( " zhangsan " , 2 ));
ts.add(
new Teacher( " lisi " , 1 ));
ts.add(
new Teacher( " wangmazi " , 3 ));
ts.add(
new Teacher( " mazi " , 3 ));
Iterator it
= ts.iterator();
while (it.hasNext()) {
System.out.println(it.next());
}
}
}

class Teacher {
int num;
String name;

Teacher(String name,
int num) {
this .num = num;
this .name = name;
}

public String toString() {
return " 学号: " + num + " 姓名: " + name;
}

static class TeacherCompare implements Comparator { // 老师自带的一个比较器

public int compare(Object o1, Object o2) {
Teacher s1
= (Teacher) o1; // 转型
Teacher s2 = (Teacher) o2; // 转型
int result = s1.num > s2.num ? 1 : (s1.num == s2.num ? 0 : - 1 );
if (result == 0 ) {
result
= s1.name.compareTo(s2.name);
}
return result;
}

}

}
深入Java集合学习系列:TreeSet 详解

 

深入Java集合学习系列:TreeSet 详解