java treeset重写
时间: 2023-11-12 09:56:27 浏览: 43
在Java中,如果你想要在TreeSet中使用自定义对象,你需要重写对象的compareTo()方法。这个方法用于比较两个对象的顺序。下面是一个示例:
```java
import java.util.TreeSet;
class Person implements Comparable<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 int compareTo(Person other) {
// 先按照姓名排序
int nameComparison = this.name.compareTo(other.name);
if (nameComparison != 0) {
return nameComparison;
}
// 如果姓名相同,则按照年龄排序
return Integer.compare(this.age, other.age);
}
}
public class Main {
public static void main(String[] args) {
TreeSet<Person> people = new TreeSet<>();
people.add(new Person("Alice", 25));
people.add(new Person("Bob", 30));
people.add(new Person("Charlie", 20));
for (Person person : people) {
System.out.println(person.getName() + " - " + person.getAge());
}
}
}
```
输出结果:
```
Alice - 25
Bob - 30
Charlie - 20
```
在上面的示例中,我们创建了一个Person类,它实现了Comparable接口,并重写了compareTo()方法。这个方法首先按照姓名进行比较,如果姓名相同,则按照年龄进行比较。通过重写compareTo()方法,我们可以定义自己的排序规则。