Java如何制作一个权重随机数工具类呢?
下文笔者讲述Java编写一个权重随机数工具类的方法分享,如下所示
实现思路: 借助hutool的RandomUtil类 即可开发一个随机数生成工具类例:
import cn.hutool.core.lang.WeightRandom; import cn.hutool.log.Log; import cn.hutool.log.LogFactory; public class RandomUtil extends cn.hutool.core.util.RandomUtil { /** * 根据权重列表生成固定总量的数组 * * @param weightlist 权重列表 * @param total 随机列表数的总和 * @param fuDong 浮动指数 * @return */ public static <T> Map<T, Integer> randomListFromWeight(List<WeightRandom.WeightObj<T>> weightList, Integer total, double fuDong) { if (!(fuDong > 0 && fuDong < 1)) { throw new IllegalArgumentException("浮动指数必须在0~1之间"); } Map<T, Integer> result = new HashMap<>(); // 将权重列表倒序排列,保证权重高的优先取值 weightList.sort(new Comparator<WeightRandom.WeightObj>() { @Override public int compare(WeightRandom.WeightObj o1, WeightRandom.WeightObj o2) { if (o1.getWeight() > o2.getWeight()) { return -1; } else if (o1.getWeight() < o2.getWeight()) { return 1; } else { return 0; } } }); // 统计权重总和 double sumWeight = weightList.stream() .map(WeightRandom.WeightObj::getWeight).reduce((a, b) -> a + b).orElse(1.0 * weightList.size()); // 计算平均权重 double avgWeight = sumWeight / weightList.size(); // 计算平均值 double avg = 1.0 * total / weightList.size(); int before = 0; for (int i = 0; i < weightList.size() - 1; i++) { // 实际值 = 平均值 * 权重 / 平均权重 double countForWeight = avg * weightList.get(i).getWeight() / avgWeight; double min = before + countForWeight * (1 - fuDong); double max = before + countForWeight * (1 + fuDong); min = min > total ? before : min; max = max > total ? total : max; int after = min < max ? (int) Math.round(RandomUtil.randomDouble(min, max)) : total; result.put(weightList.get(i).getObj(), after - before); before = after; } // 最后一个的值 result.put(weightList.get(weightList.size() - 1).getObj(), total - before); return result; } }
版权声明
本文仅代表作者观点,不代表本站立场。
本文系作者授权发表,未经许可,不得转载。