Java枚举类型：查找枚举

Question

我刚刚阅读了有关枚举的教程，并有一个问题。我学习了以下示例：

public enum Planet {
    MERCURY (3.303e+23, 2.4397e6),
    VENUS   (4.869e+24, 6.0518e6),
    EARTH   (5.976e+24, 6.37814e6),
    MARS    (6.421e+23, 3.3972e6),
    JUPITER (1.9e+27,   7.1492e7),
    SATURN  (5.688e+26, 6.0268e7),
    URANUS  (8.686e+25, 2.5559e7),
    NEPTUNE (1.024e+26, 2.4746e7),
    PLUTO   (1.27e+22,  1.137e6);

    private final double mass;   // in kilograms
    private final double radius; // in meters
    Planet(double mass, double radius) {
        this.mass = mass;
        this.radius = radius;
    }
    public double mass()   { return mass; }
    public double radius() { return radius; }

    // universal gravitational constant  (m3 kg-1 s-2)
    public static final double G = 6.67300E-11;

    public double surfaceGravity() {
        return G * mass / (radius * radius);
    }
    public double surfaceWeight(double otherMass) {
        return otherMass * surfaceGravity();
    }
}

问题：如果我知道质量和半径，如何找到例如水星的枚举类型？

谢谢。

- jitm

---

"PLUTO不是行星"，他们这样说。 - irreputable

质量和半径是唯一的键，因此您应该能够在任何一个上进行搜索。 - Peter Lawrey

http://www.petitiononline.com/bg1bk8po/petition.html - whiskeysierra

Answer 1

O(n) - 遍历所有枚举值并进行比较：

for (Planet planet : Planet.values()) {
   if (..) {..}
}

最好的位置是将此方法放在枚举类本身的 static 方法中。

- Bozho

Answer 2

给Planet枚举类型添加一个静态的search方法，该方法接受这两个事实并查找它。对于这样大小的问题，简单的线性探测策略应该足够快。

- Hank Gay

---

具体来说，一个静态搜索方法。 - StriplingWarrior

好观点；我本以为这是显而易见的，但更好的做法是明确表达。 - Hank Gay

Answer 3

对于一个enum，values()方法将返回一个包含所有enum值的数组，按照它们被声明的顺序排列。因此，您可以循环遍历该数组，寻找与您的条件匹配的Planet。

for (Planet p : Planet.values()) {
    if (p.mass() == searchMass && p.radius == searchRadius) {
      //do something with p
    }
}

enum 不太可能有大量的值，因此在性能方面通常会很好。

- David Webb

---

比较浮点数时应当小心使用等号“==”。 - Darron

Answer 4

讨论的线性搜索模式非常适合所提出的问题。然而，在枚举类增长的情况下（或者如果您正在使用Java 1.5之前的类型安全枚举，使用EnumSyntax创建运行时配置的枚举），您可能需要更快的东西。

在这种情况下，您可以定义一个静态初始化块，用值填充Map，以便通过键值对进行查找。在这种情况下，您将定义由质量和半径组成的键的Map>。

然后，您将提供一个静态方法，返回从地图中查找的结果。

在大多数情况下，这是过度设计，因为线性搜索对性能来说已经足够了。但是，如果您多次执行这些查找操作，则此解决方案会在初始化时提供一次性命中。

示例代码：

public enum Planet {
    MERCURY (3.303e+23, 2.4397e6),
    VENUS   (4.869e+24, 6.0518e6),
    EARTH   (5.976e+24, 6.37814e6),
    MARS    (6.421e+23, 3.3972e6),
    JUPITER (1.9e+27,   7.1492e7),
    SATURN  (5.688e+26, 6.0268e7),
    URANUS  (8.686e+25, 2.5559e7),
    NEPTUNE (1.024e+26, 2.4746e7),
    PLUTO   (1.27e+22,  1.137e6);

    static {
       map = new HashMap<Double, Map<Double, Planet>>();
       for (Planet p : Planet.values()) {
          if (!map.containsKey(p.getMass())) {
             p.put(p.getMass(), new HashMap<Double, Planet>());
          }
          p.get(p.getMass()).put(p.getRadius(), p));
       }
    }

    private final double mass;   // in kilograms
    private final double radius; // in meters

    private static final Map<Double, Map<Double, Planet>> map;

    Planet(double mass, double radius) {
        this.mass = mass;
        this.radius = radius;
    }
    public double mass()   { return mass; }
    public double radius() { return radius; }

    // universal gravitational constant  (m3 kg-1 s-2)
    public static final double G = 6.67300E-11;

    public double surfaceGravity() {
        return G * mass / (radius * radius);
    }
    public double surfaceWeight(double otherMass) {
        return otherMass * surfaceGravity();
    }

    public static Planet getPlanet(double mass, double radius) {
       if (map.contains(mass)) {
          return map.get(mass).get(radius);
       }
       return null;
    }
}

- John Engelman

Answer 5

你可以使用 Planet.values() 获取所有 Planet 的数组，并遍历它们，查找具有指定质量和半径的行星。

- Richard Fearn