TreeSet排序compareTo、Comparator

Set：元素不可以重复，是无序。

Set接口中的方法和Collection一致

|----HashSet：内部数据结构是哈希表，是不同步的。
如何保证该集合的元素唯一性呢？
是通过对象的hashCode和equals方法来完成对象的唯一性的。如果对象的hashCode值不同，那么不用判断equals方法，就直接存储到哈希表中；如果对象的hashCode值相同，那么要再次判断对象的equals方法是否为true，如果为true，视为相同元素，不存，如果为false，那么视为不同元素，就进行存储。
记住：如果元素要存储到HashSet集合中，必须覆盖hashCode方法和equals方法；一般情况下，如果定义的类会产生很多对象，比如人，学生，书，通常都需要覆盖equals和hashCode方法，建立对象判断是否相同的依据

|----TreeSet：可以对Set集合中的元素进行排序，是不同步的。
判断元素唯一性的方式：就是根据比较方法的返回结果是否为0，是0，就是相同元素，就存。

TreeSet对元素进行排序的方式一：compareTo方法
让元素自身具备比较功能，元素就需要comparable接口，覆盖compareTo方法。
但是如果不要按照对象中具备的自然顺序进行排序，即对象中不具备自然顺序（更简明地来说说，对象是别人的，我们只能调用，不能修改）怎么办？

即使用TreeSet集合的第二种排序方式：Comparator比较器
让集合自身具备比较功能，即在集合对象创建时，由此想到TreeSet的构造方法,

TreeSet(Comparator comparator)

为此需要定义一个类实现Comparator接口，覆盖compare方法，将该类对象作为参数传递给TreeSet集合的构造函数。

1.如下代码，TreeSet进行迭代会出现问题，无法转换类

package com.vv.treehset.demo;

import java.util.Iterator;
import java.util.TreeSet;

import com.vv.p.bean.Person;

public class TreeSetDemo {

    public static void main(String[] args) {
        // TODO Auto-generated method stub
        TreeSet ts = new TreeSet();
        ts.add(new Person("zhangsan", 20));
        ts.add(new Person("lisi", 22));
        ts.add(new Person("wangwu", 21));
        ts.add(new Person("zhouqi", 29));
        ts.add(new Person("zhaoliu", 28));

        Iterator it = ts.iterator();
        while (it.hasNext()) {
            Person p = (Person) it.next();
            System.out.println(p.getName() + ":" + p.getAge());
        }
    }

}

结果如下：

Exception in thread "main" java.lang.ClassCastException: com.vv.p.bean.Person cannot be cast to java.lang.Comparable
    at java.util.TreeMap.compare(TreeMap.java:1294)
    at java.util.TreeMap.put(TreeMap.java:538)
    at java.util.TreeSet.add(TreeSet.java:255)
    at com.vv.treehset.demo.TreeSetDemo.main(TreeSetDemo.java:13)

java.lang.Comparable即是出错的包，故在API文档中查出错问题

1.对此接口的描述为：此接口强行对实现它的每个类的对象进行整体排序。这种排序被称为类的自然排序，类的 compareTo 方法被称为它的自然比较方法。

2.故此查看compareTo方法：即比较此对象与指定对象的顺序。如果该对象小于、等于或大于指定对象，则分别返回负整数、零或正整数。

同时由于对象是无法进行比较的，因此我们定义其按照Person的年龄按照从小到大进行比较，故需要在compareTo方法中添加我们需要的程序，（注意：TreeSet中的排序和hashCode完全没有关系）

修改代码如下：按照年龄进行排序

package com.vv.p.bean;
//alt+shift+s使用快捷键set等方法
public class Person implements Comparable{

    private String name;
    private int age;
    //输出person的结果
    @Override
    public String toString() {
        // TODO Auto-generated method stub
        return name+":" + age;
    }
        public String getName() {
        return name;
    }
    public void setName(String name) {
        this.name = name;
    }
    public int getAge() {
        return age;
    }
    public void setAge(int age) {
        this.age = age;
    }
    //一创建对象就初始化姓名和年龄
    public Person(String name, int age) {
        super();
        this.name = name;
        this.age = age;
    }
    @Override
    public int compareTo(Object o) {
        Person p = (Person) o;
        //优化后的代码

        //按照年龄进行排序
//      int temp = this.age - p.age;
//      return temp == 0? this.name.compareTo(p.name):temp;

        //      按照姓名进行排序
        int temp = this.name.compareTo(p.name);
        return temp == 0 ? this.age - p.age : temp;
        /**
        if(this.age > p.age){
            return 1;
        }else if(this.age < p.age){
            return -1;
        }else{
            return this.name.compareTo(p.name);
        }
        */  }

}

TreeSet排序方式二：

package com.vv.treehset.demo;

import java.util.Comparator;

import com.vv.p.bean.Person;

public class ComparatorByName implements Comparator {

    //根据姓名进行排序
    @Override
    public int compare(Object o1, Object o2) {
        Person p1 = (Person) o1;
        Person p2 = (Person) o2;

        int temp = p1.getName().compareTo(p2.getName());
        return temp == 0 ? p1.getAge() - p2.getAge() : temp;
        
    }

}

TreeSetDemo只需要做如下修改，而且Person中不需要写CompareTo方法（第一种排序方式）即可：

TreeSet ts = new TreeSet(new ComparatorByName());

TreeSet集合的排序方式：二叉树

image.png

TreeSet集合是自动对字符串进行排序的

package com.vv.treehset.demo;

import java.util.Iterator;
import java.util.TreeSet;

import com.vv.p.bean.Person;

public class TreeSetDemo1 {

    public static void main(String[] args) {
        // TODO Auto-generated method stub
        TreeSet ts = new TreeSet();
        ts.add("aaaa");
        ts.add("scsdf");
        ts.add("efef");
        ts.add("rtrg");
        Iterator it = ts.iterator();
        while (it.hasNext()) {
            System.out.println(it.next());
        }
        
    }

}

结果如下：

aaaa
efef
rtrg
scsdf

TreeSet练习：对字符串长度进行排序

创建TreeSet对象的构造方法：

package com.vv.treehset.demo;

import java.util.Iterator;
import java.util.TreeSet;

import com.vv.comparator.ComparatorByLength;
import com.vv.p.bean.Person;

public class TreeSetDemo1 {

    public static void main(String[] args) {
        // TODO Auto-generated method stub
        TreeSet ts = new TreeSet(new ComparatorByLength());
        ts.add("aaaasdsad");
        ts.add("scsdf");
        ts.add("efefsa");
        ts.add("afefsa");
        ts.add("rtrg");
        Iterator it = ts.iterator();
        while (it.hasNext()) {
            System.out.println(it.next());
        }
        
    }

}

长度比较的方法：

package com.vv.comparator;

import java.util.Comparator;

public class ComparatorByLength implements Comparator {

    @Override
    public int compare(Object o1, Object o2) {

        String s1 = (String) o1;
        String s2 = (String) o2;

        int temp = s1.length() - s2.length();
        
        return temp == 0 ? s1.compareTo(s2) : temp;
    }

}