java中的自定义比较器

在java中很多时候需要我们自定义比较器, 例如在向TreeSet中添加元素时, 元素必须可比较. 本文介绍两种自定义比较器的方法:

1. 实现Comparable接口

Comparable接口中只有一个方法, compareTo()方法. 重写该方法使该类元素强制具有比较性. CompareTo()方法返回正数, 说明该对象大于传进来的对象, 返回0, 代表二者相等, 返回负数, 代表该对象小于传进来的对象.
例如在Student类中, 实现该接口, 可以使得两个Student的对象进行比较.
这种方式叫做元素的自然顺序, 或者叫默认顺序.
注意如果在TreeSet中使用该方法时, 如果返回0表示两个元素相同, 新添加的元素不会被加入TreeSet.

class Student implements Comparable{
    private int no;
    private String name;
    @Override
    public int compareTo(Object o) {
        if (!(o instanceof Student)){
            throw new RuntimeException("不是学生对象!")
        }
        Student s = (Student) o;
        if(this.no > s.no){
            return 1;
        }
        else if(this.no == s.no){
            return this.name.compareTo(s.name);
        }
        else {
            return -1;
        }

    }
}

2. 实现Comparator接口

当元素自身不具备比较性或具备的比较性不是所需要的, 就需要让集合自身具备比较性.在集合初始化时就有了比较方式.
定义了比较器, 将比较器的对象作为参数传递给集合的构造函数.

class MyComparator implements Comparator{
        @Override
        public int compare(Object o1, Object o2) {
            Student s1 = (Student) o1;
            Student s2 = (Student) o2;
            int num = s1.getName().compareTo(s2.getName());
            if(num == 0){
                if(s1.getAge() > s2.getAge()){
                    num = 1;
                }
                else if(s1.getAge() == s2.getAge()){
                    num = 0;
                }
                else
                    num = -1;
            }
            return num;
        }
    }

在定义TreeSet时, 就要把比较器作为参数传进去.

TreeSet ts = new TreeSet(new MyComparator());