Set集合 TreeSet 自然排序Comparable的使用

浏览 32 次     时间 2021-08-23 23:26:34     作者 有勇气的牛排    标签 Java

1 Comparable的使用

1、用TreeSet集合存储自定义对象，无参构造方法使用的是自然排序对元素进行排序的

2、自然排序，就是让元素所属的类实现Comparable接口，重写comparable接口，重写compareTo，重写compareTo方法

3、重写方法时，一定要注意排序规则必须按照要求的主要条件和次要条件

2 需求介绍

1、存储学生对象并遍历，创建TreeSet集合使用无参构造方法

2、要求：按照年龄从小到大排序，年龄相同时，按照姓名的字母顺序排序

3 实战演练

Student.java

package 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;
    }

    // 重写Compare里面的方法
    @Override
    public int compareTo(Student s) {
//        return 0;     // 认为重复元素
//        return 1;     // 升序存储
//        return -1;    // 降序存储

        // 按照年龄 从小到大排序
//        int num = this.age - s.age; // s2-s1

        // 按照年龄 从大到小排序
        int num = s.age - this.age; // s1-s2

        // 年龄相同 名字不同 不存储 所以修改
        int num2 = num ==0?this.name.compareTo(s.name):num;

        return num2;
    }
}

TreeSetDemo2.java

package TreeSet;

import java.util.TreeSet;

public class TreeSetDemo2 {
    public static void main(String[] args) {

        // 创建学生对象
        TreeSet<Student> ts = new TreeSet<Student>();

        // 创建学生对象
        Student s1 = new Student("xishi", 20);
        Student s2 = new Student("wangzhaojun", 26);
        Student s3 = new Student("diaochan", 30);
        Student s4 = new Student("yangyuhuan", 22);


        Student s5 = new Student("tom", 22);

        // 把学生添加到集合
        ts.add(s1);
        ts.add(s2);
        ts.add(s3);
        ts.add(s4);
        ts.add(s5);

        // 遍历集合
        for (Student s : ts) {
            System.out.println(s.getName() + "," + s.getAge());
        }

    }
}