踩坑记录:TreeSet自定义排序导致数据丢失变少

当使用自定义Comparable的TreeSet进行排序时,若compareTo方法返回0导致元素重复,会丢失数据。TreeSet内部依赖TreeMap,当比较结果为0时,只会保留第一个元素。解决办法包括避免重复排序或利用Java 8的Stream API进行去重和排序。
摘要由CSDN通过智能技术生成

问题代码

自定义Comparable
Set set = new TreeSet(自定义Comparable);
set.addAll(List); //List原来有5个,结果set只有4个,原因是List的compareTo返回的结果相同,导致后面的item被舍弃掉了。

原理

通过跟踪TreeSet源码发现,TreeSet中使用的TreeMap来存储对象,将TreeSet的元素作为Map的key,继续跟踪TreeMap的put方法源码当比compareTo返回值是0时,key值不会变化(实际是更新key对应的value,TreeSet的value是无意义的),也就是说当出现多个返回0的情况时TreeSet中只会放入第一个元素,后面的全都舍弃掉,源码比较简单,这里就不贴了。
总结:当使用TreeSet排序时,如果实现的compareTo方法中有返回0的情况时就会造成数据丢失。

解决方案:

  • 方法一:不要让有重复的排序出现就可以了,实际业务如果不能避免那就换一个排序方式。
  • 方法二:使用jdk8的list.stream().distinct().sort(排序规则).collect(Collectors.toList());
    Stream#distinct是根据toString来进行去重,可以重写下List里的元素的toString方法。如果sort相同,也不会出现只会放入第一个元素,后面的全都舍弃掉的现象
  • 2
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

我叫985

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值