通过实现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方法即可。