如何在Java中实现权重随机抽取_基于TreeMap的离散分布范围查找

雨瑶姑娘_3055

雨瑶姑娘_3055

2026-03-19

161人浏览

原创

应使用treemap存储累积权重右边界实现o(log n)加权随机抽取,而非random.nextdouble()×totalweight遍历匹配,因后者时间复杂度o(n)在高频调用时显著拖慢性能。

如何在java中实现权重随机抽取_基于treemap的离散分布范围查找

为什么不用 Random.nextDouble() 直接乘总权重

直接用 nextDouble() * totalWeight 得到一个浮点数,再遍历累加权重匹配区间——看似简单,但实际在权重数量大(比如上千个选项)或高频调用时,每次都要从头累加,时间复杂度是 O(n)。线上服务里,单次抽取耗时从纳秒级跳到微秒级,积少成多就容易拖慢吞吐。

TreeMap 的核心价值不是“排序”,而是利用其底层红黑树的 floorEntry() 实现 O(log n) 范围查找:把累积权重当 key,选项当 value,一次二分就能定位。

  • 必须预计算前缀和,不能边插边查;插入新选项需重建整个 TreeMap
  • key 必须是 long 或 double,但推荐 long(避免浮点精度导致 floorEntry() 找错区间)
  • TreeMap 的 key 不能重复,所以累积权重必须严格递增——哪怕两个选项权重为 0,也要保证 key 唯一(例如用索引补位)

怎么构建带累积权重的 TreeMap<long t></long>

关键不是存“权重”,而是存“右边界”。比如选项 A 权重 3、B 权重 5、C 权重 2,对应累积数组是 [3, 8, 10],TreeMap 应存:{3=A, 8=B, 10=C}。抽取时生成 [1, 10] 内随机 long,用 floorEntry(randomValue) 拿到最接近且不大于它的 entry。

Alibabacloud Sdk Client Initialization For Java
Alibabacloud Sdk Client Initialization For Java

在 Java 中初始化和管理阿里云 SDK客户端。包括单例模式、线程安全、endpoint 与 region 配置、VPC 终端节点、同步与异步等。

下载
  • 随机数下界是 1,不是 0:因为权重为 0 的选项本就不该被抽中,nextLong(totalWeight) 返回 [0, totalWeight),要 +1 对齐左闭右闭语义
  • 用 nextLong(totalWeight) + 1,而非 nextInt():避免 int 溢出(总权重大于 Integer.MAX_VALUE 很常见)
  • 构建时用 new TreeMap() 即可,别用 ConcurrentSkipListMap:它不支持 floorEntry() 的强一致性语义,且并发写入仍需外部同步
Map<long string> weightedMap = new TreeMap();
long sum = 0;
for (var item : items) {
    sum += item.weight;
    weightedMap.put(sum, item.value);
}
// 抽取:
long rand = random.nextLong(sum) + 1;
String result = weightedMap.floorEntry(rand).getValue();
</long>

floorEntry() 返回 null 怎么办

只会在 rand 或 <code>weightedMap 为空时发生。前者不可能(因为 rand = nextLong(sum) + 1,最小为 1),后者是逻辑错误。真正要防的是:sum == 0 时 nextLong(0) 抛 IllegalArgumentException。

  • 构建前必须校验 totalWeight > 0,否则直接抛 IllegalStateException,别等运行时崩
  • 如果业务允许“零权重选项存在”,构建 TreeMap 前应先过滤掉权重 ≤ 0 的项,而不是给它们分配 key=0
  • 不要用 ceilingEntry() 回退:它可能跳过第一个区间,导致分布偏移

比 ArrayList + 线性扫描快多少

实测 1000 个选项时,TreeMap 查找 P99 耗时约 50ns,ArrayList 线性扫描约 300ns;到 10000 个选项,前者仍稳定在 60ns,后者升至 3μs 以上。但注意:TreeMap 构建开销比 ArrayList 大 3–5 倍,所以适用于“构建一次、反复抽取”的场景。

  • 如果权重频繁变更(如实时竞价中每毫秒更新),宁可用数组+二分搜索(Arrays.binarySearch()),自己维护前缀和数组,避免 TreeMap 的对象创建和树平衡开销
  • Java 17+ 可考虑 TreeMap.tailMap(key, true).firstEntry(),但语义等价于 floorEntry(),无实际优势
  • 别为了“看起来高级”而用 TreeMap:只有当选项数 ≥ 50 且抽取频次高时,log n 才真有价值

最易被忽略的一点:累积权重的类型必须和随机数类型严格一致。用 long 累加却用 nextInt() 生成随机数,或者反过来,会导致范围错位、永远抽不到末尾项——这种 bug 在压测时才暴露,但原因极难定位。

Java免费学习笔记:立即使用
解锁 Java 大师之旅:从入门到精通的终极指南

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

java

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
java
java

Java是一个通用术语,用于表示Java软件及其组件,包括“Java运行时环境 (JRE)”、“Java虚拟机 (JVM)”以及“插件”。php中文网还为大家带了Java相关下载资源、相关课程以及相关文章等内容,供大家免费下载使用。

2023.06.15

9157

6

java正则表达式语法
java正则表达式语法

java正则表达式语法是一种模式匹配工具,它非常有用,可以在处理文本和字符串时快速地查找、替换、验证和提取特定的模式和数据。本专题提供java正则表达式语法的相关文章、下载和专题,供大家免费下载体验。

2023.07.05

6322

9

java自学难吗
java自学难吗

Java自学并不难。Java语言相对于其他一些编程语言而言,有着较为简洁和易读的语法,本专题为大家提供java自学难吗相关的文章,大家可以免费体验。

2023.07.31

5652

8

java配置jdk环境变量
java配置jdk环境变量

Java是一种广泛使用的高级编程语言,用于开发各种类型的应用程序。为了能够在计算机上正确运行和编译Java代码,需要正确配置Java Development Kit(JDK)环境变量。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.01

1004

3

java保留两位小数
java保留两位小数

Java是一种广泛应用于编程领域的高级编程语言。在Java中,保留两位小数是指在进行数值计算或输出时,限制小数部分只有两位有效数字,并将多余的位数进行四舍五入或截取。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.02

848

3

java基本数据类型
java基本数据类型

java基本数据类型有:1、byte;2、short;3、int;4、long;5、float;6、double;7、char;8、boolean。本专题为大家提供java基本数据类型的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

1196

5

java有什么用
java有什么用

java可以开发应用程序、移动应用、Web应用、企业级应用、嵌入式系统等方面。本专题为大家提供java有什么用的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

2409

5

java在线网站
java在线网站

Java在线网站是指提供Java编程学习、实践和交流平台的网络服务。近年来,随着Java语言在软件开发领域的广泛应用,越来越多的人对Java编程感兴趣,并希望能够通过在线网站来学习和提高自己的Java编程技能。php中文网给大家带来了相关的视频、教程以及文章,欢迎大家前来学习阅读和下载。

2023.08.03

19751

3

配置java环境变量
配置java环境变量

配置Java环境变量是为了让操作系统能够识别和使用Java的相关命令和功能。本专题为大家提供配置java环境变量相关文章,帮助大家解决问题。

2023.08.03

1095

8

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
dev.java 官方:Learn Java
dev.java 官方:Learn Java

共0课时 | 0人学习

Java JDBC数据库连接官方教程
Java JDBC数据库连接官方教程

共0课时 | 0人学习