import java.util.Comparator;
import java.util.TreeMap;
public class TreeMapDemo2 {
public static void main(String[] args) {
TreeMap<Person, String> pdata = new TreeMap<Person, String>(new Comparator<Person>() {
@Override // 按照名字进行比较
public int compare(Person o1, Person o2) {
/* return o1.getName().compareTo(o2.getName());
* //这是比较一个属性,会产生名字相同时替换的情况*/
if (o1.getName().compareTo(o2.getName())>0) {
return 1;
}else if (o1.getName().compareTo(o2.getName())<0) {
return -1;
}else {
//如果名字一样则按照年龄来比较
return o1.getAge()-o2.getAge();
}
}
/*
* @Override//按照年龄比较也可以按照名字比较 public int compare(Person o1, Person
* o2) { if (o1.getAge()-o2.getAge()>0) { return 1; }else if
* (o1.getAge()-o2.getAge()<0) { return -1; } return 0; }
*/
});
// 如果Person类中不实现Comparable这个比较方法,这不能够正常运行,因为TreeMap是比较键的
pdata.put(new Person("zhangsan", 21), "张三");
pdata.put(new Person("lisi", 31), "李四");
pdata.put(new Person("rose", 32), "玫瑰");
pdata.put(new Person("zhangsan", 33), "张三");
// 当比较一个属性时假设两个名字一样则会被替换
System.out.println(pdata);
}
}
// 声明一个Person类来实现Comparable的方法
class Person /* implements Comparable<Person> */ {// 定义两个变量
private String name;
private int age;
// 构造方法
public Person(String name, int age) {
super();
this.name = name;
this.age = age;
}
// 实现get和set方法
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 "Person [name=" + name + ", age=" + age + "]";
}
/*
* //自定义的比较方法,具体的比较规则自己定
*
* @Override public int compareTo(Person o) { if (this.age-o.getAge()>0) {
* return 1; }else if (this.age-o.getAge()<0) { return -1; } return 0; }
*/
}