TreeSet集合以及两种排序

Set集合

Set集合的概述和特性

• Set集合不可以存储重复的元素
• 无索引，不能通过经典的for循环进行遍历，可以通过迭代器和foreach循环进行遍历
• 需要排序规则进行遍历

Set集合的使用

• 由于Set是一个接口，所以需要使用Set的子类来实现里面的方法
  • Set常用的子类有 TreeSet 类 和 HashSet 类

例：使用 TreeSet 存储字符串并进行遍历

public class SetListDemo {
    public static void main(String[] args) {
        setListDemo_1();
   }

    private static void setListDemo_1() {
        //        set集合没有索引，不能存重复的元素
//         存取顺序不一致
        Set<String> set = new TreeSet<>();
        set.add("a");
        set.add("d");
        set.add("a");
        set.add("c");
        Iterator i = set.iterator();
        while (i.hasNext()){
            System.out.println(i.next());
        }

        System.out.println("--------------");
//        foreach循环
        for(String s: set){
            System.out.println(s);
        }
    }
}

控制台输出结果：
从结果可以看到

TreeSet集合

TreeSet集合的概述和特点

• TreeSet实现了Set类，也是不可以存储存储重复的元素
• 没有索引，但是可以将元素按照规则进行排序
  • TreeSet()：使用无参构造根据其元素的自然排序进行排序
  • TreeSet(new Comparator )：根据指定的比较器进行排序

TreeSet集合的基本使用

• TreeSet可以对集合中的元素进行排序,在添加元素的时候会自动去调用Comparable接口的compareTo方法
• 有些泛型类已经写好了排序规则，比如String 和 Integer 都已经实现了Comparable接口，也重写了compareTo方法
• String类是按照字典顺序的规则进行排序，比如上一个例子就是按照String重写的规则进行的排序；而 Integer 则是按照整数大小规则进行排序

例：存储Integer类型的数据并进行遍历

public class TreeSetDemo01 {
    public static void main(String[] args) {
        //创建集合对象
        TreeSet<Integer> ts = new TreeSet<Integer>();

        //添加元素
        ts.add(10);
        ts.add(40);
        ts.add(30);
        ts.add(50);
        ts.add(20);

        ts.add(30);

        //遍历集合
        for(Integer i : ts) {
            System.out.println(i);
        }
    }
}

控制台输出结果：

TreeSet的自然排序Comparable的使用

• 当使用的泛型类是使用自己定义的时候，就需要在泛型类中实现Comparable接口，重写comparaTo方法
• 重写comparaTo方法时，需要注意排序规则要按照要求的主要条件和次要条件来编写

例：存储学生对象并进行遍历，按照年龄从小到大排序，年龄相同时，按照姓名首字母顺序排序

先写一个学生类，里面存入学生的年龄和姓名，重写comparaTo方法

public class Student implements Comparable<Student>{
    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;
    }

    @Override
    public String toString() {
        return "Student{" +
                "name='" + name + '\'' +
                ", age=" + age +
                '}';
    }

    @Override
    public int compareTo(Student o) {
        //按照对象的年龄进行排序
        //主要判断条件: 按照年龄从小到大排序
        //当result小于0的时候，就将新存入的数据放在左边，大于0就将新存入的数据放在右边，等于0的时候不进行存取
        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);
        }
    }
}

控制台输出结果：

比较器Comparator的使用

• 使用TreeSet集合存储自定义对象，TreeSet的带参构造方法是使用比较器排序对元素进行排序
• 比较器排序，让TreeSet的构造方法接受Comparator的实现类对象（多态），重写compare(T o1, T o2)方法
• 同样，要注意排序规则的主次条件

例如：存储老师对象并进行遍历，按照年龄从小到大进行排序，年龄相同时，按照姓名的字目进行排序

先写一个老师类：

public class Teacher {
    private String name;
    private int age;

    public Teacher() {
    }

    public Teacher(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;
    }

    @Override
    public String toString() {
        return "Teacher{" +
                "name='" + name + '\'' +
                ", age=" + age +
                '}';
    }
}

在测试类中重写comparator方法：

public class MyTreeSet4 {
    public static void main(String[] args) {
      	//创建集合对象
        TreeSet<Teacher> ts = new TreeSet<>(new Comparator<Teacher>() {
            @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，表示当前存入的元素跟集合中元素重复了，不存
  • 如果返回值为正数，表示当前存入的元素是较大值，存右边