TreeMap详细用法及案例分析

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; }
*/

}