通过实现Comparable接口结合TreeSet来对对象自动排序

时间:2023-03-09 08:50:46
通过实现Comparable接口结合TreeSet来对对象自动排序

经过会遇到这样的情况,对于某个对象数组或者链表要按照一定的规则进行排序,那么我们该怎么做呢?

如遇到这样的需求:

1.需求1

对于学生对象按照年龄进行排序,年龄小的排在前面。

单单看到这样的需求,实现起来是比较简单的,自己写个函数实现一下快速排序或者比较土的冒泡排序,按照年龄来排序就可以了。我们再看看下面的需求

2.需求2

对于学生对象,按照年龄进行排序,年龄小的排在前面,年龄相同的,身高小的排在前面。

3.需求3

对于学生对象,按照年龄进行排序,年龄小的排在前面;年龄相同的,身高小的排在前面;身高相同时,体重轻的排在前面。

。。。

这样的需求,可能会随着页面的复杂化,越来越来复杂,如果要自己去做的话,估计得多层嵌套了,逻辑非常复杂,那是否有现成的比较简单的机制呢。

答案就是实现Comparable接口。

具体样例如下:

public class Student implements Comparable{ 

    private int age; 

     

    public Student(int age) 

    { 

        this.age = age; 

    } 

    @Override 

    public int compareTo(Object obj) { 

        Student stu = (Student)obj; 

        if(this.age>stu.age) 

            return 1; 

        else if(this.age<stu.age) 

            return -1; 

        else 

            return 0; 

    } 

    public int getAge() { 

        return age; 

    } 

    public void setAge(int age) { 

        this.age = age; 

    } 

 

}

调用代码:

public static void main(String[] args) { 

    TreeSet<Student> treeSet = new TreeSet<Student>(); 

    treeSet.add(new Student(1)); 

    treeSet.add(new Student(3)); 

    treeSet.add(new Student(2)); 

    for(Student student:treeSet) 

    { 

        System.out.println(student.getAge()); 

    } 

}

输出结果:

1

2

3

可见自动按照年龄进行了排序,我们再修改一下代码,让学生按照年龄由大到排列:

public int compareTo(Object obj) { 

    Student stu = (Student)obj; 

    if(this.age>stu.age) 

        return -1; 

    else if(this.age<stu.age) 

        return 1; 

    else 

        return 0; 

}

输出结果:

3

2

1

由此可见,这种通过对象实现Comparable接口,结合TreeSet的方式来进行对象排序,还是十分方便的,而且逻辑比较简单,后续如果有变更,直接修改compareTo方法即可。