📅  最后修改于: 2023-12-03 15:23:24.744000             🧑  作者: Mango
在Java中使用TreeSet可以实现有序的存储和快速查询,而使用比较器可以自定义对象的比较规则。因此,我们可以通过用户定义对象来创建具备自定义比较器的TreeSet。
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);
}
}
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());
}
}
}
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的步骤,我们可以使用这种方法来实现对自定义对象按照一定的规则进行排序和查询。