獎品
實體
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);
}
}
結果與預期差不多。