package en.itcast.set.demo;
import java.util.Comparator;
import en.itcast.demain.Student;
/**
* 自定义一个比较器,用来对学生对象按照姓名进行排序。
* 步骤:实现Comparator接口,覆盖compare方法。将Comparator接口的对象作为参数
* 传递给TreeSet集合的构造函数。
* @author 许荣
* 注意:传入的参数为两个对象(区别于Comparable接口中的compareTo方法只有一个参数)
*
*/
public class ComparatorByName implements Comparator {
@Override
public int compare(Object o1, Object o2) {
Student s1 = (Student)o1;
Student s2 = (Student)o2;
int temp = s1.getName().compareTo(s2.getName());
return temp == 0?s1.getAge() - s2.getAge():temp;
}
}
package en.itcast.set.demo;
import java.util.Iterator;
import java.util.Set;
import java.util.TreeSet;
import en.itcast.demain.Student;
public class TreeSetDemo2 {
public static void main(String[] args) {
//初始化TreeSet集合明确一个比较器
/**
* 1.
* 需求中也有一种情况,元素具备的比较功能不是所需要的,也就是说不想按照自然排序的方法,
* 而是按照自定义的排序方式,对元素进行排序。而且,存储到TreeSet中的元素万一没有
* 比较功能,应该如何排序,为此,可用第二种比较方式--比较器,让集合具备比较功能。
* 2.
* 步骤:实现Comparator接口,覆盖compare方法。将Comparator接口的对象作为参数
* 传递给TreeSet集合的构造函数。
* 3.
* 当传入多个比较器时,那么Comparator接口中的equals方法将就起作用,用于判断比较器是否相同。
*/
Set set = new TreeSet(new ComparatorByName());
set.add(new Student(20, "xiaoqiang"));
set.add(new Student(24, "daniu"));
set.add(new Student(22, "xiaoming"));
set.add(new Student(18, "tudou"));
set.add(new Student(18, "xurong"));
set.add(new Student(19, "dahuang"));
//3.只能用迭代器取出
for(Iterator it = set.iterator();it.hasNext();) {
Student student = (Student)it.next();
System.out.println(student.getAge()+"::"+student.getName());
}
}
}
输出结果:
19::dahuang
24::daniu
18::tudou
22::xiaoming
20::xiaoqiang
18::xurong