【集合框架】JDK1.8源码分析之Comparable && Comparator(九)

时间:2021-12-04 17:58:01

一、前言

  在Java集合框架里面,各种集合的操作很大程度上都离不开Comparable和Comparator,虽然它们与集合没有显示的关系,但是它们只有在集合里面的时候才能发挥最大的威力。下面是开始我们的分析。

二、示例

  在正式讲解Comparable与Comparator之前,我们通过一个例子来直观的感受一下它们的使用。

  首先,定义好我们的Person类  

class Person {
    String name;
    int age;
    
    public Person(String name, int age) {
        this.name = name;
        this.age = age;
    }
    
    public String toString() {
        return "[name = " + name + ", age = " + age + "]";
    }
}

  其次编写测试代码,代码如下 

public class Test {
    public static void main(String[] args) {
        List<String> nameLists = new ArrayList<String>();
        nameLists.addAll(Arrays.asList("aa", "ab", "bc", "ba"));
        Collections.sort(nameLists);
        System.out.println(nameLists);
        
        List<Person> personLists = new ArrayList<Person>();
        personLists.addAll(Arrays.asList(new Person("leesf", 24), new Person("dyd", 24), new Person("ld", 0)));
        Collections.sort(personLists); // 出错
        System.out.println(personLists); } }

  说明:上述代码是两份同样的逻辑,同样的操作,但是,对于List<String>不会报错,对于List<Person>类型就会报错,为什么?为了解决这个问题,我们需要讲解今天的主角Comparable && Comparator。如果知道怎么解决的园友也不妨瞧瞧,开始分析。

三、源码分析

  3.1 Comparable

  1. 类的继承关系 

public interface Comparable<T>

  说明:Comparable就是一个泛型接口,很简单。

  2. compareTo方法

public int compareTo(T o);

  说明:compareTo方法就构成了整个Comparable源码的唯一的有效方法。

  3.2 Comparator

  1. 类的继承关系  

public interface Comparator<T>

  说明:同样,Comparator也是一个泛型接口,很简单。

  2. compare方法 

int compare(T o1, T o2);

  说明:Comparator接口中一个核心的方法。

  3. equals方法

boolean equals(Object obj);

  说明:此方法是也是一个比较重要的方法,但是一般不会使用,可以直接使用Object对象的equals方法(所有对象都继承自Object)。

  其他在JDK1.8后添加的方法对我们的分析不产生影响,有感兴趣的读者可以自行阅读源码,了解更多细节。

四、解决思路

  4.1. 分析问题

  在我们的程序中,List<String>类型是可以通过编译的,但是List<Person>类型却不行,我们猜测肯定是和元素类型String、Person有关系。既然是这样,我们来看String在Java中的定义。

public final class String
    implements java.io.Serializable, Comparable<String>, CharSequence

  说明:我们平时说String为final类型,不可被继承,查看源码,确实是这样。注意查看String实现的接口,直觉告诉我们Comparable<String>很重要,之前我们已经分析过了Comparable接口,既然String实现了这个接口,那么肯定也实现了compareTo方法,顺藤摸瓜,String的compareTo方法如下:

public int compareTo(String anotherString) {
    // this对象所对应的字符串的长度
    int len1 = value.length; 
    // 参数对象所对应字符串的长度
    int len2 = anotherString.value.length; 
    // 取长度较小者
    int lim = Math.min(len1, len2); 
    // value是String底层的实现,为char[]类型数组
    // this对象所对应的字符串
    char v1[] = value;
    // 参数对象所对应的字符串
    char v2[] = anotherString.value;
    
    int k = 0;
    // 遍历两个字符串
    while (k < lim) {
        char c1 = v1[k];
        char c2 = v2[k];
        // 如果不相等,则返回
        if (c1 != c2) {
            return c1 - c2;
        }
        // 继续遍历
        k++;
    }
    // 一个字符串是另外一个字符串的子串
    return len1 - len2;
}

  说明:我们可以看到String中compareTo方法具体的实现。比较同一索引位置的字符大小。

  分析了String的compareTo方法后,并且按照在compareTo方法中的逻辑进行排序,之于如何排序涉及到具体的算法问题,以后我们会进行分析。于是乎,我们知道了之前示例程序的问题所在:Person类没有实现Comparable接口。

  4.2. 解决问题

  1. 修改我们的Person类的定义,修改为如下:  

Person implements Comparable<Person>

  2. 实现compareTo方法,并实现我们自己的想要比较的逻辑,如我们想要首先根据年龄比较(采用升序),若年龄相同,则根据姓名的ASCII顺序来比较。那么我们实现的compareTo方法如下: 

    int compareTo(Person anthor) {
        if (this.age < anthor.age)
            return -1;
        else if (this.age == anthor.age)
            return this.name.compareTo(anthor.name);
        else
            return 1;
    }

  说明:于是乎,修改后的程序如下:

  Person类代码如下  

class Person implements Comparable<Person> {
    String name;
    int age;
    
    public Person(String name, int age) {
        this.name = name;
        this.age = age;
    }
    
    public String toString() {
        return "[name = " + name + ", age = " + age + "]";
    }
    
    @Override
    public int compareTo(Person anthor) {
        if (this.age < anthor.age)
            return -1;
        else if (this.age == anthor.age)
            return this.name.compareTo(anthor.name);
        else
            return 1;
    }
}

  测试类代码不变

  运行结果如下:

  [aa, ab, ba, bc]
  [[name = ld, age = 0], [name = dyd, age = 24], [name = leesf, age = 24]]

  说明:我们可以看到Person类的排序确实按照了在compareTo方法中定义的逻辑进行排序。这样,就修正了错误。

五、问题提出

  上面的Comparable接口解决之前出现的问题。但是,如果我现在不想按照刚刚的逻辑进行排序了,想按照一套新的逻辑排序,如只根据姓名比较来进行排序。此时,我们需要修改Comparable接口的compare方法,添加新的比较逻辑。过了一久,用户又希望采用别的逻辑进行排序,那么,又得重新修改compareTo方法里面的逻辑,可以通过标志位来做if判断,用来判断用户想要使用哪种比较逻辑,这样会造成会造成代码很臃肿,不易于维护。此时,一种更好的解决办法就是使用Comparator接口。

  5.1 比较逻辑一

  首先根据年龄比较(采用升序),若年龄相同,则根据姓名的ASCII顺序来比较。

  那么我们可以定义这样的Comparator,具体代码如下: 

class ComparatorFirst implements Comparator<Person> {    
    public int compare(Person o1, Person o2) {
        if (o1.age < o2.age)
            return -1;
        else if (o1.age == o2.age)
            return o1.name.compareTo(o2.name);
        else 
            return 1;
    }    
}

  测试代码做如下修改:

  将Collections.sort(personLists) 改成 Collections.sort(personLists, new ComparatorFirst());

  sort的两种重载方法,后一种允许我们传入自定义的比较器。

  运行结果如下:

  [aa, ab, ba, bc]
  [[name = ld, age = 0], [name = dyd, age = 24], [name = leesf, age = 24]]

  结果说明:我们看到和前面使用Comparable接口得到的结果相同。

  5.2 比较逻辑二

  直接根据姓名的ASCII顺序来比较。

  则我们可以定义如下比较器  

class ComparatorSecond implements Comparator<Person> {
    public int compare(Person o1, Person o2) {
        return o1.name.compareTo(o2.name);
    }    
}

  测试代码做如下修改:

  将Collections.sort(personLists) 改成 Collections.sort(personLists, new ComparatorSecond());

  运行结果:

  [aa, ab, ba, bc]
  [[name = dyd, age = 24], [name = ld, age = 0], [name = leesf, age = 24]]

  说明:我们可以看到这个比较逻辑和上一个比较器的逻辑不相同,但是也同样完成了用户的逻辑。

  我们还可以按照我们的意愿定义其他更多的比较器,只需要在compareTo中正确完成我们的逻辑即可。

  5.3 Comparator优势

  从上面两个例子我们应该可以感受到Comparator比较器比Comparable接口更加灵活,可以更友好的完成用户所定义的各种比较逻辑。

六、总结

  分析了Comparable和Comparator,掌握了在不同的场景中使用不同的比较器,写此篇博客后对两者的使用和区别也更加的清晰了。谢谢各位园友的观看~