import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;
public class TreeSetDemo {
public static void main(String[] args) {
TreeSet<Person1>pts=new TreeSet<Person1>(new Comparator<Person1>() {
@Override//通过使用Comparator接口来实现比较的方法也是可以的
public int compare(Person1 o1, Person1 o2) {
if (o1.getAge()-o2.getAge()>0) {
return 1;
}else if (o1.getAge()-o2.getAge()<0) {
return -1;
}
return 0;
}
});
pts.add(new Person1("zhangsan", 15));
pts.add(new Person1("lisi", 18));
pts.add(new Person1("wangwu", 20));
pts.add(new Person1("rose", 19));
//在实现comparable接口之前输出会报错ClassCastException
System.out.println(pts);
Iterator<Person1> it=pts.iterator();
while (it.hasNext()) {
Person1 person1 = (Person1) it.next();
System.out.println(person1.getName()+">>>"+person1.getAge());
}
}
}
//通过实现Comparable接口的方法来比较,也可以不用这个方法,使用comparator
class Person1 /*implements Comparable<Person1>*/{
private String name ;
private int age;
public Person1(String name, int age) {
super();
this.name = name;
this.age = age;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public int getAge() {
return age;
}
public void setAge(int age) {
this.age = age;
}
@Override
public String toString() {
return "Person1 [name=" + name + ", age=" + age + "]";
}
/*@Override
public int compareTo(Person1 o) {
if (this.age-o.getAge()>0) {
return 1;
}else if (this.age-o.getAge()<0) {
return -1;
}
return 0;
}
*/
}