TreeSet介绍

一、TreeSet原理：

1.TreeSet存储对象的时候, 可以排序, 但是需要指定排序的算法

2.Integer能排序(有默认顺序), String能排序(有默认顺序), 自定义的类存储的时候出现异常(没有顺序)

3.如果想把自定义类的对象存入TreeSet进行排序, 那么必须实现Comparable接口，或者实现一个比较器

             在类上implements Comparable
             重写compareTo()方法
             在方法内定义比较算法, 根据大小关系, 返回正数负数或零
             在使用TreeSet存储对象的时候, add()方法内部就会自动调用compareTo()方法进行比较, 根据比较结果使用二叉树形式进行存储

4.TreeSet依靠TreeMap来实现

二、TreeSet实现有序的2种方式

1）、元素实现Comparable接口：

实例：

实体类User（实现Comparable接口）：

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50

package com.model.elgin.collections; public   class User implements Comparable<User>{         private   int id;     private   String name;     private   String phone;          public   int getId() {         return id;     }     public   void setId( int id) {         this .id = id;     }     public   String getName() {         return name;     }     public   void setName( String name) {         this .name = name;     }     public   String getPhone() {         return phone;     }     public   void setPhone( String phone) {         this .phone = phone;     }          public User( int id, String name, String phone) {         super ();         this .id = id;         this .name = name;         this .phone = phone;     }          @Override     public   String toString() {         return   "User [id=" + id + ", name=" + name + ", phone=" + phone + "]" ;     }          @Override     public   int compareTo(User user) {         if ( this .id>user.getId()){             return   1 ;         } else   if ( this .id==user.getId()){             return   0 ;         }         return - 1 ;     } }

测试类：

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31

package com.model.elgin.collections; import java.util.Set; import java.util.TreeSet; public   class TestTreeSet {          public   static   void main( String [] args) {                  /*          * TreeSet有序、唯一。          * 实现有序的2种方式：          * 1.集合中的元素对象实现Comparable接口，TreeSet声明使用无参构造          * 2.声明一个比较器 Comparator ，TreeSet声明使用有参构造，参数传入比较器的一个实例对象          *           */         Set<User> set= new TreeSet<User>();         User user1= new User( 10 , "jim" , "13232326565" );         User user2= new User( 20 , "jack" , "13232326566" );         User user3= new User( 30 , "lily" , "13232326567" );         User user4= new User( 40 , "lucy" , "13232326568" );         User user5= new User( 40 , "lucy" , "13232326568" );         set.add(user1);         set.add(user3);         set.add(user4);         set.add(user2);         set.add(user5);         System.out.println(set);     } }

运行之后发现结果有序（根据id由小到大）排列

2）、通过自定义比较器实现（User类不需要实现Comparable接口）：

比较器类： UserComparator

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16

package com.model.elgin.collections; import java.util.Comparator; public   class UserComparator implements Comparator<User> {     @Override     public   int compare(User user1, User user2) {         if (user1.getId()>user2.getId()){             return   1 ;         } else   if (user1.getId()==user2.getId()){             return   0 ;         }         return - 1 ;     }    }

测试类：

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32

package com.model.elgin.collections; import java.util.Comparator; import java.util.Set; import java.util.TreeSet; public   class TestTreeSet {          public   static   void main( String [] args) {                  /*          * TreeSet有序、唯一。          * 实现有序的2种方式：          * 1.集合中的元素对象实现Comparable接口，TreeSet声明使用无参构造          * 2.声明一个比较器 Comparator ，TreeSet声明使用有参构造，参数传入比较器的一个实例对象          *           */         Comparator<User> comparator= new UserComparator();         Set<User> set= new TreeSet<User>(comparator);         User user1= new User( 10 , "jim" , "13232326565" );         User user2= new User( 20 , "jack" , "13232326566" );         User user3= new User( 30 , "lily" , "13232326567" );         User user4= new User( 40 , "lucy" , "13232326568" );         User user5= new User( 40 , "lucy" , "13232326568" );         set.add(user1);         set.add(user3);         set.add(user4);         set.add(user2);         set.add(user5);         System.out.println(set);     } }

运行，同样可以得到上述有序的结果。