compare方法中,object1代表当前对象,object2代表与当前对象进行比较的对象。
当向Set集合中新添加一个元素的时候,这个元素会和之前集合里面的所有元素进行比较,从而选择插入的位置。
compare方法的返回值是int类型。例如”return o1 - o2“,
返回一个正数,表示o1比o2大,把o1排在o2后面。
返回负数相反。
返回0,表示无所谓。
实例:
import java.util.Comparator;
import java.util.TreeSet;
//集合输入1-10十个数,将1-5放在后面升序排序,6-10放在前面降序排序
public class Student{
public static void main(String[] args) {
TreeSet<Integer> ss = new TreeSet(new Comparator() { //匿名内部类的方式重写compare方法
@Override
public int compare(Object o1, Object o2) {
Integer a1 = (int)o1; //将对象强转为int类型
Integer a2 = (int)o2;
if ( a1 < 5 && a2 < 5){ //总体规则,返回负数,a1再a2前面,反之后面,0的话不做变化
return a1 - a2;
}else if ( a1 > 5 && a2 > 5){
return a2 - a1;
}else if ( a1 > 5 && a2 < 5){
return -1;
}else if ( a1 < 5 && a2 > 5){
return 1;
}else {
return 0;
}
}
});
for (int i = 0 ; i < 10 ; i++ ){ //向集合中添加元素
ss.add(i+1);
}
System.out.println(ss); //输出集合
}
}
[10, 9, 8, 7, 6, 1, 2, 3, 4]