使用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
查看此处

Guess you like

Origin blog.csdn.net/qq_45895520/article/details/120029330