在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按照与字典顺序相反的顺序进行了排序。