讲师中心 微信公众号
AI工具推荐 视频效率加速

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

雨瑶姑娘_3055

雨瑶姑娘_3055

发布时间:2026-03-19 15:36:27

|

161人浏览过

|

来源于php中文网

原创

应使用TreeMap<Long, T>存储累积权重右边界实现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>

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

  • 随机数下界是 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();

floorEntry() 返回 null 怎么办

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

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

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

下载

立即学习“Java免费学习笔记(深入)”;

  • 构建前必须校验 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 在压测时才暴露,但原因极难定位。

热门AI工具

更多
Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

相关专题

更多
java
java

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

9237

2023.06.15

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

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

6402

2023.07.05

java自学难吗
java自学难吗

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

5712

2023.07.31

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

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

1024

2023.08.01

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

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

848

2023.08.02

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

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

1216

2023.08.02

java有什么用
java有什么用

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

2429

2023.08.02

java在线网站
java在线网站

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

19771

2023.08.03

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

160

2026.09.23

热门下载

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

精品课程

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

共0课时 | 0人学习

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

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn