Java TreeMap/TreeSet

首先要说明的是，TreeMap仍然是Map，而不是Tree。同样的，TreeSet仍然是Set，而不是Tree。也就是说TreeMap和TreeSet不能当成树来使用，只是它们的底层是使用红黑树来实现的。

TreeMap实现了java.util.SortedMap接口，是一个已经排好序的Map。

TreeMap映射根据其键的自然顺序进行排序，或者根据创建映射时提供的 Comparator 进行排序，具体取决于使用的构造方法。

有两种方式使用TreeMap。

1.元素自然排序，即放入容器中的元素实现java.lang.Comparable接口。

package org.zzj;

import java.util.TreeMap;

public class TreeMapTest {

public static void main(String[] args) {
TreeMap<Person, String> treeMap = new TreeMap<Person, String>();

Person p1 = new Person("张三", 21);
Person p2 = new Person("李四", 15);
Person p3 = new Person("孙五", 26);
Person p4 = new Person("赵六", 18);

treeMap.put(p1, "");
treeMap.put(p2, "");
treeMap.put(p3, "");
treeMap.put(p4, "");

System.out.println(treeMap);

p3.setAge(12);

System.out.println(treeMap);
}
}

class Person implements Comparable<Person> {
private String name;
private int age;

public Person() {
}

public Person(String name, int age) {
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) {
return this.age - o.age;
}

@Override
public String toString() {
return "name:" + name + "; age:" + age + "\n";
}

}

输出

{name:李四; age:15
=, name:赵六; age:18
=, name:张三; age:21
=, name:孙五; age:26
=}
{name:李四; age:15
=, name:赵六; age:18
=, name:张三; age:21
=, name:孙五; age:12
=}

当把孙五的年龄改为最小的时候，他仍然排在了最后，这说明元素的顺序在放入容器的时候就已经确定了。

2.使用排序器

package org.zzj;

import java.util.Comparator;
import java.util.TreeMap;

public class TreeMapTest {

public static void main(String[] args) {
TreeMap<Person, String> treeMap = new TreeMap<Person, String>(
new Comparator<Person>() {

@Override
public int compare(Person o1, Person o2) {
return o1.getAge() - o2.getAge();
}
});

Person p1 = new Person("张三", 21);
Person p2 = new Person("李四", 15);
Person p3 = new Person("孙五", 26);
Person p4 = new Person("赵六", 18);

treeMap.put(p1, "");
treeMap.put(p2, "");
treeMap.put(p3, "");
treeMap.put(p4, "");

System.out.println(treeMap);

p3.setAge(12);

System.out.println(treeMap);
}
}

class Person {
private String name;
private int age;

public Person() {
}

public Person(String name, int age) {
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 "name:" + name + "; age:" + age + "\n";
}

}

输出：

{name:李四; age:15
=, name:赵六; age:18
=, name:张三; age:21
=, name:孙五; age:26
=}
{name:李四; age:15
=, name:赵六; age:18
=, name:张三; age:21
=, name:孙五; age:12
=}

如果既没有让元素实现java.lang.Comparable接口，也没有在TreeMap的构造器中传入排序器，则会在添加元素的抛出错误。

显然，使用排序器有利于解耦。

HashMap通常比TreeMap快一些(树和哈希表的数据结构使然)，建议多使用HashMap，在需要排序的Map时候才用TreeMap。

类似于HashSet，TreeSet也是基于TreeMap实现的。