首先说明用TreeSet集合存储自定义时,无参构造方法使用的是自然排序对元素进行排序的
然后,使用自然排序的前提是,让元素所属的类实现Comparable接口并重写comparaTo(T o)方法,来完成。
其次要注意,重写方法时,一定要注意排序规则必须按照要求的主要条件和条件来写。 下面通过一个需求来具体实现一下。
需求:存储学生对象并遍历,创建TreeSet集合使用无参构造方法。
要求:按照年龄从小到大排序,年龄相同时,按照姓名的字母顺序排序
分析可知,需要两个类:一个学生类(Student)一个操作类(TreeSetDemo)学生类中有两个成员遍历:姓名(name)、年龄(age)。(仅记录重写代码部分)
操作类代码:
public class TreeSetDemo { public static void main(String[] args) { //创建集合对象 TreeSet<Student> treeSet = new TreeSet<Student>(); //创建学生对象 Student s1 = new Student("xishi",29); Student s2 = new Student("wangzhaojun",30); Student s3 = new Student("diaocan",28); Student s4 = new Student("yangyuhuan",31); //添加元素 treeSet.add(s1); treeSet.add(s2); treeSet.add(s3); treeSet.add(s4); //遍历 for (Student s : treeSet){ System.out.println(s); } } }
接下来测试,在学生类中实现了Comparable<E>接口,并重写compareTo(T o)方法后,操作类的运行情况
在Student类中重写compareTo(T o)方法代码:(通过返回值的不同,来影响排序结果,具体看运行结果)
@Override public int compareTo(Student o) { return 0; } /* 运行结果:xishi,29 */ @Override public int compareTo(Student o) { return 1; } /* 运行结果: xishi,29 wangzhaojun,30 diaocan,28 yangyuhuan,31 */ @Override public int compareTo(Student o) { return -1; } /* 运行结果: yangyuhuan,31 diaocan,28 wangzhaojun,30 xishi,29 */
通过三次修改返回值,从运行结果中可以得出结论:
当返回值为0时,认为是重复元素,不进行存储。
当返回值为1时,按照存储顺序,正向排序(升序)(因为是正向排序,可以理解从小到大的顺序)
当返回值为-1时,按照存储顺序,逆向排序(降序)(同理)
于是,可以通过控制返回值的不同来控制不同的排序结果。接下来实现上面的需求。
对compareTo方法在进行修改,先使得年龄以升序的方式排序,代码:
@Override public int compareTo(Student s) { int num = this.age - s.age; return num; } /* 运行结果: diaocan,28 xishi,29 wangzhaojun,30 yangyuhuan,31 */
接下来完善要求,年龄相同时,按照姓名的字母顺序排序,完整代码如下:
//操作类代码 public class TreeSetDemo { public static void main(String[] args) { //创建集合对象 TreeSet<Student> treeSet = new TreeSet<Student>(); //创建学生对象 Student s1 = new Student("xishi",29); Student s2 = new Student("wangzhaojun",30); Student s3 = new Student("diaocan",28); Student s4 = new Student("yangyuhuan",31); Student s5 = new Student("linqingxia",31); //添加元素 treeSet.add(s1); treeSet.add(s2); treeSet.add(s3); treeSet.add(s4); treeSet.add(s5); //遍历 for (Student s : treeSet){ System.out.println(s.getName()+","+s.getAge()); } } } //学生类中重写代码部分及运行结果 @Override public int compareTo(Student s) { int num = this.age - s.age; //如果年龄相同则比较姓名字母顺序(升序)排列。 int num2 = num == 0 ? this.name.compareTo(s.name) : num; return num2; } /* 运行结果: diaocan,28 xishi,29 wangzhaojun,30 linqingxia,31 yangyuhuan,31 */
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。