java 按概率生成随机数算法

最近项目有需求，一组数据按照概率分布随机取出一个，相当于抽奖，所以写了一个算法。

 

    /**
     * 从一组数据按照概率分布随机取出一个
     * @param nums
     * @return 随机生成的数据的id
     */
    public static int getRandomNum(List<ModelNum> nums) {
        List<Double> percent = new ArrayList<>();
        int sumWeight = 0;

        //计算总权重
        for (int i = 0; i < nums.size(); i++) {
            sumWeight += nums.get(i).getWeight();
        }

        for (int i = 0; i < nums.size(); i++) {
            //计算出每个数据所占百分比，保留两位小数
            percent.add(Double.parseDouble(stringFormat(Arith.div(String.valueOf(nums.get(i).getWeight()),String.valueOf(sumWeight)))));
        }

        //按照每个数据对应的概率分布，生成100个数据
        List<ModelNum> list = new ArrayList<>();
        for (int j = 0; j < percent.size(); j++) {
            for (int i = 0; i < 100 * percent.get(j); i++) {
                list.add(nums.get(j));
            }
        }

        //产生0-99的整数值,也是list的索引
        int index = (int) (Math.random() * 100);
        //从生成的100个数据中，随机取出
        int rand = list.get(index).getId();
        return rand;
    }

BY LiYing