TreeSet的理解

1.TreeSet集合概述和特点【应用】

• 不可以存储重复元素

• 没有索引

• 可以将元素按照规则进行排序

  • TreeSet()：根据其元素的自然排序进行排序

  • TreeSet(Comparator comparator) ：根据指定的比较器进行排序

 TreeSet就是通过定义的排序规则进行排序,下面介绍自然排序Comparable的使用和定制排序Comparator的使用

主要代码示例:

//定义一个学生类
public class Student implements Comparable<Student>{
    private String name;
    private int age;

.....构造器,get,set,toString方法已省略.........

      @Override
    public 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.自然排序Comparable的使用

在包装类和String类型由于重写了compareTo方法,所以能够添加到TreeSet集合中,对于自定义的类就要按照以下步骤:

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

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

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

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

3. 重写接口中的compareTo方法

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

 3.定制排序Comparator的使用

• 用TreeSet集合存储自定义对象，带参构造方法使用的是比较器排序对元素进行排序的
• 比较器排序，就是让集合构造方法接收Comparator的实现类对象，重写compare(T o1,T o2)方法
• 重写方法时，一定要注意排序规则必须按照要求的主要条件和次要条件来写

 

//学生类同上




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

在这里也可以发现定制排序优先级高于自然排序

4.两种比较方式总结

• 两种比较方式小结

  • 自然排序: 自定义类实现Comparable接口,重写compareTo方法,根据返回值进行排序

  • 比较器排序: 创建TreeSet对象的时候传递Comparator的实现类对象,重写compare方法,根据返回值进行排序

  • 在使用的时候,默认使用自然排序,当自然排序不满足现在的需求时,必须使用比较器排序

• 两种方式中关于返回值的规则

  • 如果返回值为负数，表示当前存入的元素是较小值，存左边

  • 如果返回值为0，表示当前存入的元素跟集合中元素重复了，不存

  • 如果返回值为正数，表示当前存入的元素是较大值，存右边