TreeSet

TreeSet实现了SortedSet接口，能够对集合中的对象进行排序 。

它支持两种排序方式：

一、自然排序

      使用自然排序时，向TreeSet集合里加入的对象都要实现了Comparable接口（里面的compareTo（）方法用于指定排序方式）。

package test.java;

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

/**
 * Created by Lenovo on 2015/2/10.
 */
public class TreeSetTest implements Comparable {


    public String name;
    public int age;

    public TreeSetTest(String name, int age) {    //构造方法没有返回值
        this.name = name;
        this.age = age;
    }
    public String getName(){
        return this.name;
    }
    public int getAge(){
        return this.age;
    }
    public boolean equals(Object o) {
        if (this == o) return true;
        if (!(o instanceof Customer)) {
            return false;
        }
        TreeSetTest other = (TreeSetTest) o;  // 记得！！！
        return (this.name.equals(other.getName()) && this.getAge() == other.age);

    }

    public int hashCode() {
        int result;
        result = (name == null) ? 0 : name.hashCode();
        result = 29 * result + age;
        return result;
    }

    @Override
    public int compareTo(Object o) {

        TreeSetTest other = (TreeSetTest) o;
        if (this.name.compareTo(other.name) > 0)
            return 1;
        if (this.name.compareTo(other.name) < 0)
            return -1;
        if(this.age>other.age)
            return 1;
        if(this.age<other.age)
            return -1;
        return 0;
    }

    public static void main(String[] args) {
        Set<TreeSetTest> set=new TreeSet<TreeSetTest>();
        TreeSetTest t1=new TreeSetTest("Tom",15);
        TreeSetTest t2=new TreeSetTest("Tom",16);
        TreeSetTest t3=new TreeSetTest("Alice",33);

        set.add(t1);
        set.add(t2);
        set.add(t3);
        Iterator<TreeSetTest> it=set.iterator();
        while(it.hasNext()) {
            TreeSetTest temp = it.next();
            System.out.println(temp.name + "  " + temp.age);
        }
    }
}

        为了保证TreeSetTest能够正确的排序，要求它的compareTo方法和equals方法按相同的规则比较两个对象是否一样，当一个类实现了equals 方法，它的hashCode方法也要实现且保证当两个对象相等时它的哈希码是一样的 。

      对于TreeSet中已经存在的对象，若修改了他们的属性，TreeSet不会对集合重新排序。因此适合用TreeSet进行排序的是不可变类。

 

 

二、客户排序

      客户化排序实现java.util.Comparator<Type> 接口，<Type>指定被排序的对象的类型，里面的compare 方法指定排序的规则。

Customer类：

package test.java;

import java.util.Objects;

public class Customer implements Cloneable{

    private String name;
    private  int age;

    public Customer(){
        this("unknow",0);  //显示的默认构造方法中 第一个语句必须是this语句。

    }

    public Customer(String name,int age){    //构造方法没有返回值
        this.name=name;
        this.age=age;
    }

    public String getName(){
        return this.name;
    }

    public int getAge(){
        return this.age;
    }

    public Object clone()throws CloneNotSupportedException{
        return super.clone();
    }

    public boolean equals(Object o){
        if(this==o) return true;
        if(!(o instanceof Customer)){
            return false;
        }
        Customer other= (Customer ) o;  // 记得！！！
        return (this.name.equals( other.name)&&this.age==other.age);

    }

    public String toString(){return this.name+"  "+this.age ;}


}

 

package test.java;

import java.util.Comparator;
import java.util.Iterator;
import java.util.Set;
import java.util.TreeSet;

/**
 * Created by Lenovo on 2015/2/10.
 */
public class CustomerComparator implements Comparator<Customer> {


    @Override
    public int compare(Customer o1, Customer o2) {
        if (o1.getName().compareTo(o2.getName()) > 0) return -1;
        if (o1.getName().compareTo(o2.getName()) < 0) return 1;

        if(o1.getAge()<o2.getAge())
            return -1;
        if(o1.getAge()>o2.getAge())
            return 1;
        return 0;
    }

    public static void main(String[] args) {
        Set<Customer> set = new TreeSet<Customer>(new CustomerComparator());
        Customer t1 = new Customer("Tom", 15);
        Customer t2 = new Customer("Tom", 16);
        Customer t3 = new Customer("Alice", 33);

        set.add(t1);
        set.add(t2);
        set.add(t3);

        Iterator<Customer> it = set.iterator();
        while (it.hasNext()) {
            Customer temp = it.next();
            System.out.println(temp.getName() + " & " + temp.getAge());
        }
    }
}