1.首先来看TreeMap源码来看它排序的本质
1.1 两种定义比较规则的方式
一:当实例化TreeMap(Set)时实现了Comparator比较器的话会按照**compare()**方法中的比较规则进行比较排序,源码如下图所示:
comparator在类中通过有参构造传入
二:实例化TreeMap(Set)时没有实现Comparator比较器的话,放进TreeSet 的对象要实现Comparable接口并重写compareTo方法实现比较排序,源码如下图所示:
一些常用的包装类都是实现了comparable接口中的compareTo()方法的,如下图的String类;
所以将这些对象放入TreeMap中时即使不实现Comparator构造器也会按照类中compareTo()的排序规则进行排序。
2.两种方式的具体代码实现
2.1 通过有参构造传入比较器
通过匿名内部类传入比较器
TreeSet set = new TreeSet(new Comparator() {
@Override
public int compare(Object o1, Object o2) {
return 1;
}
});
这下在TreeMap中就会通过compare定义的比较规则进行排序了
下面来看一个使用TreeMap的冷知识
定义一个Person类
public class Person {
String name;
int age;
public Person(String name,int age){
this.name = name;
this.age = age;
}
@Override
public String toString() {
return "Person{" +
"name='" + name + '\'' +
", age=" + age +
'}';
}
}
将person对象add进TreeSet,编译运行
TreeSet set = new TreeSet();
set.add(new Person("1",10));
set.add(new Person("2",11));
System.out.println(set);
会发现出现了报错,错误如下图所示
解答:因为Person类既没有实现Comparable接口,又没有调用TreeSet的有参构造传入比较器,所以会报ClassCastException: se.d1214.Person cannot be cast to java.lang.Comparable这样一个错。
2.2 传入的对象所在的类实现Comparable接口中的compareTo()方法
public class Person implements Comparable {
String name;
int age;
public Person(String name,int age){
this.name = name;
this.age = age;
}
@Override
public String toString() {
return "Person{" +
"name='" + name + '\'' +
", age=" + age +
'}';
}
@Override
public int compareTo(Object o) {
return 0;
}
}
这样代码就不会报错了。
总结:Comparable和Comparator接口两种方式实现TreeSet 的排序