java - 如何管理数组中的概率？

Question

我有 2 个 2D 数组，用于存储分形数据。我的第一个数组有 3 行，每行的最后一个值是概率。

第 1 行的概率为 0.33， 第 2 行的概率为 0.33，并且 第 3 行的概率为 0.34

现在我正在做的是从 0-1 生成一个随机 double 值。 我使用 if 语句来确定应该使用哪一行。

这是我的数组:

    static double[][] SierpGasketMatrix = { { 0.5, 0, 0, 0.5, 0, 0, 0.33 }, //row 1
                                            { 0.5, 0, 0, 0.5, 0, 0.5, 0.33 }, //row 2
                                            { 0.5, 0, 0, 0.5, 0.43, 0.25, 0.34 } }; //row 3

    static double[][] BarnsleyFernMatrix = { { 0, 0, 0, 0.16, 0, 0, 0.01 }, //row 1
                                             { 0.85, 0.04, -0.04, 0.85, 0, 1.6, 0.85 }, //row 2 
                                             { 0.2, -0.26, 0.23, 0.22, 0, 1.6, 0.07 }, //row 3
                                             { -0.15, 0.28, 0.26, 0.24, 0, 0.44, 0.07 } }; //row 4

这里是确定使用哪一行的代码:

            double randNum = Math.random();
            int row = -1;

            if (randNum < 0.33)
                row = 0;
            else if (randNum < 0.66)
                row = 1;
            else if (randNum < 1.00)
                row = 2; 

我觉得应该有更好的方法来做到这一点。另外，我的第二个数组有 4 行和不同的概率，我想对该数组使用相同的方法。

Answer 1

如果你想要一个通用的解决方案，只需迭代数组直到达到阈值。结果实际上并不比您正在做的慢。

static double[] getRandomRow(double[][] data) {
    double randNum = Math.random();
    double threshold = 0.0;
    for (double[] row : data) {
        threshold += row[row.length - 1]; // probability = last value of row
        if (threshold > randNum)
            return row;
    }
    // We should never get here, but float math is imprecise, so just return last row.
    // Could also be that input probabilities don't add to 1, so handle that if necessary.
    return data[data.length - 1];
}