带权重的随机数算法的实现

在做一个银行的网页游戏的时候，涉及到一个随机抽奖的模块。具体的需求是：当用户的积分可以抽奖的时候，点击抽奖则消耗指定的积分随机抽取奖品，各种奖品的概率如下：笔记本（10%），手机（20%），充值卡（30%），积分（40%）。因为，奖品的概率是可以设置的，因此考虑Java容器中的TreeMap集合实现该算法。
核心逻辑：

累加每个物品的权重笔记本(10%)-手机(30%)-充值卡(60%)-积分(100%)，则4个物品的的权重有效长度区间分别为（0,10]、(10,30]、(30,60]、(60,100]。然后随机出一个[0,10)之间的随机数。落在哪个区间，则该区间之后的元素即为按权重命中的元素。

为什么用TreeMap？

因为TreeMap底层用红黑树实现，TreeMap中的key是唯一且有序的，因此TreeMap提供了如下方法：

NavigableMap<K,V>  tailMap(K fromKey, boolean inclusive) 返回此映射的部分视图，其键大于（或等于，如果 inclusive 为 true）fromKey。K lastKey() 返回映射中当前最后一个（最高）键。K  firstKey() 返回此映射中当前第一个（最低）键。

试想，当我们用普通的不可动态扩展的方法如if/switch方法实现时，必然是产生一个随机数，然后判断该随机数在哪个概率区间。TreeMap因为有序，因此刚好提供了两个方法tailMap，firstKey用于实现落点区间判断的逻辑，简洁又高效，并且方便动态扩展。
具体的代码实现：

package icerno.com.weightRandom;import java.util.ArrayList;
import java.util.List;
import java.util.SortedMap;
import java.util.TreeMap;import org.apache.commons.math3.util.Pair;import com.cookingfox.guava_preconditions.Preconditions;public class WeightRandom<K,V extends Number> {private TreeMap<Double, K> weightMap = new TreeMap<Double, K>();public WeightRandom(List<Pair<K, V>> list) {Preconditions.checkNotNull(list, "list can NOT be null!");for (Pair<K, V> pair : list) {double lastWeight = this.weightMap.size() == 0 ? 0 : this.weightMap.lastKey().doubleValue();//统一转为doublethis.weightMap.put(pair.getValue().doubleValue() + lastWeight, pair.getKey());//权重累加}System.out.println("------Tree init OK!--------");}public K random() {double randomWeight = this.weightMap.lastKey() * Math.random();SortedMap<Double, K> tailMap = this.weightMap.tailMap(randomWeight, true);return this.weightMap.get(tailMap.firstKey());}@SuppressWarnings({ "rawtypes", "unchecked" })public static void main(String[] args){List<Pair<String,Double>> list = new ArrayList<Pair<String,Double>>();list.add(new Pair<String,Double>("A",0.20));list.add(new Pair<String,Double>("C",0.50));list.add(new Pair<String,Double>("B",0.30));WeightRandom weightRandom = new WeightRandom(list);Object random = weightRandom.random();System.out.println("----Key----"+random);}
}

注意：TreeMap为异步的非线程安全的，因此在并发要求高时，可以使用如下代码加锁。

public static Map<Double, K> treemap = Collections.synchronizedMap(new TreeMap<Double, K>());

参考文章：
权重随机算法的java实现
Java TreeMap 介绍和使用

带权重的随机数算法的实现相关推荐

【酷熊科技】工作积累 ----------- 随机数 (带权重的随机数)
依靠权重来随机思路 : 先随机每个随机段权重然后通过随机出来的权重判断在哪个随机段内然后再在这个随机段中随机出来需要的随机数举例: 需求: 50 ~ 999 随 ...
python概率随机抽奖_Python利用带权重随机数解决抽奖和游戏爆装备
关于带权随机数为了帮助理解,先来看三类随机问题的对比: 1.已有n条记录,从中选取m条记录,选取出来的记录前后顺序不管. 实现思路:按行遍历所有记录,约隔n/m条取一个数据即可 2.在1类情况下,还 ...
关于寻路算法的一些思考（3）：A*算法的实现
概述剥除代码,A* 算法非常简单.算法维护两个集合:OPEN 集和 CLOSED 集.OPEN 集包含待检测节点.初始状态,OPEN集仅包含一个元素:开始位置.CLOSED集包含已检测节点.初始状态 ...
计算机LCG/PCG/MWC/XorShift等PRNG算法，以及V8中Math.random()、webkit中crypto等随机算法的实现
计算机LCG/PCG/MWC/XorShift等PRNG算法,以及V8中Math.random().webkit中crypto等随机算法的实现本文篇幅较长,如想直接看 js 的随机数实现可定位本文E ...
传授“带权重的负载均衡实现算法”独家设计思路！
作者|孙玄/陈东分布式系统中,大部分系统调用都会涉及到负载均衡,例如:客户端发往服务端的请求首先到达反向代理,然后反向代理再通过负载均衡算法将请求转发到业务系统:或者后端业务系统各模块间的调用前,也 ...
干货回顾丨TensorFlow四种Cross Entropy算法的实现和应用
交叉熵介绍交叉熵(Cross Entropy)是Loss函数的一种(也称为损失函数或代价函数),用于描述模型预测值与真实值的差距大小,常见的Loss函数就是均方平方差(Mean Squared Er ...
选择性模糊及其算法的实现。
选择性模糊及其算法的实现. 我们常见的模糊算法比如均值模糊.高斯模糊等其基本的过程都是计算一个像素周边的的某个领域内,相关像素的某个特征值的累加和及对应的权重,然后得到结果值.比如均值模糊的各像素的权 ...
RSA算法和RSA数字签名算法的实现
RSA算法和RSA数字签名算法的实现 http://blog.chinaunix.net/uid-21880738-id-1813146.html 顾婷婷李涛 (四川大学计算机系(西区) 成都 61 ...
游戏中常用的寻路算法的分享（3）：A*算法的实现
概述剥除代码,A* 算法非常简单.算法维护两个集合:OPEN 集和 CLOSED 集.OPEN 集包含待检测节点.初始状态,OPEN集仅包含一个元素:开始位置.CLOSED集包含已检测节点.初始状态 ...

带权重的随机数算法的实现

带权重的随机数算法的实现相关推荐

最新文章

热门文章