阅读量:2
要自定义排序,可以通过实现Comparator接口来实现。Comparator接口有一个compare方法,可以根据自定义的排序规则来比较两个对象的大小。
以下是一个示例,自定义了一个Person类,并在TreeSet中按照年龄进行排序:
import java.util.Comparator; import java.util.TreeSet; class Person { private String name; private int age; public Person(String name, int age) { this.name = name; this.age = age; } public String getName() { return name; } public int getAge() { return age; } } class AgeComparator implements Comparator<Person> { @Override public int compare(Person p1, Person p2) { return p1.getAge() - p2.getAge(); } } public class Main { public static void main(String[] args) { TreeSet<Person> set = new TreeSet<>(new AgeComparator()); set.add(new Person("Alice", 25)); set.add(new Person("Bob", 30)); set.add(new Person("Charlie", 20)); for (Person person : set) { System.out.println(person.getName() + " - " + person.getAge()); } } }
输出结果为:
Charlie - 20 Alice - 25 Bob - 30
在这个示例中,AgeComparator类实现了Comparator接口,并重写了compare方法,根据Person对象的age属性进行比较。然后在创建TreeSet对象时,将AgeComparator对象作为参数传入,这样TreeSet会根据AgeComparator定义的排序规则来进行排序。