在学生类中重写comparaTo方法,使学生按照id的顺序,从小到大排序,并且不允许出现重复的学生。我们知道,comparaTo方法中,返回值大于零,则上一个数大于这个数,若为0,则相等,若是小于号,则上一个数小于这个数。我们让上同学的id减去这个同学的id,则实现了数据根据id 从小到大排序,但是需要注意的一点是,有些同学的id虽然相等,但是他们的姓名其他却不相等,这个时候因为id相等,想减为0,所以后面那个同学就添加不进去,所以需要对comparaTo方法进行改写,当id相等时,比较他们的姓名,若姓名不等,则可以正常添加。
学生类(对该方法进行了重写)
@Override
public int compareTo(Student o) {
int num = this.id-o.id;
int num2 = num==0?this.name.compareTo(o.name):num;
return num2;
}
测试类
import java.util.*;
public class Main {
public static void main(String[] args) {
TreeSet<Student> treeSet = new TreeSet<Student>();
Student s1 = new Student(2, "张三", "男", "西安");
Student s2 = new Student(1, "李四", "女", "咸阳");
Student s3 = new Student(3, "王五", "男", "西安");
Student s4 = new Student(4, "刘双", "女", "安康");
Student s5 = new Student(4, "刘双", "女", "安康");
//添加学生对象到集合中
treeSet.add(s1);
treeSet.add(s2);
treeSet.add(s3);
treeSet.add(s4);
treeSet.add(s5);
//遍历集合
for (Student s : treeSet) {
System.out.println(s.toString());
}
}
}