java - 有效地将元素添加到列表的顶部

Question

我有一个像这样的 ENUM，我总是从中得到我的 localFruit ，它可以是 APPLE 或 ORANGE 或 香蕉。

public enum Fruits {
    // it can have more elements here
    APPLE, ORANGE, BANANA;

    // some code
}

假设 APPLE 是我的 localFruit，那么 ORANGE 和 BANANA 将是我的 remoteFruits。我需要打乱我的 remoteFruits，然后确保我的 localFruit 位于我列表的顶部，然后是 remoteFruits。

下面是我的代码，我在其中进行洗牌并将其添加到原始 result 列表中:在下面的代码中，CURRENT_FRUIT 可以是 APPLE 或ORANGE 或 BANANA。

private static List<Fruits> getFruitsInOrder() {
    EnumSet<Fruits> localFruit = EnumSet.of(CURRENT_FRUIT);
    EnumSet<Fruits> remoteFruits = EnumSet.complementOf(localFruit);

    List<Fruits> result = new ArrayList<Fruits>(remoteFruits);
    Collections.shuffle(result);

    // first element in the list will always be the local fruit
    result.addAll(0, new ArrayList<Fruits>(localFruit));
    return result;
}

由于这段代码会被调用很多次，所以想看看我在做的事情是否有问题，这可能是性能瓶颈？我的代码在性能方面是否还可以？

我的主要目标是让 localFruit 位于列表的顶部，然后是 remoteFruits(但是我需要在添加到结果列表之前将它们打乱顺序).

Answer 1

所有这些解决方案都太难了。 Java 集非常高效，但简单的数组访问更是如此。此外，构建两个集合(一个具有补码操作)、复制到 ArrayList，然后使用 addAll 创建一个新的 ArrayList 是大量无用的工作和内存垃圾。

枚举按顺序为您提供其值的数组。用它!只需将本地元素交换到位置零，然后打乱数组的其余部分。这样您就创建了一个数据结构:您要返回的 ArrayList。剩下的只是重新排序它的元素。

当然，除非你有一个巨大的枚举或者正在调用这个函数数百万次，否则这个讨论是学术性的。您不会注意到性能差异。

import java.util.Arrays;
import java.util.Collections;
import java.util.List;

enum Fruit { APPLE, PEAR, PEACH, PLUM, BANANA }

public class Hack {

    static List<Fruit> getFruitInOrder(Fruit local) {
        List<Fruit> list = Arrays.asList(Fruit.values());
        Collections.swap(list, 0, local.ordinal());
        Collections.shuffle(list.subList(1, list.size()));
        return list;
    }

    public static void main(String[] args) {
        for (int i = 0; i < 10; i++) {
            System.out.println(getFruitInOrder(Fruit.PLUM));
        }
    }
}

在我的 MacBook 上:

run:
[PLUM, BANANA, PEAR, APPLE, PEACH]
[PLUM, PEACH, PEAR, APPLE, BANANA]
[PLUM, PEACH, BANANA, PEAR, APPLE]
[PLUM, PEAR, BANANA, APPLE, PEACH]
[PLUM, PEAR, APPLE, BANANA, PEACH]
[PLUM, BANANA, APPLE, PEACH, PEAR]
[PLUM, APPLE, BANANA, PEACH, PEAR]
[PLUM, APPLE, PEACH, PEAR, BANANA]
[PLUM, APPLE, PEAR, PEACH, BANANA]
[PLUM, PEACH, APPLE, PEAR, BANANA]
BUILD SUCCESSFUL (total time: 0 seconds)