TreeSet的排序

TreeSet的的排序使用

TreeSet的两种排序方式：

• 自然排序
• 比较器排序

自然排序

• 新建一个实体类，注意这个实体类要实现接口  Comparable 。见代码详解：

package zh.stu.sort.TreeSetTest;

public class Person implements Comparable<Person>{
    
    private  String uname;
    
    private int age;
    
    public Person(){
        super();
    }
    
    public Person(String uname, int age) {
        super();
        this.uname = uname;
        this.age = age;
    }
    
    public String getUname() {
        return uname;
    }
    
    public void setUname(String uname) {
        this.uname = uname;
    }
    
    public int getAge() {
        return age;
    }
    
    public void setAge(int age) {
        this.age = age;
    }
    
    public int compareTo(Person p) {
        // return -1   -1 表示放在红黑树的左边，就是逆序输出
        // return 1     1 表示放在红黑树的右边，就是顺序输出
        // return 0     0 表示元素相同，仅存放第一个元素
        
        // 主要条件 姓名的长度 ，如果姓名长度小的那么就放在左子树，否则放在右子树
        int num=this.uname.length()- p.uname.length();
        
        // 此时姓名的长度相同，不代表姓名相同。
        // 如果按照字典顺序比，string  对象位于参数字符串之前，那么比较结果是一个负整数
        // 如果按照字典顺序比， string 对象位于参数字符串之后，那么比较结果为一个正整数
        // 如果两个字符串相等，那么返回0
        int num1= num ==0 ? this.uname.compareTo(p.uname) : num;
        // 内容相同，年龄不一定相同。在按照年龄排序
        int num2 = num1==0 ? this.age-p.age : num1;
        return num2;
    }
}

• 做排序测试

  package zh.stu.sort.TreeSetTest;
  
  import java.util.TreeSet;
  
  public class SortTest {
      
      public static void main(String[] args) {
          // 自然排序
          TreeSet<Person> treeSet = new TreeSet<Person>();
          // 在使用的是汉字排序的时候
          Person p=new Person("詹姆斯",18);
          Person p1=new Person("艾弗森",20);
          Person p2=new Person("科比",20);
          Person p3=new Person("乔丹",22);
          Person p4=new Person("戴维斯",12);
          treeSet.add(p);
          treeSet.add(p1);
          treeSet.add(p2);
          treeSet.add(p3);
          treeSet.add(p4);
          for (Person person : treeSet) {
              System.out.println(person.getUname() +"----"+ person.getAge());
          }
      }
  }

• 在控制台输出结果

  乔丹----22
  科比----20
  戴维斯----12
  艾弗森----20
  詹姆斯----18

比较器排序

• 新建一个实体类

  package zh.stu.sort.TreeSetTest.TheComparator;
  
  public class User {
      
      private String name;
      
      private int age;
      
      public User(){
          super();
      }
      
      public User(String name, int age) {
          super();
          this.name = name;
          this.age = age;
      }
      
      public String getName() {
          return name;
      }
      
      public void setName(String name) {
          this.name = name;
      }
      
      public int getAge() {
          return age;
      }
      
      public void setAge(int age) {
          this.age = age;
      }
  }

• 新加一个比较器，但是要实现接口Comparator

  package zh.stu.sort.TreeSetTest.TheComparator;
  
  import java.util.Comparator;
  
  // 实现接口Comparator
  public class MyComparator implements Comparator<User> {
      
      public int compare(User o1, User o2) {
          
          int num=o1.getName().length()-o2.getName().length();
          
          int num1= num == 0 ? o1.getName().compareTo(o2.getName()) : num;
          
          int num2 = num1 == 0? o1.getAge() - o2.getAge() : num1;
          
          return num2;
      }
  }

• 最后在类中做测试

  package zh.stu.sort.TreeSetTest.TheComparator;
  
  import java.util.Comparator;
  import java.util.TreeSet;
  
  public class Test {
      
      public static void main(String[] args) {
          TreeSet<User> users = new TreeSet<User>(new MyComparator());
      
          User user =new User("乔丹",22);
          User user1 =new User("艾弗森",24);
          User user2 =new User("乔丹",21);
          User user3 =new User("Zion",62);
          User user4 =new User("欧文",22);
          
          users.add(user);
          users.add(user1);
          users.add(user2);
          users.add(user3);
          users.add(user4);
      
          for (User user5 : users) {
              System.out.println(user5.getName() +"===="+ user5.getAge() );
          }
      }
  }

  在控制台输出的结果为

  乔丹====21
  乔丹====22
  欧文====22
  艾弗森====24
  Zion====62