黑马程序员对TreeSet集合中的元素排序的两种方式

------ Java培训、Android培训、iOS培训、.Net培训、期待与您交流！ -------

TreeSet对元素进行排序的方式一：让元素自身具有比较功能，元素需要实现Comparable接口，覆盖compareTo方法

package org.blog.test2;
import java.util.TreeSet;
class Person implements Comparable{
    private String name;
    private int age;
    Person(String n,int a){
     this.setName(n);
     this.setAge(a);
    }
    public String getName(){
     return name;
    }
    public void setName(String n){
     this.name=n;
    }
    public int getAge(){
     return age;
    }
    public void setAge(int a){
     this.age=a;
    }
    //先按年龄比较，如果年龄相同再按姓名比较是否相同
 public int compareTo(Object o) {
  Person p=(Person)o;
  int temp=this.age-p.age;  
  return temp==0?this.name.compareTo(p.name):temp;
 }
 public String toString(){
  return "姓名："+name+"年龄："+age;
 } 
}
public class TreeSetDemo {
 public static void main(String[] args) {
        TreeSet<Person> ts=new TreeSet<Person>();
        ts.add(new Person("xiaoqiang",23));
        ts.add(new Person("lisi",20));
        ts.add(new Person("wangwu",23));
        ts.add(new Person("suimei",19));
        System.out.println(ts);
 }
}

输出结果：[姓名：suimei年龄：19, 姓名：lisi年龄：20, 姓名：wangwu年龄：23, 姓名：xiaoqiang年龄：23]
由此可以看出年龄都为23岁时，wangwu和xiaoqiang又进行了姓名的排序。

 TreeSet对元素进行排序的方式二：让集合具备比较功能，定义一个类实现Comparator接口，覆盖Compare方法，然后把该类对象作为参数传递给TreeSet集合的构造函数。

package org.blog.test2;
import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;
//定义学生类
class Student{
 private String name;
 private int age;
 Student(String n,int a){
  this.setName(n);
  this.setAge(a);
 }
 public String getName(){
  return name;
 }
 public void setName(String n){
  this.name=n;
 }
 public int getAge(){
  return age;
 }
 public void setAge(int a){
  this.age=a;
 }
 //覆写toString方法
 public String toString(){
  return "姓名："+name+"年龄："+age;
 }
}
//定义一个类实现Comparator接口
class ComparatorByName implements Comparator<Student>{
 //覆盖Compare方法
 public int compare(Student o1, Student o2) {
  Student s1=(Student)o1;
  Student s2=(Student)o2;
  int temp=s1.getName().compareTo(s2.getName());
  return temp==0?s1.getAge()-s2.getAge():temp;//如果姓名相同就比较年龄
 } 
}
public class TreeSetDemo2 {
 public static void main(String[] args) {
    //该类对象作为参数传递给TreeSet集合的构造函数
       TreeSet<Student> ts=new TreeSet<Student>(new ComparatorByName());
       //把学生类作为对象存储到集合中
       ts.add(new Student("wangwu",25));
       ts.add(new Student("lisi",22));
       ts.add(new Student("xiaomei",20));
       ts.add(new Student("lisi",23));
       ts.add(new Student("hanhong",27));
       Iterator<Student> it=ts.iterator();
       while(it.hasNext()){
        //向下转型
        Student s=(Student)it.next();
        System.out.println(s.getName()+s.getAge());
       }      
 }
}

输出结果：hanhong27
                    lisi22
                    lisi23
                    wangwu25
                    xiaomei20