Java中TreeSet、HashSet、Collection重写比较器

关注

原创 2023-08-02 17:25:56 1.1k 阅读 AI 写同款· GEO 优化 ›

一、什么是比较器？

比较器是一种可以对集合或数组中的元素按照自定义的方式进行排序的对象，它是实现了Comparator接口的类的实例，可以决定元素在集合或映射中如何被排序和存储。

比较器有两种基本类型：自然排序和定制排序。自然排序是指让要排序的类实现Comparable接口，并重写compareTo方法，来定义元素的默认排序方式。定制排序是指创建一个实现了Comparator接口的类，并重写compare方法，来定义元素的特定排序方式。定制排序可以覆盖自然排序，也可以在自然排序不适用的情况下使用。

二、TreeSet为什么重写比较器？

TreeSet是一个有序的集合，它可以按照自然排序或者定制排序来排列元素。自然排序是指让要排序的类实现Comparable接口，并重写compareTo方法，来定义元素的默认排序方式。定制排序是指创建一个实现了Comparator接口的类，并重写compare方法，来定义元素的特定排序方式。

TreeSet实现了Set接口，可以去重内容相同的元素。比如存入Integer类型的两个数据，6和6，遍历输出，只会输出一个6。但是对于自定义类，添加进TreeSet排序，会报错，因为比较器无法比较一个类。因此我们重写比较器，按照类中的指定属性进行排序

代码展示：

直接使用这个模板也是可以的，根据什么排序，就使用什么来做减法。等于0时会去重，正数会从小到大排序，负数会从大到小排序。

// 创建一个按照年龄升序排列的TreeSet
TreeSet<Person> ts = new TreeSet<>(new Comparator<Person>() {
    @Override
    public int compare(Person p1, Person p2) {
        return p1.getAge() - p2.getAge();
    }
});

三、HashSet为什么重写比较器？

HashSet是一个无序的集合，它不保证元素的排列顺序，并且不允许重复元素。HashSet是基于哈希表实现的，它通过调用元素的hashCode和equals方法来判断元素是否相等和决定元素在哈希表中的位置。

当我们添加进HashSet中相同属性的两个类，想去重时，可以选择重写hashcode方法和equal方法。（HashSet的比较器底层实现通过hashcode和equal，两个返回值都为true时，才会去重）

代码展示：

// 定义一个Person类，并重写hashCode和equals方法
class Person {
    private String name;
    private int age;

    // 省略构造方法和get/set方法

    @Override
    public int hashCode() {
        return Objects.hash(name, age);
    }

    @Override
    public boolean equals(Object obj) {
        if (this == obj) return true;
        if (obj == null || getClass() != obj.getClass()) return false;
        Person person = (Person) obj;
        return age == person.age && Objects.equals(name, person.name);
    }
}

// 创建一个HashSet对象，并添加Person对象
HashSet<Person> hs = new HashSet<>();
hs.add(new Person("张三", 20));
hs.add(new Person("李四", 21));
hs.add(new Person("张三", 20)); // 这个对象不会被添加，因为和第一个对象相等

四、Collection为什么重写比较器？

Collection是一个接口，它是Java集合框架的根接口，它定义了一些通用的操作方法，如添加、删除、遍历、排序等。Collection接口有很多实现类，如ArrayList、LinkedList、HashSet、TreeSet等。Collection接口本身并不需要重写比较器，但是它提供了一个sort方法，可以对实现了List接口的集合进行排序。该方法有两个重载形式，一个是使用自然排序，另一个是使用定制排序。

当我们想对ArryList集合进行排序时（ArryList本身无法排序，没有比较器），可以通过重写比较器来进行排序。

// 创建一个ArrayList对象，并添加一些字符串
ArrayList<String> al = new ArrayList<>();
al.add("apple");
al.add("banana");
al.add("orange");
al.add("pear");

// 使用自然排序对ArrayList进行排序
Collections.sort(al);

// 使用定制排序对ArrayList进行排序，按照字符串长度升序排列
Collections.sort(al, (s1, s2) -> s1.length() - s2.length());