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