轮盘赌算法

首先，这个算法可以如下表述：

如果已知A类对象生成概率为P(A)，B类对象生成概率为P(B),C类对象···，K类对象，他们的概率总和为1，问如何在A~K中随机生成一个对象

算法理解如下：

即我们需要先对0~1区段按照概率大小划分长度，设随机生成一个0~1之间的数，这种随机数生成结果是在0~1均匀分布的，然后看其究竟落在哪个概率区段即为生成哪种对象

算法代码：

package test;

import org.junit.Test;

public class Roulette {

    @Test
    public  void test() {
        double[] P={0.2,0.3,0.5};//假设的概率分布
        int appearnum=0;
        for(int i=0;i<100;i++){
            int result=roulette(P);
            if(result==2)//统计第2类对象生成的数目，可以看出大致为50%的生成概率
                appearnum++;
        }
        System.out.println(appearnum);
    }
    /**
     * 轮盘赌函数
     * @param P 各类对象概率分布
     * @return 生成的对象类
     */
    private int roulette(double[] P){
        
        double rand=Math.random();
        double pointer=0;//pointer指示每个区段的右边界，从左往右扫描判断
        for(int i=0;i<P.length;i++){
            pointer+=P[i];
            if(rand<=pointer)
                return i;
        }
        return -1;
    }

}

 

原文：轮盘赌算法