自定义类的对象作为TreeSet元素的两种方法排序浅谈

来源:互联网 发布:网络卫星电视机顶盒 编辑:程序博客网 时间:2024/06/06 05:48

我们知道TreeSet是Set接口下的一个实现类,TreeSet中的元素的排列顺序是和添加的顺序是没有关系的,并且它里面元素也是不能重复的
但是,TreeSet集合有它独有的特点,比如:

import java.util.*;public class Test{    public static void main(String[] args){        TreeSet ts = new TreeSet();        ts.add("x");        ts.add("a");        ts.add("c");        ts.add("j");        System.out.println(ts);    }}

看输出结果:

这里写图片描述

集合里元素的排序,虽然和你添加的顺序无关,但它会根据元素的unicode自然排序

但是,当你给集合中添加不同类型的数据时:

import java.util.*;public class Test{    public static void main(String[] args){        TreeSet ts = new TreeSet();        ts.add("x");        ts.add("a");        ts.add("c");        ts.add(1);        System.out.println(ts);    }}

出现的结果是:

这里写图片描述

就会报ClassCastException 类型转换异常
String 不能转成Integer类型,要排序,类型必须一致
说了这么多,TreeSet的排序好像就这么多了??
当然不是了,再往下看:
如果现在给集合里添加自定义类的对象怎么办?它又是怎么排序的呢?

import java.util.*;class Person{    private String name ;    private int age;    public Person(String name,int age){        this.name = name;        this.age = age;    }    public String toString(){        return "Person[name="+name+",age="+age+"]";    }}public class Test{    public static void main(String[] args){        TreeSet ts = new TreeSet();        ts.add(new Person("ab",20));        ts.add(new Person("ac",10));        ts.add(new Person("aa",20));        ts.add(new Person("ad",50));        System.out.println(ts);    }}

再看运行结果:

这里写图片描述

可以看出,给出的异常信息是:Person类不能比较,为什么呢?
它给的异常信息指示是应该去实现Comparable接口,一般系统内置的类都已经实现了Comparable接口,像上面的String类,Integer类

不卖关子了,说一下,自定义类的对象作为TreeSet元素的两种方法排序
第一种:让自定义的类实现Comparable接口,并实现里面的compareTo()抽象方法,此时自定义的类的对象就有了可比性
第二种:当自定义的类没有实现Comparable接口时,这时自定义的类的对象也就不具备可比性了,这时需要让集合自身具备可比性,在集合初始化时,可以指明集合的比较方式,即在new TreeSet()时,在里面写一个匿名内部类,该匿名内部类实现Comparator接口,再重写它的compare()方法,具体看代码

先看第一种:
看下面的代码,排序规则是自定义的,先按照年龄排序,在年龄相同的情况下,再按名字排序

import java.util.*;class Person implements Comparable{    private String name ;    private int age;    public Person(String name,int age){        this.name = name;        this.age = age;    }    public String toString(){        return "Person[name="+name+",age="+age+"]";    }    // compareTo()方法是一个回调方法!    // 返回值有3种情况:    // 1. 返回大于0的数字,表示传入的参数比当前对象大    // 2. 返回小于0的数字,表示传入的参数比当前对象小    // 3. 返回0,表示当前对象等于传入的参数    public int compareTo(Object obj){        if(!(obj instanceof Person)){            //这里先判断Obj是不是Person的实例,如果不是直接抛异常            //不用再进行下面的比较            throw new ClassCastException("这不是人");        }        Person p = (Person)obj;        int n = this.age - p.age;        //先按年龄比,年龄相同时,再按名字比        if(n == 0){            return this.name.compareTo(p.name);        }        return n;    }}public class Test{    public static void main(String[] args){        Set set = new TreeSet();        set.add(new Person("ab",20));        set.add(new Person("ac",10));        set.add(new Person("aa",20));        set.add(new Person("ad",50));        System.out.println(set);    }}

看输出结果:

这里写图片描述

看中间两项,当年龄相同,都是20,此时,两者再按照名字排序

第二种:
此时,不实现Comparable接口,在new TreeSet()时,在里面写一个匿名内部类,该匿名内部类实现Comparator接口,再重写它的compare()方法

import java.util.*;class Person{    private String name ;    private int age;    public Person(String name,int age){        this.name = name;        this.age = age;    }    public String toString(){        return "Person[name="+name+",age="+age+"]";    }    public int getAge(){        return age;    }    public String getName(){        return name;    }}public class Test{    public static void main(String[] args){        Set set = new TreeSet(new Comparator(){            //这里的compare()传两个参数            //第一种方法里的compareTo()传一个参数            public int compare(Object obj1,Object obj2){                if(!((obj1 instanceof Person)&&(obj1 instanceof Person))){                    throw new ClassCastException("这不是人");                }                Person p1 = (Person)obj1;                Person p2 = (Person)obj2;                System.out.println(p1 + "vs" + p2);                return p1.getAge()-p2.getAge();            }        });        set.add(new Person("ab",20));        set.add(new Person("ac",10));        set.add(new Person("aa",20));        System.out.println(set);    }}

看输出结果:

这里写图片描述
最后一行是最终结果,前面三行是比较过程
这里只按照年龄排序,没有再按名字排序,所以最后输出的set只有两项,第三项和第一项重复,直接被过滤掉

注意一点,当两种方法都存在时,到底以哪种方法为主呢?
答案是,以第二种方法,以比较器comparator()为主
我们可以验证一下:看下面,把两个方法都放进去

import java.util.*;class Person implements Comparable{    //这里实现了Comparable接口    private String name ;    private int age;    public Person(String name,int age){        this.name = name;        this.age = age;    }    public String toString(){        return "Person[name="+name+",age="+age+"]";    }    public int getAge(){        return age;    }    public String getName(){        return name;    }    public int compareTo(Object obj){        if(!(obj instanceof Person)){            throw new ClassCastException("这不是人");        }        Person p = (Person)obj;        int n = this.age - p.age;        if(n == 0){            return this.name.compareTo(p.name);        }        return n;    }}public class Test{    public static void main(String[] args){        Set set = new TreeSet(new Comparator(){            //这里也实现了Comparator接口            //那么以哪种为主呢?            //看最下面的结果            public int compare(Object obj1,Object obj2){                if(!((obj1 instanceof Person)&&(obj1 instanceof Person))){                    throw new ClassCastException("这不是人");                }                Person p1 = (Person)obj1;                Person p2 = (Person)obj2;                return p1.getAge()-p2.getAge();            }        });        set.add(new Person("ab",20));        set.add(new Person("ac",10));        set.add(new Person("aa",20));        System.out.println(set);    }}

如果输出的集合中有三项,则说明以第一种为主;
如果输出的集合中有两项,则说明以第二种为主

最终结果为:

这里写图片描述

输出的只有两项,说明了:当两种排序方式都存在时,以比较器为主

其实第一种方式是有安全隐患的,后续再来完善…..

0 0