TreeSet,TreeMap实现自定义排序(Comparable 接口和Comparator 接口的使用)

概要

我们知道java中的TreeSet和TreeMap是一种可以进行自动排序的结构,他不同于Collections.sort(sortedList)这种外部的主动排序,而是根据它自身的结构特点,在插入的那一刻就是按照某个顺序进行排列的.那么我们要如何自定义这个排序呢?比如我想让我的集合实现里面元素的某个字段进行排序.有两种方式去实现这样的自定义排序,一个是实现Comparable 接口,另一个是实现Comparator 接口.

1.Comparable 接口:

1.集合的元素类实现Comparable 接口;
2.重写里面的compareTo方法
3.将 该元素类作为TreeSet的集合元素时,他会自动排序

假如我想实现一个按照Person里面的age字段升序排列,age小的person排在前面,age大的排在后面:

直接看代码:
实现Comparable 接口:

//1.实现Comparable 接口
public class Person implements Comparable<Person> {
    private String name;
    private int age;

    // 构造方法和 getter/setter
	//2.重写compareTo方法方法
    @Override
    public int compareTo(Person other) {
    	if (this.age > other.age) {//当本身this的年龄大于形参的年龄时
    		return 1;  // 返回1表示this排在形参后面
    	} else if (this.age < other.age) {//当本身this的年龄小于形参的年龄时
        	return -1; // 返回1表示this排在形参前面
    	} else {
        	return 0;  // 两个对象相等，返回 0
    	}
    }
}

插入元素时自动排序:

public static void main(String[] args) {
    // 插入时自动排序
    Set<Person> people = new TreeSet<>();
    people.add(new Person("Alice", 30));
    people.add(new Person("Bob", 25));
    people.add(new Person("Charlie", 35));

    System.out.println(people); // 输出：[Bob: 25, Alice: 30, Charlie: 35]
}

可以看到,Comparable 接口是使用在集合的元素类上面的,让集合的元素类去实现这个接口,TreeMap或TreeSet在插入该类元素时就可以自动进行自定义的排序.

2.Comparator 接口:

1. 定义一个比较器类,即是用一个类去实现Comparable 接口
2. 重写里面的compare(obj o1, obj o2)方法
3. 用该比较器去初始化TreeSet或TreeMap

直接看代码:
还是这个元素类:

public class Person {
    private String name;
    private int age;

    // 构造方法和 getter/setter
}

定义比较器,实现对应方法:

//1.自定义的比较器
public static class AgeComparator implements Comparator<Person> {
	//2.实现的compare()方法(升序)
	
    @Override
    public int compare(Person p1, Person p2) {
    	if (p1.getAge() > p2.getAge()) {//当第一个参数的年龄大于第二个参数的年龄时
        	return 1;  // 返回1的时候表示第一个参数排在第二个参数的后面
    	} else if (p1.getAge() < p2.getAge()) {//当第一个参数的年龄小于第二个参数的年龄时
        	return -1; // 返回-1的时候表示第一个参数排在第二个参数的前面
    	} else {
        	return 0;  // 年龄相同，视为相等
    	}
    }
}

初始化TreeSet时将比较器传进去:

public static void main(String[] args) {
    // 3.将自定义的比较器在TreeSet初始化时当作参数传进去
    Set<Person> people = new TreeSet<>(new AgeComparator());

    people.add(new Person("Alice", 30));
    people.add(new Person("Bob", 25));
    people.add(new Person("Charlie", 35));

    System.out.println(people); // 输出：[Bob: 25,Alice: 30, Charlie: 35]
}

可以看到,不同于Comparable 接口是让集合的元素类去实现,实现Comparator 接口更像是去定义一个比较器,在初始化集合的时候传递,当作一种初始化集合的手段.

3.如何重写他们的函数,达到自己想要的排序方式

可以先理解compare(obj o1,obj o2)和compareTo(obj o)他们两个函数返回不同返回值时所代表的含义:

1. return 1 表示第一个参数排在第二个参数的后面(对于compare(obj o1,obj o2)函数),或者是this排在参数的后面(对于compareTo(obj o)函数);
2. return -1 表示第一个参数排在第二个参数的前面(对于compare(obj o1,obj o2)函数),或者是this排在参数的前面(对于compareTo(obj o)函数);
3. return 0 表示两个并列位置.

再根据他们的返回值的意义去设置return 1,和return -1 返回时的条件,就可以做到自定义排序了.