Java实现自定义概率抽奖算法

使用随机数生成器

private static final RandomGenerator randomGenerator = RandomGenerator.of("L64X128MixRandom")

定义抽奖池的数据类型

使用Map数组来存放奖品池，Map中含有“prize”：奖品，“odds”:概率

概率必须是int类型，概率综合不一定为100%，抽中概率为奖品概率/总概率

例：

List<Map<String, Object>> pool = Arrays.asList(
        Map.of("prize", "张三", "odds", 69), // 69%
        Map.of("prize", "李四", "odds", 30), // 30%
        Map.of("prize", "王五", "odds", 1) // 1%
);

计算总概率

int totalOdds = pool.stream().mapToInt(item -> (Integer) item.get("odds")).sum()

生成随机数作为中奖范围

int randomNum = randomGenerator.nextInt(totalOdds)

中奖逻辑判断

int current = 0;
for (Map<String, Object> item : pool) {
    int odds = (Integer) item.get("odds");
    current += odds;
    if (randomNum < current) {
        return item.get("prize");
    }
}

完整代码

import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.random.RandomGenerator;

/**
 * @author Mingkai
 * @date 2025/5/20 09:39
 * @description 可自定义概率的抽奖
 */
public class PrizeDrawUtil {

    /**
     * 随机生成器
     */
    private static final RandomGenerator randomGenerator = RandomGenerator.of("L64X128MixRandom");

    /**
     * 抽奖
     *
     * @param pool 奖品/概率池
     * @return 奖品
     * @implNote
     *  参数为ListMap 数组：prize: 奖品名称；odds: 奖品概率（必须是int）<br>
     *  概率综合不一定为100%，抽中概率为奖品概率/总概率
     *  <br>
     *  示例：
     *  <pre>
     *   List<Map<String, Object>> pool = Arrays.asList(
     *       Map.of("prize", "张三", "odds", 69), // 69%
     *       Map.of("prize", "李四", "odds", 30), // 30%
     *       Map.of("prize", "王五", "odds", 1) // 1%
     *    );
     *  </pre>
     */
    public static Object prizeDraw(List<Map<String, Object>> pool) {
        // 获取总概率
        int totalOdds = pool.stream().mapToInt(item -> (Integer) item.get("odds")).sum();
        // 生成一个随机数
        int randomNum = randomGenerator.nextInt(totalOdds);
        // 遍历，判断随机数是否在当前奖品的概率范围内
        int current = 0;
        for (Map<String, Object> item : pool) {
            int odds = (Integer) item.get("odds");
            current += odds;
            if (randomNum < current) {
                return item.get("prize");
            }
        }
        // 理论上不会走到这里
        return null;
    }

    public static void main(String[] args) {
        List<Map<String, Object>> pool = Arrays.asList(
                Map.of("prize", "张三", "odds", 69),
                Map.of("prize", "李四", "odds", 30),
                Map.of("prize", "王五", "odds", 1)
        );
        int 张三 = 0;
        int 李四 = 0;
        int 王五 = 0;
        for (int i = 0; i < 10000; i++) {
            String prize = String.valueOf(prizeDraw(pool));
            if ("张三".equals(prize)) {
                张三++;
            }
            if ("李四".equals(prize)) {
                李四++;
            }
            if ("王五".equals(prize)) {
                王五++;
            }
        }
        System.out.println("张三:" + 张三 + " 概率:" + 张三 / 10000.0 *  100 + "%");
        System.out.println("李四:" + 李四 + " 概率:" + 李四 / 10000.0 *  100 + "%");
        System.out.println("王五:" + 王五 + " 概率:" + 王五 / 10000.0 *  100 + "%");
    }

}

实例中输出结果：

张三:6840 概率:68.4%
李四:3070 概率:30.7%
王五:90 概率:0.8999999999999999%