怎么重写TreeMap排序规则重写compareto排序方法

转载

香奈儿 2024-06-06 23:15:47

文章标签 怎么重写TreeMap排序规则 java 比较器构造方法排序规则 文章分类 机器学习人工智能

文章目录

前言
一、自然排序

1.自然排序Comparable的实现
2.案例实现

二、比较器排序Comparator

1.比较器排序Comparator的实现
2.案例实现

三、两种排序方式的对比

1.不同点
2.相同点

总结

前言

TreeSet集合的特点可以将元素按照规则进行排序,这篇笔记里的两种排序方式通常是实现TreeSet集合排序的常用规则。

提示：以下是本篇文章正文内容，下面案例可供参考

一、自然排序

自然排序，就是让元素所属的类实现Comparable接口，重写compareTo(T o)方法

1.自然排序Comparable的实现

用TreeSet集合存储自定义对象，无参构造方法使用的是自然排序对元素进行排序的。自定义的类就必须要实现Comparable接口，重写compareTo(T o)方法，重写compareTo方法时一定要注意排序规则必须按照要求的主要条件和次要条件来写。

2.案例实现

案例需求
存储学生对象并遍历，创建TreeSet集合使用无参构造方法
要求：按照年龄从小到大排序

代码如下：
学生类

public class Student implements Comparable<Student>{
    private String name;
    private int age;

    public Student() {
    }

    public Student(String name, int age) {
        this.name = name;
        this.age = age;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public int getAge() {
        return age;
    }

    public void setAge(int age) {
        this.age = age;
    }

    @Override
    public String toString() {
        return "Student{" +
                "name='" + name + '\'' +
                ", age=" + age +
                '}';
    }

    /***
     * 比较两个对象的大小
     * @param o 表示集合中的元素
     * @return 如果是正值，表示要存入的当前对象大，放右边
     *         如果是负值，表示要存入的当前对象小，入左边
     *         如果返回0,  表示两个对象相等，不存
     */
    @Override
    public int compareTo(Student o) {
        //按照对象的年龄进行排序
        //主要判断条件
        int result = this.age - o.age;//年龄小的放前面，年龄大的放后面，从小到大排
        return result;
    }
}

测试类

代码如下：

public class MyTreeSet2 {
      public static void main(String[] args) {
          //创建集合对象
          TreeSet<Student> ts = new TreeSet<>();
  	    //创建学生对象
          Student s1 = new Student("zhangsan",28);
          Student s2 = new Student("lisi",27);
          Student s3 = new Student("wangwu",29);
          Student s4 = new Student("zhaoliu",28);
          Student s5 = new Student("qianqi",30);
  		//把学生添加到集合
          ts.add(s1);
          ts.add(s2);
          ts.add(s3);
          ts.add(s4);
          ts.add(s5);
  		//遍历集合
          for (Student student : ts) {
              System.out.println(student);
          }
      }
  }

二、比较器排序Comparator

比较器排序，就是让集合构造方法接收Comparator的实现类对象，重写compare(T o1,T o2)方法。

int compare(T o1,T o2)

		比较其两个参数的顺序。 返回负整数，零或正整数，因为第一个参数小于，等于或大于第二个参数

1.比较器排序Comparator的实现

用TreeSet集合存储自定义对象，带参构造方法使用的是比较器排序对元素进行排序的
比较器排序，就是让集合构造方法接收Comparator的实现类对象，重写compare(T o1,T o2)方法
重写方法时，一定要注意排序规则必须按照要求的主要条件和次要条件来写

2.案例实现

案例需求

存储老师对象并遍历，创建TreeSet集合使用带参构造方法
要求：按照年龄从小到大排序，年龄相同时，按照姓名的字母顺序排序

代码如下（示例）：

老师类

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

    public Teacher() {
    }

    public Teacher(String name, int age) {
        this.name = name;
        this.age = age;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public int getAge() {
        return age;
    }

    public void setAge(int age) {
        this.age = age;
    }

    @Override
    public String toString() {
        return "Teacher{" +
                "name='" + name + '\'' +
                ", age=" + age +
                '}';
    }
}

public class MyTreeSet4 {
    public static void main(String[] args) {
        TreeSet<Teacher> ts = new TreeSet<>(
                new Comparator<Teacher>() {
                    /***
                     * 比较两个对象的大小
                     * @param o1 表示要存入的元素
                     * @param o2 表示已经存入到集合中的元素
                     * @return 如果是正值，表示要存入的当前对象大，放右边
                     *          如果是负值，表示要存入的当前对象小，入左边
                     *          如果返回0,  表示两个对象相等，不存
                     */
            @Override
            public int compare(Teacher o1, Teacher o2) {
                //o1表示现在要存入的那个元素
                //o2表示已经存入到集合中的元素

                //主要条件
                int result = o1.getAge() - o2.getAge();
                //次要条件
                result = result == 0 ? o1.getName().compareTo(o2.getName()) : result;
                return result;
            }
        });
        Teacher t1 = new Teacher("zhangsan",23);
        Teacher t2 = new Teacher("lisi",22);
        Teacher t3 = new Teacher("wangwu",24);
        Teacher t4 = new Teacher("zhaoliu",24);

        ts.add(t1);
        ts.add(t2);
        ts.add(t3);
        ts.add(t4);

        System.out.println(ts);
    }
}