使用TreeSet,关于重写compareTo()方法 升序降序问题(适合Java小白)

首先,我们要了解TreeSet

点击 TreeSet 的源码我们可以看到 TreeSet 的底层是 TreeMap
而 TreeMap 的底层是基于红黑树实现的,那么什么是红黑树呢?
那我们就不得不提到二叉树,因为红黑树是二叉树的升级版

二叉树

直接举一个栗子吧
开始我们只有一个 50
在这里插入图片描述
然后我们要插入一个元素 30 ,因为比 50 小,插入左边
在这里插入图片描述
然后我们要插入一个元素 60 ,因为比 50 大,所以要插入右边
在这里插入图片描述
然后我们要插入一个元素 50,因为已有了 50,所以不必再插入,同样现在有30和40,若要再插入元素 30 和 40,也是如此的

然后我们要插入一个元素 55,则 55 会先与 50 比较,比 50 大来到右边,发现右边已有元素60,然后 55 会与 60 比较,小就在 60 左边,大就在 60 右边
在这里插入图片描述
以此类推

那么会出现一种情况:
来的元素一直都比 50 大,结果全都在 50 右边,影响了效率
在这里插入图片描述

这个时候,红黑树来了

红黑树

红黑树就是二叉树的升级版
在基于二叉树的基础上,插入元素红黑树会随机生成两种不同的标记
例如:插入的元素有黑色和蓝色(注:这种随机是根据哈希计算的)
在这里插入图片描述
发现右边有是三个黑块,所以意识到了不对劲,就会重新选取中间的节点,比如 58
在这里插入图片描述
这样就效率高了不少

compareTo方法返回

为什么返回值为 0 ,只是存取一个元素,返回 -1 就会倒序存储,返回 1 就会怎么存怎么取呢?

因为TreeSet 底层是一个二叉树(红黑树就是二叉树的升级版嘛),没插入一个元素就会调用 comapreTo() 方法做比较,就会按照我们上面提到的插入规则来

返回值如果写死 0,认为是相同的元素,所以 TreeSet 中不用插入新元素,用原来的即可
返回值如果写死 1,认为是大的元素,所以会插入到已有元素的右边,读取的时候,就是正序排列的
返回值如果写死 -1,认为是小的元素,所以会插入到已有元素的左边,读取的时候,就是倒序排列的

而下面的情况

@Override
public int compareTo(Object o) { 
	Book book = (Book) o; 
	return (int) (this.getBookPrice() - book.getBookPrice());
}

这个为什么是降序排列呢?
这个和上面说的红黑树的原理不同,这里采用的是比较器中的方法
看这篇文章

为什么
return (int) (book.getBookPrice() - this.getBookPrice());等价于
return this.getBookPrice()>book.getBookPrice()?-1:1
查看此处

  • 6
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Java中,TreeSet是一个基于红黑树实现的有序集合。当我们向TreeSet中添加元素时,它会自动根据元素的自然顺序进行排序。 在这个过程中,TreeSet会调用元素类的compareTo()方法来确定元素的顺序。因此,如果我们需要自定义元素的排序方式,就需要重写方法。 以下是一个重写compareTo()方法的示例: ```java public class Student implements Comparable<Student> { private String name; private int age; public Student(String name, int age) { this.name = name; this.age = age; } // 重写compareTo方法 @Override public int compareTo(Student o) { if (this.age > o.age) { return 1; } else if (this.age < o.age) { return -1; } else { return this.name.compareTo(o.name); } } } ``` 在这个例子中,我们创建了一个Student类,它包含一个name属性和一个age属性。我们在这个类中实现了Comparable接口,并重写compareTo方法,按照年龄升序排序,如果年龄相同则按照姓名升序排序。 当我们将Student对象添加到TreeSet中时,会根据重写后的compareTo方法进行排序。 ```java TreeSet<Student> set = new TreeSet<>(); set.add(new Student("Tom", 21)); set.add(new Student("Jerry", 19)); set.add(new Student("Alice", 20)); for (Student s : set) { System.out.println(s.getName() + " " + s.getAge()); } ``` 输出结果为: ``` Jerry 19 Alice 20 Tom 21 ``` 注意:如果我们在重写compareTo方法时只考虑年龄,而不考虑姓名,则会出现相同年龄的Student对象无法添加到TreeSet中的问题。因此,在重写compareTo方法时,需要考虑到所有可能影响元素顺序的因素。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值