在Java集合体系中,EnumSet和EnumMap是专门为枚举类型设计的集合实现,它们充分利用了枚举的特性,在性能和易用性上都优于普通的HashSet、HashMap等集合类,非常适合处理枚举相关的批量数据操作场景。

EnumSet的基本使用
EnumSet是一个专门为枚举类型设计的Set实现,内部采用位向量存储,因此占用内存小、操作速度快,不允许添加null元素。
创建EnumSet实例
EnumSet提供了多个静态工厂方法来创建实例,最常用的是noneOf、allOf和of方法。
首先定义一个测试用的枚举类型:
// 定义季节枚举
enum Season {
SPRING, SUMMER, AUTUMN, WINTER
}
创建EnumSet的示例代码如下:
import java.util.EnumSet;
public class EnumSetDemo {
public static void main(String[] args) {
// 创建一个包含Season所有枚举值的EnumSet
EnumSet<Season> allSeasons = EnumSet.allOf(Season.class);
System.out.println("所有季节:" + allSeasons);
// 创建一个空的EnumSet,指定元素类型为Season
EnumSet<Season> emptySeasons = EnumSet.noneOf(Season.class);
System.out.println("空集合:" + emptySeasons);
// 创建包含指定枚举值的EnumSet
EnumSet<Season> someSeasons = EnumSet.of(Season.SPRING, Season.AUTUMN);
System.out.println("指定季节:" + someSeasons);
// 创建指定范围枚举值的EnumSet,包含起始和结束枚举
EnumSet<Season> rangeSeasons = EnumSet.range(Season.SPRING, Season.AUTUMN);
System.out.println("范围季节:" + rangeSeasons);
}
}
EnumSet的常用操作
EnumSet支持常规的Set操作,比如添加、删除、包含判断等,同时也提供了补集、并集等批量操作方法。
import java.util.EnumSet;
public class EnumSetOperationDemo {
public static void main(String[] args) {
EnumSet<Season> set1 = EnumSet.of(Season.SPRING, Season.SUMMER);
EnumSet<Season> set2 = EnumSet.of(Season.SUMMER, Season.AUTUMN);
// 判断包含元素
System.out.println("set1是否包含SPRING:" + set1.contains(Season.SPRING));
// 添加元素
set1.add(Season.WINTER);
System.out.println("添加WINTER后的set1:" + set1);
// 求并集
set1.addAll(set2);
System.out.println("set1和set2的并集:" + set1);
// 求交集
EnumSet<Season> intersection = EnumSet.copyOf(set1);
intersection.retainAll(set2);
System.out.println("set1和set2的交集:" + intersection);
// 求补集(set1中不包含在set2里的元素)
EnumSet<Season> complement = EnumSet.complementOf(set2);
System.out.println("set2的补集:" + complement);
}
}
EnumMap的基本使用
EnumMap是一个专门为枚举类型作为键设计的Map实现,内部采用数组存储,因此查询、插入、删除操作的时间复杂度都是O(1),同样不允许键为null,但值可以为null。
创建EnumMap实例
EnumMap的构造方法需要指定键对应的枚举类型,示例代码如下:
import java.util.EnumMap;
import java.util.Map;
public class EnumMapDemo {
public static void main(String[] args) {
// 创建键为Season类型的EnumMap
EnumMap<Season, String> seasonDescMap = new EnumMap<>(Season.class);
// 添加键值对
seasonDescMap.put(Season.SPRING, "温暖多雨");
seasonDescMap.put(Season.SUMMER, "炎热干燥");
seasonDescMap.put(Season.AUTUMN, "凉爽宜人");
seasonDescMap.put(Season.WINTER, "寒冷少雨");
// 遍历EnumMap
for (Map.Entry<Season, String> entry : seasonDescMap.entrySet()) {
System.out.println(entry.getKey() + "的描述:" + entry.getValue());
}
// 获取指定键的值
System.out.println("SPRING的描述:" + seasonDescMap.get(Season.SPRING));
}
}
EnumMap的适用场景
当需要将枚举值和其他数据关联时,EnumMap是比HashMap更好的选择,比如状态机、枚举对应的配置信息等场景。例如下面的状态转换场景:
import java.util.EnumMap;
// 定义订单状态枚举
enum OrderStatus {
CREATED, PAID, SHIPPED, FINISHED, CANCELLED
}
public class OrderStatusDemo {
public static void main(String[] args) {
// 定义订单状态对应的可操作状态映射
EnumMap<OrderStatus, EnumSet<OrderStatus>> statusTransitionMap = new EnumMap<>(OrderStatus.class);
statusTransitionMap.put(OrderStatus.CREATED, EnumSet.of(OrderStatus.PAID, OrderStatus.CANCELLED));
statusTransitionMap.put(OrderStatus.PAID, EnumSet.of(OrderStatus.SHIPPED, OrderStatus.CANCELLED));
statusTransitionMap.put(OrderStatus.SHIPPED, EnumSet.of(OrderStatus.FINISHED));
// 判断CREATED状态可以转换到哪些状态
EnumSet<OrderStatus> availableStatus = statusTransitionMap.get(OrderStatus.CREATED);
System.out.println("已创建订单可转换为:" + availableStatus);
}
}
EnumSet和EnumMap的优势
和普通集合相比,EnumSet和EnumMap主要有以下优势:
- 性能更高:内部采用位向量和数组存储,避免了哈希计算的开销,操作速度更快
- 内存占用更小:不需要存储额外的哈希表结构,内存利用率更高
- 类型安全:编译期就能检查键或元素的类型,避免类型转换错误
- 使用便捷:提供了针对枚举的专用工厂方法,代码更简洁
使用注意事项
在使用这两个类时需要注意以下几点:
- EnumSet和EnumMap都不允许键或元素为null,添加null会抛出NullPointerException
- EnumSet的迭代顺序是枚举定义的顺序,不会发生变化
- EnumMap的迭代顺序也是键枚举的定义顺序
- 如果需要存储大量非枚举类型的元素,还是应该选择普通的集合类