TreeMap
TreeMap实现SortedMap接口,能够把它保存的记录根据键排序,默认是按键值的升序排序,也可以指定排序的比较器,当用Iterator遍历TreeMap时,得到的记录是排过序的。如果使用排序的映射,建议使用TreeMap。在使用TreeMap时,key必须实现Comparable接口或者在构造TreeMap传入自定义的Comparator,否则会在运行时抛出java.lang.ClassCastException类型的异常。
之前总结TreeSet类中实现排序的源码,发现是由TreeMap类进行实现的,所以在TreeMap中key如果是自定义类则需要自己定义规则来进行排序,有两种办法
在Student类中实现Comparable,重写compareTo方法
在构造函数中new Comparator,匿名内部类,重写compare 方法
以下的代码将展示第二种排序方式
public class TestTreeMap {
public static void main(String[] args) {
//1.创建集合
TreeMap<Student, String> map = new TreeMap<Student, String>(new Comparator<Student>() {
//按照年龄来排序,年龄相同按照姓名来排序
@Override
public int compare(Student o1, Student o2) {
if(o1.getAge()==o2.getAge()){
return o1.getName().compareTo(o2.getName());
}
return o1.getAge()-o2.getAge();
});
//2.创建学生对象并往集合中增加
Student s1 = new Student("张三",27);
Student s2 = new Student("李四",29);
Student s3 = new Student("王五",16);
Student s4 = new Student("张三",27);
map.put(s1, "2001");
map.put(s2, "2002");
map.put(s3, "2003");
map.put(s4, "2004");
//3.遍历集合 ,排序完成
Set<Student> set = map.keySet();
for(Student student : set){
String value = map.get(student);
System.out.println(student.getName()+"=="+student.getAge()+"=="+value);
}
}
}
附上TreeSet类的地址便于回顾:
点击打开链接