TreeMap集合

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

文件1 Example19.java

 1    import java.util.Map;
 2    import java.util.TreeMap;
 3    public class Example19 {                  
 4        public static void main(String[] args) {
 5            Map map = new TreeMap();   
 6            map.put("2", "Rose");
 7            map.put("1", "Jack");
 8            map.put("3", "Lucy");
 9            System.out.println(map);
 10        }
 11    }

运行结果如图1所示。

图1 运行结果

在文件1中，首先创建了一个TreeMap集合，并使用put()方法按顺序向集合中添加了3个元素然后打印出集合信息。从运行结果可以看出，取出的元素按照键对象的自然顺序进行了排序，这是因为添加的元素中键对象是String类型，String类实现了Comparable接口，因此默认会按照自然顺序对元素进行排序。

同TreeSet集合一样，在使用TreeMap集合时，也可以通过自定义比较器Comparator的方式对所有的键进行定制排序。接下来对文件6-19进行修改，将集合中的元素按照键对象由大到小进行排序，如文件2所示。

文件2 Example20.java

 1    import java.util.*;
 2    // 自定义比较器
 3    class CustomComparator implements Comparator { 
 4        public int compare(Object obj1, Object obj2) {
 5            String key1 = (String) obj1;  
 6            String key2 = (String) obj2;
 7            return key2.compareTo(key1);  // 将比较之后的值返回
 8        }
 9    }
 10    public class Example20 {
 11        public static void main(String[] args) {
 12            Map map = new TreeMap(new CustomComparator());  
 13            map.put("2", "Rose");
 14            map.put("1", "Jack");
 15            map.put("3", "Lucy");
 16            System.out.println(map);
 17        }
 18    }

运行结果如图2所示。

图2 运行结果

文件2中定义了比较器CustomComparator针对String类型的键对象k进行比较，在实现compare()方法时，调用了String对象的compareTo()方法。由于方法中返回的是“key2.compareTo(key1)”，因此最终输出结果中的元素按照键对象从大到小的顺序进行了排序。