TreeSet和TreeMap的输出

如果加入TreeSet和TreeMap的元素没有实现comprable中的compareTo()方法，那么会报错“treeset cannot be cast to java.lang.Comparable”.

要解决这个问题有两种方法：

（1）让元素自身具有比较性；可以实现Comparable接口，实现compareTo()方法；

（2）让集合自身具有比较性；可以定义比较器，即让集合实现Comparator接口，然后实现compare()方法；

方法一：

因此需要对元素类型实现Comparable借口，并实现compareTo()方法，即告诉TreeSet和TreeMap你要怎么排列，它才能进行排序。

基础类：

package niukewang;

import java.util.Objects;

public class setClass implements Comparable<setClass>{

    String a;
    String b;
    public setClass(String a, String b)
    {
        this.a=a;
        this.b=b;
    }

    public int hashCode() {
        return a.hashCode();
    }

    public boolean equals(Object obj)
    {
        if(obj==null) return false;

        if(getClass() != obj.getClass()) return false;

        final setClass s=(setClass)obj;

        return Objects.equals(this.a, s.a);
    }

    @Override
    public int compareTo(setClass o) {
        int num=this.b.compareTo(o.b);
        return num;
    }
}

测试类：

package niukewang;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Hashtable;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.TreeMap;
import java.util.TreeSet;
import java.util.Vector;

public class test1 {

    public static void main(String args[])
    {
        setClass s1=new setClass("http://www.yjbys.com/", "same0");
        setClass s2=new setClass("http://www.yjbys.com1/", "same0");
        setClass s3=new setClass("http://www.yjbys.com2/", "same2");
        setClass s4=new setClass("http://www.yjbys.com2/", "same3");

        //hasSet
        System.out.println("hasSet......");
        Set<setClass> set=new HashSet<setClass>();
        set.add(s1);
        set.add(s2);
        set.add(s3);
        set.add(s4);
        Iterator it= set.iterator();
        while(it.hasNext())
        {
            setClass ob=(setClass)it.next();
            System.out.println(ob.a);
        }

        //TreeSet
        System.out.println("\nTreeSet........");
        Set<setClass> tree=new TreeSet<setClass>();
        tree.add(s1);
        tree.add(s2);
        tree.add(s3);
        tree.add(s4);
        Iterator treeit=tree.iterator();
        while(treeit.hasNext())
        {
            setClass ob=(setClass) treeit.next();
            System.out.println(ob.a);
        }

        //TreeMap
        System.out.println("\nTreeMap.........");
        Map<setClass,String> treemap=new TreeMap<setClass,String>();
        treemap.put(s1, "TreeMap one");
        treemap.put(s2, "TreeMap two");
        treemap.put(s3, "TreeMap three");
        treemap.put(s4, "TreeMap four");
        for(Map.Entry<setClass, String> entry: treemap.entrySet())
        {
            System.out.println("The treemap key is "+entry.getKey().a+" The value is "+ entry.getValue());
        }

        //HasMap
        System.out.println("\nHashMap......");
        Map<setClass,String> hashmap=new HashMap<setClass, String>();
        hashmap.put(s1, "HashMap one");
        hashmap.put(s2, "HashMap two");
        hashmap.put(s3, "HashMap three");
        hashmap.put(s4, "HashMap four");
        for(Map.Entry<setClass, String> entry : hashmap.entrySet())
        {
            System.out.println("The key is "+entry.getKey().a+" The value is "+entry.getValue());
        }

        //HasTable
        System.out.println("\nHashTable......");
        Map<setClass,String> hashtable=new Hashtable<setClass, String>();
        hashtable.put(s1, "HashTable one");
        hashtable.put(s2, "HashTable two");
        hashtable.put(s3, "HashTable three");
        hashtable.put(s4, "HashTable four");
        for(Map.Entry<setClass, String> entry: hashtable.entrySet())
        {
            System.out.println("The HashTable key is "+entry.getKey().a+" The HashTable value is "+entry.getValue());
        }

        //LinkedList
        System.out.println("\nLinkedList.....");
        List list=new LinkedList();
        list.add(s1);
        list.add(s2);
        list.add(s3);
        list.add(s4);
        Iterator listit=list.iterator();
        while(listit.hasNext())
        {
            setClass ob=(setClass)listit.next();
            System.out.println(ob.a);
        }

        //ArrayList
        System.out.println("\nArrayList.....");
        List array=new ArrayList();
        array.add(s1);
        array.add(s2);
        array.add(s3);
        array.add(s4);
        Iterator arrayit=array.iterator();
        while(arrayit.hasNext())
        {
            setClass ob=(setClass)arrayit.next();
            System.out.println(ob.a);
        }

        //vector
        System.out.println("\nvector.....");
        Vector v=new Vector();
        v.add(s1);
        v.add(s2);
        v.add(s3);
        v.add(s4);
        Iterator vit=v.iterator();
        while(vit.hasNext())
        {
            setClass  ob=(setClass)vit.next();
            System.out.println(ob.a);
        }

    }
}

输出结果：

hasSet......
http://www.yjbys.com1/
http://www.yjbys.com/
http://www.yjbys.com2/

TreeSet........
http://www.yjbys.com/
http://www.yjbys.com2/
http://www.yjbys.com2/

TreeMap.........
The treemap key is http://www.yjbys.com/ The value is TreeMap two
The treemap key is http://www.yjbys.com2/ The value is TreeMap three
The treemap key is http://www.yjbys.com2/ The value is TreeMap four

HashMap......
The key is http://www.yjbys.com1/ The value is HashMap two
The key is http://www.yjbys.com/ The value is HashMap one
The key is http://www.yjbys.com2/ The value is HashMap four

HashTable......
The HashTable key is http://www.yjbys.com1/ The HashTable value is HashTable two
The HashTable key is http://www.yjbys.com/ The HashTable value is HashTable one
The HashTable key is http://www.yjbys.com2/ The HashTable value is HashTable four

LinkedList.....
http://www.yjbys.com/
http://www.yjbys.com1/
http://www.yjbys.com2/
http://www.yjbys.com2/

ArrayList.....
http://www.yjbys.com/
http://www.yjbys.com1/
http://www.yjbys.com2/
http://www.yjbys.com2/

vector.....
http://www.yjbys.com/
http://www.yjbys.com1/
http://www.yjbys.com2/
http://www.yjbys.com2/

TreeSet并不像hashSet那样，用hashCode和equals取决于元素是否相等。
而是取决于compareTo()方法中指定的排序的方式；

setClass s1=new setClass("http://www.yjbys.com/", "same0");
  setClass s2=new setClass("http://www.yjbys.com1/", "same0");
  setClass s3=new setClass("http://www.yjbys.com2/", "same2");
  setClass s4=new setClass("http://www.yjbys.com2/", "same3");

如果用hashCode和equals取决于元素是否相等，那么s3和s4是相等的；

而此时候，是根据compareTo()方法中的第二个字符串b来判断是否相等，则s1与s2相等，s3和s4不相等，所以才有这样的输出：

TreeSet........
http://www.yjbys.com/
http://www.yjbys.com2/
http://www.yjbys.com2/

方法二：

定义比较器，让集合实现Comparator接口，然后实现compare方法，则集合自身具有比较性。

其中Class Student 是实现了Comparable接口的，是元素本身具有比较性，在main方法中添加到TreeSet中的时候，能体现出来，通过年龄来比较大小；如果年龄和姓名相同，则是完全相等；

其中成员方法testComparator()中，定义了一个TreeSet，使其实现Comparator接口，并实现了方法compare()，通过setClass类中的a的属性的大小；因此此方法中的TreeSet具有比较性；

package niukewang;

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

import org.junit.Test;

//非线程安全
public class TreeSetTest {
    /**
     * TreeSet:它可以给Set集合中的元素进行指定方式的排序。 保证元素唯一性的方式：通过比较的结果是否为0. 底层数据结构是：二叉树。
     *
     * 排序的第一种方式： 让元素自身具备比较性。只要让元素实现Comparable接口，覆盖compareTo方法即可。
     *
     * 但是，如果元素自身不具备比较性，或者元素自身具备的比较性，不是所需要的。
     * 比如，学生的自然排序是按年龄排序，现在想要按照学生姓名排序。还可以不改动原有代码。 这时怎么办呢？
     *
     * 排序的第二种方式：自定比较器的方式。这时可以让集合自身具备比较性。
     *
     * 可以定义一个类实现Comparator接口，覆盖compare方法。
     * 将该Comparator接口子类对象作为实际参数传递给TreeSet集合构造函数。
     */
    public static void main(String[] args) {
        TreeSet<Student> ts = new TreeSet<Student>();
        // Student如果不实现Comparable，add时会出错，Student cannot be cast to java.lang.Comparable
        //添加第一个对象时，TreeSet里没有任何元素，所以没有问题；
        //当添加第二个Err对象时，TreeSet就会调用该对象的compareTo(Object obj)方法与集合中其他元素进行比较——
        //如果对应的类没有实现Comparable接口，则会引发ClassCastException异常
        // add方法内部会对插入的数据进行排序，此时元素自身具备比较性，因为其实现了Comparable接口
        ts.add(new Student("lisi0", 30));
        ts.add(new Student("lisixx", 29));
        ts.add(new Student("lisi9", 29));
        ts.add(new Student("lisi8", 38));
        // 重复插入无效，但是不报错（根据年龄和名字进行比较，都相同，视为同一个学生）
        ts.add(new Student("lisixx", 29));
        ts.add(new Student("lisixx", 28));
        ts.add(new Student("lisi4", 14));
        ts.add(new Student("lisi7", 27));
        System.out.println(ts);

        System.out.println("testComparator........\n");
        TreeSetTest ob=new TreeSetTest();
        ob.testComprator();
    }

    //自定义比较器，让集合自身具备比较性
    @Test
    public void testComprator()
    {
        setClass s1=new setClass("http://www.yjbys.com/", "same0");
        setClass s2=new setClass("http://www.yjbys.com1/", "same0");
        setClass s3=new setClass("http://www.yjbys.com2/", "same2");
        setClass s4=new setClass("http://www.yjbys.com2/", "same3");

        Set<setClass> treeset=new TreeSet<setClass>(new Comparator<setClass>(){

            @Override
            public int compare(setClass arg0, setClass arg1) {
                // TODO Auto-generated method stub
                int num=arg0.a.compareTo(arg1.a); //只比较属性a的大小
                return num;
            }
        });

        treeset.add(s1);
        treeset.add(s2);
        treeset.add(s3);
        treeset.add(s4);

        System.out.println(treeset);
    }

    //同姓名同年龄的学生视为同一个学生
    public static class Student implements Comparable<Student> {

        private int age;
        private String name;

        public Student(String name, int age) {
            this.age = age;
            this.name = name;
        }

        @Override
        public int compareTo(Student stu) {
            int num = new Integer(this.age).compareTo(new Integer(stu.age));
            return num == 0 ? this.name.compareTo(stu.name) : num;
        }

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

        public int getAge() {
            return age;
        }

        public void setAge(int age) {
            this.age = age;
        }

        public String getName() {
            return name;
        }

        public void setName(String name) {
            this.name = name;
        }
    }
}

运行结果：

[lisi4::14, lisi7::27, lisixx::28, lisi9::29, lisixx::29, lisi0::30, lisi8::38]
testComparator........

[http://www.yjbys.com/ :: same0, http://www.yjbys.com1/ :: same0, http://www.yjbys.com2/ :: same2]

由上图可知：
Stuent类是通过年龄判断的，所以按年龄排序；只有姓名和年龄都相同时，则认为是相同的，所以ts.add(new Student("lisixx", 29))不会重复添加；其他因为姓名不同，就算年龄相同，也认为是不相同的；

setClass类，是根据setClass中的属性a的值是否相等,就算hashCode和equals方法被重写了，它只看比较性。

注意：

如果在setClass类中，如果没有重写public String toString()方法，则会输出类似下面的输出：

[niukewang.setClass@cbf33da, niukewang.setClass@8b2747d3, niukewang.setClass@8b2747f2]

但是如果重写了public String toString()方法：

public String toString()
 {
  return this.a+" :: "+this.b;
 }

则会输出：[http://www.yjbys.com/ :: same0, http://www.yjbys.com1/ :: same0, http://www.yjbys.com2/ :: same2]