随机抽奖

/**
 * 经常要从多个选项中随机选择一个,不过,不同选项经常有不同的权重。
 *
 * 比如,给用户随机奖励,
 * 三种面额:1元、5元和10元,权重分别为70、20和10。这个怎么实现呢?
 *
 * 实现的基本思路是,使用概率中的累计概率分布。
 *
 * 以上面的例子来说,计算每个选项的累计概率值,
 * 首先计算总的权重,这里正好是100,每个选项的概率是70%、20%和10%,累计概率则分别是70%、90%和100%
 *
 */
public class Test {
    public static void main(String[] args) {
        Pair[] options = new Pair[] {
                Pair.builder().item("1元").weight(7).build(),
                Pair.builder().item("2元").weight(2).build(),
                Pair.builder().item("10元").weight(1).build()
        };
        WeightRandom rnd = new WeightRandom(options);
        for (int i = 0; i < 10; i++) {
            System.out.print(rnd.nextItem() + " ");
        }
    }
}
@Data
@Builder
public class Pair {

    Object item;

    int weight;
}
public class WeightRandom {

    private Pair[] options;

    private double[] cumulativeProbabilities;

    private Random rnd;

    public WeightRandom(Pair[] options) {
        this.options = options;
        this.rnd = new Random();
        prepare();
    }

    private void prepare() {
        int weight = 0;
        for (Pair pair : options) {
            weight += pair.getWeight();
        }
        cumulativeProbabilities = new double[options.length];
        int sum = 0;
        for (int i = 0; i < options.length; i++) {
            sum += options[i].getWeight();
            cumulativeProbabilities[i] = sum / (double) weight;
        }
    }

    public Object nextItem() {
        double randomValue = rnd.nextDouble();
        int index = Arrays.binarySearch(cumulativeProbabilities, randomValue);
        if (index < 0) {
            index = -index-1;
        }
        return options[index].getItem();
    }
}
发布了89 篇原创文章 · 获赞 38 · 访问量 4万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 精致技术 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览