TreeMap及常用API
① TreeMap类通过使用红黑树实现Map接口;
② TreeMap提供按排序顺序存储键/值对的有效手段，同时允许快速检索;
③ 不像散列(HashMap)，树映射保证它的元素按关键字升序排序;
④ TreeMap构造方法：
a) TreeMap()
b) TreeMap(Comparator comp)
c) TreeMap(Map m)
d) TreeMap(SortedMap sm)
⑤ TreeMap实现SortedMap并且扩展AbstractMap，它本身并没有定义其他方法;
TreeMap<String,String> tmap=new TreeMap<String, String>();
tmap.put("zhang", "张三");
tmap.put("jack", "小明");
tmap.put("mary", "小红");
tmap.put("free", "小叶");
tmap.put("mary", "小草");
//tmap.put(null,"小草");//键不能传入null，会抛异常
System.out.println(tmap);
默认按照键的自然顺序升序输出
输出结果：
{free=小叶, jack=小明, mary=小草, zhang=张三}
输出所有键值对
Set<Entry<String,String>> entrys=tmap.entrySet();
for(Entry<String,String> entry:entrys){
System.out.println(entry.getKey()+"--"+entry.getValue());
}
输出结果：
free--小叶
jack--小明
mary--小草
zhang--张三
编写一个Person类
class Person{
private String name;
private int age;
public Person(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;
}
}
在主方法中创建并添加元素，输出
TreeMap<Person,String> pdata=new TreeMap<Person, String>();
pdata.put(new Person("zhangsan",20), "张三");
pdata.put(new Person("lisi",25), "李四");
pdata.put(new Person("wangwu",30), "王五");
pdata.put(new Person("zhangsan",33), "张三");
System.out.println(pdata);
运行结果：
出错，原因是Person类中没有实现compareTo(T o)比较方法
方法一：实现Comparable接口
class Person implements Comparable<Person>{ }
重写方法如下：按年龄进行排序
@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;
}
然后再执行，输出结果：
{com.iotek.map.Person@2a139a55=张三, com.iotek.map.Person@15db9742=李四, com.iotek.map.Person@6d06d69c=王五, com.iotek.map.Person@7852e922=张三}
完整的Person类如下:
class Person implements Comparable<Person> {
private String name;
private int age;
public Person(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 int compareTo(Person o) {
if (this.age - o.getAge() > 0) {
return 1;
} else if (this.age - o.getAge() < 0) {
return -1;
}
return 0;
}
}
方法二：Person中不使用Comparable接口，在主方法中使用匿名内部类：
TreeMap<Person, String> pdata = new TreeMap<Person, String>(new Comparator<Person>() {
@Override
public int compare(Person o1, Person o2) {
if(o1.getAge()>o2.getAge()){
return 1;
}
else if(o1.getAge()<o2.getAge()){
return -1;
}
return 0;
}
});
输出结果与上面相同
上面都是使用int年龄进行排序，使用String名字排序方法如下：
@Override
public int compare(Person o1, Person o2) {
return o1.getName().compareTo(o2.getName());
}
输出结果：
{com.iotek.map.Person@2a139a55=李四, com.iotek.map.Person@15db9742=王五, com.iotek.map.Person@6d06d69c=张三}
因为zhangsan相同则被替换了，修改如下：
@Override
public int compare(Person o1, Person o2) {
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();
}
}
Comparator和Comparable接口
① TreeMap的key存储引用类型数据，需要满足一定条件
a) 要么引用类型实现Comparable接口
b) 要么为该TreeMap容器提供实现Comparator接口的比较器对象