这篇文章主要分析compare的比较规则;
TreeSet实现了set接口,可以完成去重、排序操作,java中的大部分API都实现了自定义Comparator接口,重写了compare方法,比如String、Integer等;但实际业务往往需要根据自定义的对象的某些属性进行排序或去重操作,这时候就需要自定义比较器
底层逻辑:compare底层是根据二叉树进行比较排序的:
代码演示:
Integer自己已经实现了比较器,这里是为了方便演示,所以我们自己重写一下
compare方法的两个参数o1、o2,o2作为父层节点,比o2小的返回-1,会存放到o2参数前面,比o2大的返回1,会存放到o2参数后面,相等返回0,不会存放到TreeSet中,这样就实现了去重、排序
debug:
o2 = 1
o1 = 1 和o2相等,返回0,不会存入integers中
o2 = 1
o1 = 2,比o2大,返回1,因为1没有右节点,因此直接存入1的右节点
o2 = 1
o1 = 10,比o2大,返回1
而后进入o2的右节点,右节点有个参数2,再和2比较,返回1,因为2没有右节点,因此直接存入2的右节点
o2 = 2
o1 = 7,比o2大,返回1,而后和10比较,返回-1,因为2没有左节点,因此直接存放存入2的左节点
o2 = 2
o1 = 0,比o1小,返回-1,而后和2的父层节点1比较,返回-1,因为1没有左节点,因此直接存放1的左节点
o2 = 2
o1 = 3,比o2大,返回1,而后和2的右节点10比较,返回1,再和2的左节点7比较,返回-1,因为7下面已经没有其他节点了,因此直接存放
因此最后的输出结果是0,1,2,3,7,10;按照先左后右,先子节点后父节点的规则读取。