当前位置：首页 > news >正文

Java 集合 - 用好 SortedMap 和 NavigableMap，优化 Java 集合排序与操作效率

news 2026/6/23 16:05:03

在Java中，SortedMap和NavigableMap接口为Map提供了对键排序的功能。这些接口有助于管理那些需要有序存储和按特定顺序访问键值对的情况。它们通过TreeMap类的实现来工作，TreeMap是一种自平衡的红黑树数据结构，它保证了键的顺序。

SortedMap是Map接口的扩展，而NavigableMap是SortedMap的一个进一步扩展。这两个接口的实现都可以使用TreeMap类。它们都能确保键值对按照键进行排序。

`SortedMap`和`NavigableMap`的功能

SortedMap和NavigableMap接口通过键值对的排序，为Map提供了额外的功能。下面是它们的一些重要方法和使用方法。

方法概述

SortedMap接口通过以下方法扩展了Map：

firstKey()和lastKey()：返回Map中最小和最大键。
headMap(toKey)：返回一个新的SortedMap，其键小于toKey。
tailMap(fromKey)：返回一个新的SortedMap，其键大于或等于fromKey。
subMap(fromKey, toKey)：返回一个新的SortedMap，其键大于或等于fromKey，且小于toKey。

这些方法返回的SortedMap是原始Map的视图，任何对这些视图的更改都会反映在原始Map中。值得注意的是，这些视图不能插入超出原始Map范围的键。

`NavigableMap`的附加功能

NavigableMap接口提供了更多的功能，包括方法navigableKeySet()，它返回一个NavigableSet，而不是普通的keySet()。NavigableSet提供了更多的操作，例如返回下一个和上一个键。

示例：`SortedMap`的使用

让我们通过一个例子来看看SortedMap和其方法如何工作：

SortedMap<Integer, String> map = new TreeMap<>(); map.put(1, "one"); map.put(2, "two"); map.put(3, "three"); map.put(5, "five"); map.put(6, "six"); // 获取最小键和最大键 System.out.println("First Key: " + map.firstKey()); // 输出 1 System.out.println("Last Key: " + map.lastKey()); // 输出 6 // 获取键小于3的部分 SortedMap<Integer, String> headMap = map.headMap(3); headMap.forEach((key, value) -> System.out.println(key + " :: " + value)); // 获取键大于或等于3的部分 SortedMap<Integer, String> tailMap = map.tailMap(3); tailMap.forEach((key, value) -> System.out.println(key + " :: " + value)); // 获取键在3到5之间的部分 SortedMap<Integer, String> subMap = map.subMap(3, 6); subMap.forEach((key, value) -> System.out.println(key + " :: " + value));

运行结果：

First Key: 1 Last Key: 6 1 :: one 2 :: two 3 :: three 3 :: three 5 :: five 6 :: six 3 :: three 4 :: four 5 :: five

错误处理示例

SortedMap和NavigableMap返回的视图在修改时有一些限制。比如，我们通过headMap()方法创建了一个视图，并试图向这个视图插入一个不在范围内的键，这会导致IllegalArgumentException异常。

SortedMap<Integer, String> headMap = map.headMap(3); headMap.put(0, "zero"); // 这个是合法的，因为 0 < 3 headMap.put(4, "four"); // 这个会抛出 IllegalArgumentException 异常，因为 4 不在 headMap 范围内

解释：

headMap(3)返回一个只包含小于3的键的SortedMap视图。因此，向该视图插入键值对(4, "four")会抛出异常，因为4超出了headMap的键范围。
但插入(0, "zero")是合法的，因为0小于3。

通过`Comparator`控制排序

与SortedSet和NavigableSet类似，SortedMap和NavigableMap的键必须实现Comparable，或者你可以在创建TreeMap时提供一个Comparator，来决定键的排序方式。如果提供了Comparator，即使键实现了Comparable，Comparator也会被优先使用。

TreeMap<Integer, String> map = new TreeMap<>(Comparator.reverseOrder()); map.put(1, "one"); map.put(2, "two"); map.put(3, "three"); map.forEach((key, value) -> System.out.println(key + " :: " + value));