java treemap用法_Java的TreeMap集合

在JDK中，Map接口还有一个常用的实现类TreeMap。TreeMap集合是用来存储键值映射关系的，其中不允许出现重复的键。在TreeMap中是通过二叉树的原理来保证键的唯一性，这与TreeSet集合存储的原理一样，因此TreeMap中所有的键是按照某种顺序排列的。接下来通过一个案例来了解TreeMap的具体用法，如例所示。import java.util.Iterator;

import java.util.Set;

import java.util.TreeMap;

public class Example {

public static void main(String[] args) {

TreeMap tm = new TreeMap();

tm.put("1"， "Jack");

tm.put("2"， "Rose");

tm.put("3"， "Lucy");

Set keySet = tm.keySet(); // 获取键的集合

Iterator it = keySet.iterator(); // 获取Iterator 对象

while (it.hasNext()) { // 判断是否存在下一个元素

Object key = it.next(); // 取出元素

Object value = tm.get(key); // 根据获取的键找到对应的值

System.out.println(key + ":" + value);

}

}

}

运行结果：1:Jack

2:Rose

3:Lucy

在例中，使用put()方法将3个学生的信息存入TreeMap集合，其中学号作为键，姓名作为值，然后对学生信息进行遍历。从运行结果可以看出，取出的元素按照学号的自然顺序进行了排序，这是因为学号是String类型，String类实现了Comparable接口，因此默认会按照自然顺序进行排序。

在使用TreeMap集合时，也可以通过自定义比较器的方式对所有的键进行排序。接下来通过一个案例将学生对象按照学号由大到小的顺序进行排序，如例所示。import java.util.Comparator;

import java.util.Iterator;

import java.util.Set;

import java.util.TreeMap;

public class Example {

public static void main(String[] args) {

TreeMap tm = new TreeMap(new MyComparator()); // 传入一个自定义比较器

tm.put("1"， "Jack"); // 向集合存入学生的学号和姓名

tm.put("2"， "Rose");

tm.put("3"， "Lucy");

Set keySet = tm.keySet(); // 获取键的集合

Iterator it = keySet.iterator(); // 获得迭代器对象

while (it.hasNext()) {

Object key = it.next(); // 获得一个键

Object value = tm.get(key); // 获得键对应的值

System.out.println(key + ":" + value);

}

}

}

class MyComparator implements Comparator { // 自定义比较器

public int compare(Object obj1， Object obj2) { // 实现比较方法

String id1 = (String) obj1; // 将Object 类型的参数强转为String 类型

String id2 = (String) obj2;

return id2.compareTo(id1); // 将比较之后的值返回

}

}

运行结果：3:Lucy

2:Rose

1:Jack

例中定义了比较器MyComparator针对String类型的id进行比较，在实现compare()方法时，调用了String 对象的compareTo()方法。由于方法中返回的是“id2.compareTo(id1)”，因此最终输出结果中的id按照与字典顺序相反的顺序进行了排序。