首頁 > Java > java教程 > 如何用Java實現加權隨機選擇?

如何用Java實現加權隨機選擇?

Patricia Arquette
發布: 2024-11-08 08:05:01
原創
333 人瀏覽過

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中文網其他相關文章!

來源:php.cn
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
作者最新文章
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板