TreeSet的自然排序和定制排序
TreeSet
- 向TreeSet中添加的元素必须时同一个类的。
- 可以按照添加进集合中的元素的指定的顺序遍历。
- 当向TreeSet中添加自定义类的对象时,有两种排序方法:
自然排序
要求自定义类实现java.lang.Comparable接口并重写其compareTo(Object object)方法,
在此方法中,指明按照自定义类的哪个属性进行排序。
代码实现(按照名字排序):
Person实体类
import java.util.Objects;
public class Person implements Comparable{
private String name;
private int age;
public Person(String name, int age) {
this.name = name;
this.age = age;
}
//此处省略空参构造器和get() set()方法
@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);
}
@Override
public String toString() {
return "name:" + name + ", age:" + age;
}
/**
* 当向treeset中添加person对象时,依据此方法,确定按照哪个属性排序
* @param o
* @return
*/
@Override
public int compareTo(Object o) {
if (o instanceof Person