首页  >  文章  >  Java  >  如何用Java实现加权随机选择?

如何用Java实现加权随机选择?

Patricia Arquette
Patricia Arquette原创
2024-11-08 08:05:01298浏览

How to Implement Weighted Random Selection in Java?

Java 中的加权随机选择

从集合中选择一个项目时,通常需要为不同的项目分配不同的概率。此方法称为加权随机选择。

考虑以下场景,其中您有一组具有关联权重的项目:

Item Weight
Sword of Misery 10
Shield of Happy 5
Potion of Dying 6
Triple-Edged Sword 1

在这种情况下,权重表示选择的可能性那个项目。例如,你获得痛苦之剑的可能性是三刃剑的 10 倍。

要在 Java 中实现加权随机选择,我们可以使用 NavigableMap:

import java.util.NavigableMap;
import java.util.Random;
import java.util.TreeMap;

public class RandomCollection<E> {
    private final NavigableMap<Double, E> map = new TreeMap<>();
    private final Random random;
    private double total = 0;

    public RandomCollection() {
        this(new Random());
    }

    public RandomCollection(Random random) {
        this.random = random;
    }

    public RandomCollection<E> add(double weight, E result) {
        if (weight <= 0) return this;
        total += weight;
        map.put(total, result);
        return this;
    }

    public E next() {
        double value = random.nextDouble() * total;
        return map.higherEntry(value).getValue();
    }
}

用法:

RandomCollection<String> rc = new RandomCollection<>()
        .add(40, "dog").add(35, "cat").add(25, "horse");

for (int i = 0; i < 10; i++) {
    System.out.println(rc.next());
}

此代码演示了如何将具有特定权重的项目添加到集合中,然后根据分配给他们的概率。

以上是如何用Java实现加权随机选择?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn