TreeSet是一个有序集合,TreeSet中的元素将按照升序排列,缺省是按照自然排序进行排列的。当往TreeSet中添加具体的自定义对象时,如果直接添加对象,会抛出ClassCastException的异常,因为TreeSet不知道根据什么规则进行比较。要想知道根据什么规则进行比较,自己自定义的类就必须要去实现Comparator接口,并且实现接口里compare方法,以此来确定比较的规则,然后TreeSet根据这个规则对里面的对象进行排序。下面是一个小的实例。
package test;
import java.util.Comparator;
import java.util.TreeSet;
public class TreeSetTest2 {
public static void main(String[] args) {
//带自定义排序规则的TreeSet
TreeSet set = new TreeSet(new MyComparator());
Person p1 = new Person(10);
Person p2 = new Person(20);
Person p3 = new Person(30);
Person p4 = new Person(40);
Person p5 = new Person(50);
set.add(p1);
set.add(p2);
set.add(p3);
set.add(p4);
set.add(p5);
System.out.println(set);
}
}
/**
* 自定义类,放入TreeSet希望根据Person的score进行排序
* @author Administrator
*
*/
class Person {
int score;
Person(int score){
this.score = score;
}
@Override
public String toString() {
return String.valueOf(score);
}
}
class MyComparator implements Comparator {
/**
* 比较的规则,由自己定义,这里是根据Person的score的属性进行比较的
* @param o1
* @param o2
* @return
*/
@Override
public int compare(Object o1, Object o2) {
Person p1 = (Person)o1;
Person p2 = (Person)o2;
return (p1.score - p2.score);
}
}