📜  在Java中通过用户定义对象创建带有比较器的 TreeSet(1)

📅  最后修改于: 2023-12-03 15:23:24.744000             🧑  作者: Mango

在Java中通过用户定义对象创建带有比较器的 TreeSet

在Java中使用TreeSet可以实现有序的存储和快速查询,而使用比较器可以自定义对象的比较规则。因此,我们可以通过用户定义对象来创建具备自定义比较器的TreeSet。

步骤
  1. 创建一个用户定义对象,并在该对象中重写equals()和hashCode()方法。
public class Person {
    private String name;
    private int age;

    public Person(String name, int age) {
        this.name = name;
        this.age = age;
    }

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

    @Override
    public int hashCode() {
        return Objects.hash(name, age);
    }
}
  1. 创建一个比较器,并在该比较器中实现对用户定义对象的比较规则。
import java.util.Comparator;

public class PersonComparator implements Comparator<Person> {
    @Override
    public int compare(Person o1, Person o2) {
        if (o1.getAge() != o2.getAge()) {
            return o1.getAge() - o2.getAge();
        } else {
            return o1.getName().compareTo(o2.getName());
        }
    }
}
  1. 创建一个TreeSet对象,并通过构造方法传入比较器。
import java.util.TreeSet;

public class Main {
    public static void main(String[] args) {
        TreeSet<Person> set = new TreeSet<>(new PersonComparator());
        set.add(new Person("Tom", 20));
        set.add(new Person("Jack", 30));
        set.add(new Person("Lucy", 25));
        System.out.println(set);
    }
}
运行结果
[Person{name='Tom', age=20}, Person{name='Lucy', age=25}, Person{name='Jack', age=30}]

总结

以上是在Java中通过用户定义对象创建具备比较器的TreeSet的步骤,我们可以使用这种方法来实现对自定义对象按照一定的规则进行排序和查询。