Java TreeSet详解

Java TreeSet详解

一、TreeSet集合概述和特点

• 不可以存储重复元素
• 没有索引
• 可以将元素按照规则进行排序
  • TreeSet()：根据其元素的自然排序进行排序
  • TreeSet(Comparator comparator) ：根据指定的比较器进行排序

二、排序规则详解 

1、自然排序Comparable的使用

（1）案例需求

存储学生对象并遍历，创建TreeSet集合使用无参构造方法

要求：按照年龄从小到大排序，年龄相同时，按照姓名的字母顺序排序

（2）实现步骤

1）使用空参构造创建TreeSet集合

        用TreeSet集合存储自定义对象，无参构造方法使用的是自然排序对元素进行排序的

2）自定义的Student类实现Comparable接口

        自然排序，就是让元素所属的类实现Comparable接口，重写compareTo(T o)方法

3）重写接口中的compareTo方法

        重写方法时，一定要注意排序规则必须按照要求的主要条件和次要条件来写

（3）代码实现

//学生类
public class Student implements Comparable{ private String name; private int age;public Student() {}public Student(String name, int age) {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;}@Overridepublic String toString() {return "Student{" +"name='" + name + '\'' +", age=" + age +'}';}@Overridepublic int compareTo(Student o) {//按照对象的年龄进行排序//主要判断条件: 按照年龄从小到大排序int result = this.age - o.age;//次要判断条件: 年龄相同时，按照姓名的字母顺序排序result = result == 0 ? this.name.compareTo(o.getName()) : result;return result;}
}//测试类
public class MyTreeSet2 { public static void main(String[] args) { //创建集合对象 TreeSet<Student> ts = new TreeSet<>(); //创建学生对象 Student s1 = new Student("zhangsan",28); Student s2 = new Student("lisi",27); Student s3 = new Student("wangwu",29); Student s4 = new Student("zhaoliu",28); Student s5 = new Student("qianqi",30); //把学生添加到集合 ts.add(s1); ts.add(s2); ts.add(s3); ts.add(s4); ts.add(s5); //遍历集合 for (Student student : ts) { System.out.println(student); } } 
}

 2、比较器排序Comparator的使用

（1）案例需求

存储老师对象并遍历，创建TreeSet集合使用带参构造方法

要求：按照年龄从小到大排序，年龄相同时，按照姓名的字母顺序排序

（2）实现步骤

1）用TreeSet集合存储自定义对象，带参构造方法使用的是比较器排序对元素进行排序的

2）比较器排序，就是让集合构造方法接收Comparator的实现类对象，重写compare(T o1,T o2)方法

3）重写方法时，一定要注意排序规则必须按照要求的主要条件和次要条件来写

（3）代码实现

//老师类
@Data
public class Teacher { private String name; private int age;public Teacher() {}public Teacher(String name, int age) {this.name = name;this.age = age;}
}//测试类
public class MyTreeSet4 { public static void main(String[] args) { //创建集合对象 TreeSet ts = new TreeSet<>(new Comparator() { @Override public int compare(Teacher o1, Teacher o2) { //o1表示现在要存入的那个元素 //o2表示已经存入到集合中的元素//主要条件int result = o1.getAge() - o2.getAge();//次要条件result = result == 0 ? o1.getName().compareTo(o2.getName()) : result;return result;}});//创建老师对象Teacher t1 = new Teacher("zhangsan",23);Teacher t2 = new Teacher("lisi",22);Teacher t3 = new Teacher("wangwu",24);Teacher t4 = new Teacher("zhaoliu",24);//把老师添加到集合ts.add(t1);ts.add(t2);ts.add(t3);ts.add(t4);//遍历集合for (Teacher teacher : ts) {System.out.println(teacher);}}
}

三、两种比较方式总结

• 两种比较方式小结
  • 自然排序: 自定义类实现Comparable接口,重写compareTo方法,根据返回值进行排序
  • 比较器排序: 创建TreeSet对象的时候传递Comparator的实现类对象,重写compare方法,根据返回值进行排序
  • 在使用的时候,默认使用自然排序,当自然排序不满足现在的需求时,必须使用比较器排序
• 两种方式中关于返回值的规则
  • 如果返回值为负数，表示当前存入的元素是较小值，存左边
  • 如果返回值为0，表示当前存入的元素跟集合中元素重复了，不存
  • 如果返回值为正数，表示当前存入的元素是较大值，存右边