Java EnumSet 元素顺序控制：深入理解与自定义排序策略_技术教程

本文深入探讨了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 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 枚举常量在定义时的自然顺序。

保持插入顺序的替代方案：使用 List

如果您的核心需求是按照元素添加的顺序进行迭代，并且对 Set 的唯一性（即不允许重复元素）要求不那么严格，或者您可以在添加前自行处理重复性，那么使用 List 是一个更直接且简单的解决方案。List 接口的实现（如 ArrayList）天然地保持元素的插入顺序。

以下示例展示了如何使用 List 来保持元素的插入顺序：

import java.time.DayOfWeek;
import java.util.List;

public class ListOrderExample {
    public static void main(String[] args) {
        List 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 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 会有更高的性能开销（由于其基于树的实现）。

根据您的具体需求和对性能、内存以及排序复杂度的权衡，选择最合适的数据结构将使您的代码更加健壮和高效。