TreeMap的基本特点

在Ja va中，Map接口还有一个常用的实现类TreeMap。它也是用来存储键值映射关系的，并且不允许出现重复的键。

在TreeMap内部是通过二叉树的原理来保证键的唯一性，这与TreeSet集合存储的原理一样。因此，TreeMap中所有的键是按照某种顺序排列的。

接下来通过一个案例来了解TreeMap的具体用法，如文件1所示。

文件1 Example19.ja va

 1    import ja va.util.Map;
 2    import ja va.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结果分析

在文件1里，先创建了一个TreeMap集合。接着通过put()方法依次往集合中放入了3个元素，最后输出集合信息。

运行结果很直观：取出来的元素会按照键对象的自然顺序完成排序。

原因就在于，这里使用的键对象是String类型，而String类本身实现了Comparable接口。所以在默认情况下，元素就会按照自然顺序自动排好。

TreeMap的定制排序

同TreeSet集合一样，在使用TreeMap集合时，也可以通过自定义比较器Comparator的方式对所有的键进行定制排序。

接下来对文件6-19进行修改，将集合中的元素按照键对象由大到小进行排序，如文件2所示。

文件2 Example20.ja va

 1    import ja va.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结果分析

在文件2里，定义了一个用于比较String类型键对象k的比较器 CustomComparator。它在实现 compare() 方法时，直接调用了 String 对象自身的 compareTo() 方法。

关键点在于，这里返回的是“key2.compareTo(key1)”而不是常见的“key1.compareTo(key2)”。

也正因为这个顺序被反过来了，最终输出结果里的元素才会按照键对象从大到小的顺序排列。

《夸克》非常好用的免费AI浏览器

下载APP查看