本文深入探讨了java `enumset`的元素排序机制，阐明其默认行为是基于枚举常量的自然声明顺序而非插入顺序。针对需要特定排序的场景，文章提供了两种解决方案：一是使用 `list` 来简单地保持插入顺序；二是通过结合 `treeset` 和自定义 `comparator`，基于枚举内部属性实现灵活的自定义排序，确保在保留 `set` 特性的同时满足复杂的排序需求。
理解 EnumSet 的默认排序行为
EnumSet 是 Java 中专门为枚举类型设计的 Set 实现，它以位向量的形式存储枚举常量，因此效率极高且内存占用小。然而，EnumSet 的一个重要特性是其元素迭代顺序并非基于元素的插入顺序，而是严格遵循枚举类型中常量声明的自然顺序。这意味着，无论您以何种顺序将枚举常量添加到 EnumSet 中，当您迭代或访问其元素时，它们总是会按照枚举定义时的顺序出现。
考虑以下示例，即使我们以非自然顺序添加 DayOfWeek 枚举常量，输出仍然是其在 DayOfWeek 枚举中声明的自然顺序：
import java.time.DayOfWeek;
import java.util.EnumSet;
public class EnumSetOrderExample {
public static void main(String[] args) {
EnumSet<DayOfWeek> days = EnumSet.of(DayOfWeek.SATURDAY, DayOfWeek.FRIDAY, DayOfWeek.THURSDAY);
System.out.println("EnumSet 迭代顺序:");
days.forEach(System.out::println);
}
}输出:
THURSDAY FRIDAY SATURDAY
从输出可以看出，尽管 SATURDAY 最先被传入 EnumSet.of() 方法，但最终的迭代顺序是 THURSDAY, FRIDAY, SATURDAY，这正是 DayOfWeek 枚举常量在定义时的自然顺序。
立即学习“Java免费学习笔记（深入）”；
保持插入顺序的替代方案：使用 List
如果您的核心需求是按照元素添加的顺序进行迭代，并且对 Set 的唯一性（即不允许重复元素）要求不那么严格，或者您可以在添加前自行处理重复性，那么使用 List 是一个更直接且简单的解决方案。List 接口的实现（如 ArrayList）天然地保持元素的插入顺序。
以下示例展示了如何使用 List 来保持元素的插入顺序：
import java.time.DayOfWeek;
import java.util.List;
public class ListOrderExample {
public static void main(String[] args) {
List<DayOfWeek> daysInOrder = List.of(DayOfWeek.SATURDAY, DayOfWeek.FRIDAY, DayOfWeek.THURSDAY);
System.out.println("List 迭代顺序 (保持插入顺序):");
daysInOrder.forEach(System.out::println);
}
}输出:
SATURDAY FRIDAY THURSDAY
通过将 EnumSet 替换为 List，我们成功地按照元素被添加的顺序获取了它们。
实现自定义排序：结合 TreeSet 与 Comparator
在某些场景下，您可能既需要 Set 的唯一性特性，又需要一种不同于枚举自然顺序或插入顺序的自定义排序。例如，您可能希望根据枚举常量的一个特定属性（如优先级、数值等）进行排序。在这种情况下，TreeSet 结合自定义 Comparator 是一个强大的解决方案。
TreeSet 是一个有序的 Set 实现，它使用红黑树结构来存储元素，并根据元素的自然顺序或构造时提供的 Comparator 进行排序。
步骤一：为枚举添加排序属性
首先，我们需要在枚举类型中定义一个用于排序的属性。例如，我们可以添加一个 order 字段。
public enum MyEnum {
ONE(3),
TWO(2),
THREE(1);
private final int order;
MyEnum(int order) {
this.order = order;
}
public int getOrder() {
return order;
}
}在这个 MyEnum 中，我们为每个枚举常量分配了一个整数 order，表示我们期望的排序优先级（数字越小，优先级越高）。
步骤二：使用 TreeSet 和自定义 Comparator 进行排序
接下来，我们可以创建一个 TreeSet，并在其构造函数中提供一个 Comparator，该 Comparator 将根据 MyEnum 的 order 属性进行比较。
import java.util.Comparator;
import java.util.Set;
import java.util.TreeSet;
import java.util.stream.Collectors;
import java.util.stream.Stream;
public class CustomOrderWithTreeSetExample {
public static void main(String[] args) {
Set<MyEnum> customOrderedSet = Stream.of(MyEnum.ONE, MyEnum.TWO, MyEnum.THREE)
.collect(Collectors.toCollection(
() -> new TreeSet<>(Comparator.comparing(MyEnum::getOrder))
));
System.out.println("TreeSet 自定义排序顺序:");
customOrderedSet.forEach(System.out::println);
}
}输出:
THREE TWO ONE
通过这种方式，我们成功地创建了一个 Set，其元素按照我们自定义的 order 属性进行了排序，而不是枚举的自然顺序或插入顺序。
注意事项与总结
在选择适合的数据结构时，理解其核心特性至关重要。
- EnumSet: 适用于只需要枚举常量自然顺序，且追求极致性能和内存效率的场景。它不保留插入顺序，也不支持自定义排序（除非您重新定义枚举常量的声明顺序）。
- List: 当需要严格保持元素的插入顺序时，List 是最简单直接的选择。它不提供 Set 的唯一性保证，需要手动处理重复元素。
- TreeSet + Comparator: 当您既需要 Set 的唯一性，又需要一种不同于自然顺序或插入顺序的自定义排序时，TreeSet 结合自定义 Comparator 提供了强大的灵活性。但相比 EnumSet，TreeSet 会有更高的性能开销（由于其基于树的实现）。
根据您的具体需求和对性能、内存以及排序复杂度的权衡，选择最合适的数据结构将使您的代码更加健壮和高效。
