怎么让TreeSet根据对象指定属性排序
时间: 2023-08-11 12:03:45 浏览: 96
java 中文姓氏 排序
5星 · 资源好评率100%
可以通过实现 `Comparable` 接口或者传入一个比较器 `Comparator` 来指定 TreeSet 的排序方式。以下是使用比较器的示例代码:
假设有一个 `Person` 类,其中有两个属性 `name` 和 `age`,我们想按照年龄从小到大排序:
```java
import java.util.Comparator;
import java.util.TreeSet;
public class Example {
public static void main(String[] args) {
Comparator<Person> comparator = new Comparator<Person>() {
@Override
public int compare(Person p1, Person p2) {
return p1.getAge() - p2.getAge();
}
};
TreeSet<Person> set = new TreeSet<>(comparator);
set.add(new Person("Alice", 20));
set.add(new Person("Bob", 18));
set.add(new Person("Charlie", 25));
System.out.println(set); // 输出 [Bob(18), Alice(20), Charlie(25)]
}
}
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;
}
@Override
public String toString() {
return name + "(" + age + ")";
}
}
```
在上面的代码中,我们首先创建了一个比较器对象 `comparator`,它比较两个 `Person` 对象的年龄。然后创建了一个 TreeSet 对象 `set`,并传入比较器对象 `comparator`。最后向 `set` 中添加三个 `Person` 对象,并输出 `set` 的内容。可以看到输出的结果是按照年龄从小到大排序的。
阅读全文