Java抽奖概率算法

川长思鸟来 2021-12-16 22:51 230阅读 0赞

### 文章目录 ###

*   *  奖品
     *  实体
     *  抽奖方法
     *  测试
     *  感谢

## 奖品 ##

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM0ODQ1Mzk0_size_16_color_FFFFFF_t_70]

## 实体 ##

package com.leigq.www.shiro.controller;
    
    import lombok.AllArgsConstructor;
    import lombok.Builder;
    import lombok.Data;
    
    /** * 奖品 DTO * <br/> * * @author ：leigq * @date ：2019/7/5 23:00 */
    @Data
    @Builder
    @AllArgsConstructor
    public class GiftDTO { 
    
        /** * 索引 */
        private Integer index;
    
    
        /** * 奖品名称 */
        private String name;
    
        /** * 奖品编号 */
        private String no;
    
    
        /** * 中奖概率 */
        private Double probability;
    }

## 抽奖方法 ##

/** * 抽奖方法 * <br/> * create by: leigq * <br/> * create time: 2019/7/5 23:08 * @param orignalRates 商品中奖概率列表，保证顺序和实际物品对应 * @return 中奖商品索引 */
     public static int lottery(List<Double> orignalRates) { 
    
         if (orignalRates == null || orignalRates.isEmpty()) { 
             return -1;
         }
    
         int size = orignalRates.size();
    
         // 计算总概率，这样可以保证不一定总概率是1
         double sumRate = 0d;
         for (double rate : orignalRates) { 
             sumRate += rate;
         }
    
         // 计算每个物品在总概率的基础下的概率情况
         List<Double> sortOrignalRates = new ArrayList<>(size);
         Double tempSumRate = 0d;
         for (double rate : orignalRates) { 
             tempSumRate += rate;
             sortOrignalRates.add(tempSumRate / sumRate);
         }
    
         // 根据区块值来获取抽取到的物品索引
         double nextDouble = Math.random();
         sortOrignalRates.add(nextDouble);
         Collections.sort(sortOrignalRates);
    
         return sortOrignalRates.indexOf(nextDouble);
     }

## 测试 ##

我这里就只用两个商品测试

public static void main(String[] args) { 
    
        List<GiftDTO> gifts = new ArrayList<>();
        gifts.add(new GiftDTO(1,  "一等奖", "P1", 0.4d));
        gifts.add(new GiftDTO(2, "谢谢参与","P2", 0.6d));
    
        // 存储概率
        List<Double> orignalRates = new ArrayList<>(gifts.size());
        for (GiftDTO gift : gifts) { 
            double probability = gift.getProbability();
            if (probability < 0) { 
                probability = 0;
            }
            orignalRates.add(probability);
        }
    
        // 统计
        Map<Integer, Integer> count = new HashMap<>();
    
        // 测试次数
        double num = 1000000;
        for (int i = 0; i < num; i++) { 
            int orignalIndex = lottery(orignalRates);
            Integer value = count.get(orignalIndex);
            count.put(orignalIndex, value == null ? 1 : value + 1);
        }
    
        for (Map.Entry<Integer, Integer> entry : count.entrySet()) { 
            System.out.println(gifts.get(entry.getKey()) + ", 命中次数=" + entry.getValue() + ", 实际概率="
                    + entry.getValue() / num);
        }
    }

![在这里插入图片描述][20190706104123189.png]  
结果与预期差不多。

## 感谢 ##

*  [https://yq.aliyun.com/articles/44578/][https_yq.aliyun.com_articles_44578]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM0ODQ1Mzk0_size_16_color_FFFFFF_t_70]: /images/20211214/53dd17c0f84e4b468495adc40008113a.png
[20190706104123189.png]: /images/20211214/132f2946cd6a4129a938105f279dd233.png
[https_yq.aliyun.com_articles_44578]: https://yq.aliyun.com/articles/44578/