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> {

	@Override
	public 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，就优先按照比较器排序