本文主要是介绍Java集合框架篇-62-TreeSet比较器排序原理和代码实现,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
前面我们知道了,TreeSet集合在存储元素的时候,会先调用compareTo方法,然后决定如何存储。下面,先看看一个需求的实现过程,然后介绍什么是比较器和如何具体实现。
1.字符串按照长度排序
我们已经知道,默认是按照unicode码表的字典排序输出,例如下面TreeSet集合存储的是String对象,就按照字典a到z的顺序排序。
package treeset;import java.util.TreeSet;public class Demo4_TreeSet {public static void main(String[] args) {TreeSet<String> ts = new TreeSet<>();ts.add("aaaaa");ts.add("z");ts.add("nba");ts.add("cba");ts.add("beijing");System.out.println(ts);}}
运行输出:
[aaaaa, beijing, cba, nba, z]
运行结果并没有按照字符串长度进行排序显示,而是按照字典排序,说明在String类里的compareTo()方法是根据字典排序。接下来我们介绍TreeSet中第二个排序方法,叫比较器排序。打开TreeSet API文档,找到TreeSet(Comparator<? super E> comparator)
这个构造方法,参数就是传一个比较器,搜索Comparator,发现是一个接口,下面有两个方法,一个是compareTo和equals方法。下面我们需要新建一个类来实现Comparator接口,并设计按照字符串长度排序。
package treeset;import java.util.Comparator;
import java.util.TreeSet;public class Demo4_TreeSet {public static void main(String[] args) {TreeSet<String> ts = new TreeSet<>(new CompareByLength());ts.add("aaaaa");ts.add("z");ts.add("nba");ts.add("cba");ts.add("beijing");System.out.println(ts);}}class CompareByLength implements Comparator<String> {@Overridepublic int compare(String s1, String s2) {int num = s1.length() - s2.length(); //长度比较是主要条件return num==0? s1.compareTo(s2) : num; //如果长度一样,然后再按照字典排序}}
运行结果:
[z, cba, nba, aaaaa, beijing]
输出结果体现了先按字符串长度比较,然后按照字符的字典顺序排序。CompareByLength就是一个我们自定义的比较器,作为类型是String类型,里面重写了compare(ob1, ob2)。
2.TreeSet的原理
TreeSet是通过以下两种排序方式保证元素的唯一性。
1.自然排序(字典顺序)
1)TreeSet类的add()方法中会把存入的对象提升为Comparable类型
2)调用对象的compareTo()方法和集合已有对象比较
3)根据compareTo()方法返回的结果进行存储
2.比较器排序
1)创建TreeSet的时候可以指定一个comparator(比较器)
2)如果传入了Comparator的子类对象,那么TreeSet就会按照比较器中的顺序排序
3)add()方法内部会自动调用Comparator接口中compare()方法来排序
两种的区别:
1)TreeSet构造函数什么都不传,默认按照Comparable的顺序
2)TreeSet如果传入Comparator,就优先按照比较器排序
这篇关于Java集合框架篇-62-TreeSet比较器排序原理和代码实现的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!