TreeSet-自然排序与比较器排序

TreeSet-自然排序与比较器排序

特性

• TreeSet是有序的(即按照一定的规则排序)，是用二叉树实现的。
• TreeSet 的存取顺序无序

自然排序

可以通过comparable接口重写compareTo方法，改写排序规则，来达到按照一定的规则进行排序。

public class Student implements Comparable//该接口强制对实现它的每个类的对象进行强制整体排序
{
    private String name;
    private int age;
    Student (String name,int age){
        this.name=name;
        this.age =age;

    }
    public String getName() {
        return name;
    }
    public int getAge() {
        return age;
    }
    public int compareTo(Object obj) {
        /*  当this.age > age 时进行升序，反之降序，相等则年龄重复，返回。
        */
        if(!(obj instanceof Student))
            throw new RuntimeException("不是学生对象");
        Student s=(Student)obj;
        System.out.println(this.name+"...........compareto..........."+s.name);
        if(this.age>s.age)
            return 1;
        if(this.age==s.age)
        {
            if(this.age==s.age)
                return this.name.compareTo(s.name);//String已经实现了这个CompareTo接口，直接调用方法即可

            else
                return -1;
        }

        return -1;

    }
}

public class treesetDemo {

    public static void main(String[] args) {

        TreeSet ts=new TreeSet();
        ts.add(new Student("lisi01",19));
        ts.add(new Student("lisi02",22));
        ts.add(new Student("lisi007",20));
        ts.add(new Student("lisi009",19));
        ts.add(new Student("lisi01",19));

        Iterator it=ts.iterator();
        while (it.hasNext()) {
            Student s=(Student)it.next();
            System.out.println(s.getName()+"     "+s.getAge());
        }
    }
}

比较器排序

构建比较器，通过构造匿名内部类来实现Comparator接口，进行排序。

public static void main(String[] args) {
        TreeSet<Student> tree = new TreeSet<>(new Comparator<Student>() {
            @Override
            public int compare(Student o1, Student o2) {
                int a = o1.getAge() - o2.getAge();
                if (a==0){
                    return o1.getName().compareTo(o2.getName());
                    //String类型的compareTo()方法
                }
                return a;
            }
        });
        Student st1 = new Student("小明",10);
        Student st2 = new Student("小刚",12);
        Student st3 = new Student("小花",13);
        Student st4 = new Student("小红",14);
        Student st5 = new Student("小强",14);
        Student st6 = new Student("小李",16);

        tree.add(st1);
        tree.add(st2);
        tree.add(st3);
        tree.add(st4);
        tree.add(st5);
        tree.add(st6);

        for (Student st :tree){
            System.out.println(st);
        }
    }

遍历方法

可以根据根节点的位置，来判断二叉树的排序方式。

• 前序遍历：根 左 右
• 中序遍历：左 根 右
• 后序遍历：左 右 根