Java编程之TreeSet排序两种解决方法(1)元素自身具备比较功能,元素需要实现Comparable接口覆盖compare(2)创建根据自定义Person类的name进行排序的Comparator

来源:互联网 发布:windows photoviewer 编辑:程序博客网 时间:2024/06/07 22:27

       当很多人问我读研到底好不好的时候,我总是说上研很苦逼,读完研之后都不知道自己能不能找到工作,所以不建议同学们读研~即使要读也读一个985或者211的研究生,这是我肺腑之言。但还有一半我没说完,读研的时候你可能会找到你喜欢的活动,会遇到一些愿意和你一起玩的玩伴,在读研期间可以很任性的在想玩耍的时候就去玩儿,这是读研的福利,很高兴我能遇到。好了开始技术!

      学习Java,肯定会学到集合,然后遇到TreeSet,我现在就遇到了它。同时使用它的时候遇到一些问题。首先说说什么是TreeSet。TreeSet是Set 的子接口,而Set接口的元素不可重复、无序的特性TreeSet也是具备的。另外,TreeSet还具有可以对集合中的元素进行指定顺序的排序的功能。

     这个功能实现的基本原理就是,当存放元素的时候进行比较,比较之后再存入。对于普通的字符串它可以完美的实现排序功能:

public static void demo1() {TreeSet ts = new TreeSet();ts.add("abc");ts.add("nba");ts.add("linweieran");ts.add("cba");Iterator it = ts.iterator();while(it.hasNext()){System.out.println(it.next());}}
执行结果如下:

运行结果正确。

但当我向TreeSet集合中保存自定义的Person对象是就会发生错误!

public static void main(String[] args) {TreeSet ts = new TreeSet(new CompareByName()) ;ts.add(new Person("linweieran",24));ts.add(new Person("linweieran1",23)); //bean.Person cannot be cast to java.lang.Comparablets.add(new Person("linweieran2",22));ts.add(new Person("weixiaobao",25));ts.add(new Person("linweieran",24));Iterator  it = ts.iterator();while(it.hasNext()){Person p = (Person)it.next();System.out.println(p.getName()+":"+p.getAge());}}
发生这样的错误:


查文档会找到这样的解释:Comparable此接口强行对实现它的每个类的对象进行强行排序,这种排序成为自然排序,类的compareTo 方法就是它的自然比较方法。所以为了实现该功能就有两个方法

      TreeSet对元素进行排序的方式一:让元素自身具备比较功能,元素需要实现Comparable接口,覆盖compareTo()方法;简而言之就是在Person类中继承Comparable接口,并重写compareTo()方法。

public class Person implements Comparable {private String name;private int age;//alt+shift+spublic Person(String name, int age) {super();this.name = name;this.age = age;}/* * 以Person对象的年龄进行从小到大的排序。 */@Overridepublic int compareTo(Object o) {//进行比较,Person p = (Person)o;/*if(this.age>p.age)return 1;if(this.age<p.age)return -1;if(this.age==p.age){//姓名怎么比较-----return this.name.compareTo(p.name);}*/System.out.println(this.name+"---compareTo---");int temp= this.age-p.age;return temp==0?this.name.compareTo(p.name):temp;}
编码思路是:先对年龄进行比较,如果年龄相容再对姓名(String)进行比较,最后可以简约成三行代码!!

这样子就可以实现排序功能。

      但是,使用上述的方式有一个问题,如果Person类不是自己写的,我们就无法在Person类中进行修改实现该功能,如果是这种情况怎么办呢?这就需要另一种方式了!

       TreeSet集合的第二种排序方式:让集合自身具备比较功能,定义一个类实现Comparator接口,覆盖Comparat方法。构造函数TreeSet(Comparator c)的比较器。

       首先我们需要写一个类,该类继承Comparator,同时实现按姓名排序的方法。

public class CompareByName implements Comparator {@Overridepublic int compare(Object o1, Object o2) {System.out.println("CompareByName----");Person p1=(Person)o1;Person p2=(Person)o2;int temp = p1.getName().compareTo(p2.getName());return temp==0?p1.getAge()-p2.getAge():temp;}}
      然后,在创建TreeSet的时候需要直接用到TreeSet(Comparator c)的构造函数。

public static void main(String[] args) {/* * 以Person对象的年龄进行从小到大的排序。 */TreeSet ts = new TreeSet(new CompareByName()) ;//new CompareByName()ts.add(new Person("linweieran",24));ts.add(new Person("linweieran1",23)); //bean.Person cannot be cast to java.lang.Comparablets.add(new Person("linweieran2",22));ts.add(new Person("weixiaobao",25));ts.add(new Person("linweieran",24));Iterator  it = ts.iterator();while(it.hasNext()){Person p = (Person)it.next();System.out.println(p.getName()+":"+p.getAge());}}
这样子就可以实现排序功能了。

哈哈,运行结果正确!!!


1 0