编程那点事编程那点事

专注编程入门及提高
探究程序员职业规划之道!

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

未经允许不得转载: 技术文章 » Java编程 » Java的TreeMap集合