Java基础知识强化之集合框架笔记47：Set集合之TreeSet保证元素唯一性和比较器排序的原理及代码实现（比较器排序：Comparator）

1. 比较器排序（定制排序）

前面我们说到的TreeSet的自然排序是根据集合元素的大小，TreeSet将它们以升序排列。

但是如果需要实现定制排序，比如实现降序排序，则要通过比较器排序（定制排序）实现。

比较器排序是通过Comparator接口帮助的，这个接口包含一个方法int compare（T o1， T o2）方法，该方法用于比较o1和o2的大小：

• 如果方法返回正整数，则表明o1 大于 o2
• 如果方法返回0，则表明o1 等于 o2
• 如果方法返回负整数，则表明o1 小于 o2

int compare(Object o1, Object o2) 返回一个基本类型的整型
　　如果要按照升序排序：
　　　　则o1 小于o2，返回-1（负数），相等返回0，o1大于o2返回1（正数）
　　如果要按照降序排序：
 　　　  则o1 小于o2，返回1（正数），相等返回0，o1大于o2返回-1（负数）

2. TreeSet保证元素唯一性和比较器排序的原理及代码实现（比较器排序）

（1）Student.java：

package cn.itcast_07;

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

    public Student() {
        super();
    }

    public Student(String name, int age) {
        super();
        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;
    }
}

（2）TreeSetDemo.java：

package cn.itcast_07;

import java.util.Comparator;
import java.util.TreeSet;

/*
 * 需求：请按照姓名的长度排序
 * 
 * TreeSet集合保证元素排序和唯一性的原理
 * 唯一性：是根据比较的返回是否是0来决定。
 * 排序：
 *         A:自然排序(元素具备比较性)
 *             让元素所属的类实现自然排序接口 Comparable
 *         B:比较器排序(集合具备比较性)
 *             让集合的构造方法接收一个比较器接口的子类对象 Comparator
 */
public class TreeSetDemo {
    public static void main(String[] args) {
        // 创建集合对象
        // TreeSet<Student> ts = new TreeSet<Student>(); //自然排序
        // public TreeSet(Comparator comparator) //比较器排序
        // TreeSet<Student> ts = new TreeSet<Student>(new MyComparator());

        // 如果一个方法的参数是接口，那么真正要的是接口的实现类的对象
        // 而匿名内部类就可以实现这个东西
        TreeSet<Student> ts = new TreeSet<Student>(new Comparator<Student>() {
            @Override
            public int compare(Student s1, Student s2) {
                // 姓名长度
                int num = s1.getName().length() - s2.getName().length();
                // 姓名内容
                int num2 = num == 0 ? s1.getName().compareTo(s2.getName())
                        : num;
                // 年龄
                int num3 = num2 == 0 ? s1.getAge() - s2.getAge() : num2;
                return num3;
            }
        });

        // 创建元素
        Student s1 = new Student("linqingxia", 27);
        Student s2 = new Student("zhangguorong", 29);
        Student s3 = new Student("wanglihong", 23);
        Student s4 = new Student("linqingxia", 27);
        Student s5 = new Student("liushishi", 22);
        Student s6 = new Student("wuqilong", 40);
        Student s7 = new Student("fengqingy", 22);
        Student s8 = new Student("linqingxia", 29);

        // 添加元素
        ts.add(s1);
        ts.add(s2);
        ts.add(s3);
        ts.add(s4);
        ts.add(s5);
        ts.add(s6);
        ts.add(s7);
        ts.add(s8);

        // 遍历
        for (Student s : ts) {
            System.out.println(s.getName() + "---" + s.getAge());
        }
    }
}

运行结果如下：