TreeSet自定义比较

Java中的TreeSet是Set的一个子类，TreeSet集合是用来对象元素进行排序的他可以保证元素的唯一。
TreeSet需要在自定义类(Person)中实现Comparable接口，并重写接口中的compareTo方法

public class Person implements Comparable {
private String name;
private int age;
…
public int compareTo(Person o) {
return 0; //当compareTo方法返回0的时候集合中只有一个元素
return 1; //当compareTo方法返回正数的时候集合会怎么存就怎么取
return -1; //当compareTo方法返回负数的时候集合会倒序存储 }
}

为什么返回0，只会存一个元素，返回-1会倒序存储，返回1会怎么存就怎么取呢？原因在于TreeSet底层其实是一个二叉树机构，且每插入一个新元素(第一个除外)都会调用compareTo()方法去和上一个插入的元素作比较，并按二叉树的结构进行排列。

1. 如果将compareTo()返回值写死为0，元素值每次比较，都认为是相同的元素，这时就不再向TreeSet中插入除第一个外的新元素。所以TreeSet中就只存在插入的第一个元素。
2. 如果将compareTo()返回值写死为1，元素值每次比较，都认为新插入的元素比上一个元素大，于是二叉树存储时，会存在根的右侧，读取时就是正序排列的。
3. 如果将compareTo()返回值写死为-1，元素值每次比较，都认为新插入的元素比上一个元素小，于是二叉树存储时，会存在根的左侧，读取时就是倒序序排列的。

利用上述规则，我们就可以按照年龄来排序了。代码如图
public int compareTo(Person o) {
int num = this.age - o.age; //年龄是比较的主要条件
return num == 0 ? this.name.compareTo(o.name) : num;//姓名是比较的次要条件
}

按照姓名排序(依据Unicode编码大小),代码如下：
public int compareTo(Person o) {
int num = this.name.compareTo(o.name); //姓名是主要条件
return num == 0 ? this.age - o.age : num; //年龄是次要条件
}

按照姓名长度排序,代码如下：
public int compareTo(Person o) {
int length = this.name.length() - o.name.length(); //比较长度为主要条件
int num = length == 0 ? this.name.compareTo(o.name) : length; //比较内容为次要条件
return num == 0 ? this.age - o.age : num; //比较年龄为次要条件
}

以上是TreeSet如何比较自定义对象，接下来我们再来看一下TreeSet如何利用比较器比较元素。

需求:现在要制定TreeSet中按照String长度比较String。

//首先定义一个类，实现Comparator接口，并重写compare()方法，class CompareByLen /extends Object/ implements Comparator {

public int compare(String s1, String s2) {       
    //按照字符串的长度比较
    int num = s1.length() - s2.length();        
    //长度为主要条件
    return num == 0 ? s1.compareTo(s2) : num;    
    //内容为次要条件    }

}

public static void demo4() {

    //需求:将字符串按照长度排序
    TreeSet<String> ts = new TreeSet<>(new CompareByLen());        
    //Comparator c = new CompareByLen();
    ts.add("aaaaaaaa");
    ts.add("z");
    ts.add("wc");
    ts.add("nba");
    ts.add("cba");
    
    System.out.println(ts);
}