TreeSet集合排序原理

1、自然顺序

即类要实现Comparable接口,并重写compareTo()方法,TreeSet对象调用add()方法时,会将存入的对象提升为Comparable类型,然后调用对象中的compareTo()方法进行比较,根据比较的返回值进行存储。
因为TreeSet底层是二叉树,当compareTo方法返回0时,不存储;当compareTo方法返回正数时,存入二叉树的右子树;当compareTo方法返回负数时,存入二叉树的左子树。如果一个类没有实现Comparable接口就将该类对象存入TreeSet集合,会发生类型转换异常。
例如下面这个Preson类:
TreeSet集合排序原理_第1张图片
此时如果将Person对象存入TreeSet集合,会按照怎样的顺序存储呢?这是就要看重写的compareTo()方法的返回值,即先按名字长度从短到长排序,如果名字的长度一样,则按照名字在字典中顺序排序,如果还一样,再按照年龄从小到大排序,如果年龄也一样,则返回0,即认为这是相同的元素,不存储。

2、比较器顺序Comparator

创建TreeSet对象的时候可以指定一个比较器,即传入一个Comparator对象,那么TreeSet会优先按照Comparator中的compare()方法排序,compare方法中有两个参数,第一个是调用该方法的对象,第二个值集合中已经存入的对象。
TreeSet集合排序原理_第2张图片
String类继承了Comparable接口,重写的compareTo()方法是按照字典顺序排列的,而传入了一个Comparator对象后,就按照compare()方法中的字符串长度从短到长排列。运行结果如下:
这里写图片描述
这里可以用匿名内部类来实现:
TreeSet集合排序原理_第3张图片

你可能感兴趣的:(Java学习笔记)