Java中如何基于给定概率与结果集生成离散随机变量？

核心思路 #

使用逆变换采样法，通过均匀随机数匹配累积概率区间，得到符合给定概率分布的离散结果。

实现步骤 #

• 第一步：参数合法性校验
  1. 校验概率数组和结果数组长度一致，否则抛出异常
  2. 校验所有概率值≥0，且概率总和与1的误差不超过允许的精度阈值（比如1e-9），避免浮点精度误差导致的问题
• 第二步：构建累积概率分布（CDF）数组
  遍历概率数组，依次累加得到累积概率，示例中的概率数组{.2,.1,.3,.4}对应的累积数组为{0.2, 0.3, 0.6, 1.0}
• 第三步：生成均匀随机数并匹配区间
  1. 生成一个范围在[0, 1)的均匀随机数r
  2. 遍历累积概率数组，找到第一个大于r的累积值对应的索引，返回相同索引的结果值即可

代码实现 #

import java.util.concurrent.ThreadLocalRandom;

public class Problem {
    private static final double EPS = 1e-9;

    public double randPMF(double[] probabilities, double[] outcomes) {
        // 1. 参数校验
        if (probabilities == null || outcomes == null || probabilities.length != outcomes.length) {
            throw new IllegalArgumentException("概率数组和结果数组长度不匹配");
        }
        int len = probabilities.length;
        if (len == 0) {
            throw new IllegalArgumentException("数组不能为空");
        }
        double sum = 0.0;
        for (double p : probabilities) {
            if (p < -EPS) {
                throw new IllegalArgumentException("概率值不能为负数");
            }
            sum += p;
        }
        if (Math.abs(sum - 1.0) > EPS) {
            throw new IllegalArgumentException("概率总和需要为1");
        }

        // 2. 计算累积概率
        double[] cdf = new double[len];
        cdf[0] = probabilities[0];
        for (int i = 1; i < len; i++) {
            cdf[i] = cdf[i-1] + probabilities[i];
        }

        // 3. 生成随机数找对应区间
        double r = ThreadLocalRandom.current().nextDouble();
        for (int i = 0; i < len; i++) {
            if (r < cdf[i]) {
                return outcomes[i];
            }
        }
        // 浮点精度兜底，返回最后一个元素
        return outcomes[len - 1];
    }

    // 测试示例
    public static void main(String[] args) {
        Problem problem = new Problem();
        double[] probabilities = {.2,.1,.3,.4};
        double[] outcomes = {4,5,8,11.5};
        double discreteRV = problem.randPMF(probabilities, outcomes);
        System.out.println(discreteRV);
    }
}

优化方案 #

如果概率数组长度很大，可以将遍历查找替换为二分查找，时间复杂度从O(n)降低到O(logn)，Java中可以直接使用Arrays.binarySearch()方法实现。

内容的提问来源于stack exchange，提问作者cvsrt