C++如何实现高效的数组元素随机采样(Sampling)

酷涛酱_4640

酷涛酱_4640

2026-06-22

328人浏览

原创

用 std::shuffle 做无放回随机采样最稳妥:先全局打乱再取前 k 个,避免误用不存在的 partial_shuffle;k≪n 时改用 uniform_int_distribution 配 unordered_set 拒绝采样;有放回直接生成索引;加权采样必用 discrete_distribution。

c++如何实现高效的数组元素随机采样(sampling)

用 std::shuffle 做无放回随机采样最稳妥

如果你要从数组中等概率抽取 k 个不重复元素(比如选 5 个用户做 A/B 测试),std::shuffle 配合 std::vector::begin() 截断是最直接、最不容易出错的方式。它底层用 Fisher–Yates 算法,时间复杂度 O(n),且标准库已做优化,比手写循环更可靠。

常见错误是只 shuffle 前 k 位(误用 partial_shuffle)——C++ 标准库里根本没有 partial_shuffle,有人搜到的都是过时提案或第三方实现,强行用会导致未定义行为。

实操建议:

  • 先确保容器支持随机访问(std::vector、原生数组都行,std::list 不行)
  • 用 std::shuffle(vec.begin(), vec.end(), rng) 全局打乱,再取前 k 个;不要试图“只 shuffle 前 k 个”来省时间——那会破坏均匀性
  • rng 必须是可复制的随机数引擎,推荐 std::mt19937 配 std::random_device 初始化,别用 rand()
std::vector<int> data = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
std::mt19937 g{std::random_device{}()};
std::shuffle(data.begin(), data.end(), g);
std::vector<int> sample(data.begin(), data.begin() + 3); // 取前 3 个
</int></int>

当 k 时,用 <code>std::uniform_int_distribution + std::set 避免全量 shuffle

如果数组有 1000 万条记录,但你只要随机抽 10 个,全 shuffle 就是浪费——此时更适合用“拒绝采样 + 去重”策略,空间换时间。

关键点在于:不能用 std::vector 存已选索引然后每次 std::find,那样是 O(k²);用 std::unordered_set 或 std::set 查重才是 O(log k) 或均摊 O(1)。

注意陷阱:

C++14
C++14

C++14 对 C++11 的修正与增强版本,适合旧系统维护和较老工具链兼容。

下载
  • 若 k 接近 n(比如 n=100, k=95),拒绝采样可能反复碰撞,实际性能反而比 shuffle 差
  • std::uniform_int_distribution 的上下界必须是闭区间 [0, n-1],写成 (0, n) 会漏掉首尾
  • 别在循环里反复构造 std::uniform_int_distribution 实例,它不是轻量对象
std::vector<int> data = {/* ... large array ... */};
std::mt19937 g{std::random_device{}()};
std::uniform_int_distribution<size_t> dist(0, data.size()-1);
std::unordered_set<size_t> chosen;
std::vector<int> sample;
<p>while (chosen.size() 
sample.push_back(data[idx]);
}
}
</p></int></size_t></size_t></int>

有放回采样直接用 std::uniform_int_distribution 循环生成索引

需要允许重复(比如蒙特卡洛模拟中按权重重采样前一步结果),那就完全不需要去重逻辑,也不用 shuffle,纯索引生成即可。

性能上这是最轻量的方式:O(k) 时间、O(1) 额外空间。但务必确认业务是否真允许重复——很多场景表面说“随机选”,实际隐含“不重复”约束,混淆会导致数据偏差。

常见疏忽:

  • 忘记把分布对象 dist 定义在循环外,导致每次调用都重新构造,开销陡增
  • 用 rand() % n 替代 std::uniform_int_distribution,在 n 不是 2 的幂时会产生偏置(低位周期短、高位未充分利用)
  • 没检查 data 是否为空,data.size()-1 在空容器下会变成极大正数(size_t 下溢)
if (data.empty()) throw std::runtime_error("sampling from empty container");
std::uniform_int_distribution<size_t> dist(0, data.size()-1);
for (int i = 0; i <h3>自定义权重采样得用 <code>std::discrete_distribution</code>,别手算前缀和</h3>
<p>如果每个元素被选中的概率不同(比如按 PV 加权抽用户),C++11 起就该用 <code>std::discrete_distribution</code>,它内部已优化为 O(1) 查表(Alias Method 或 Walker’s Method),比手动二分查找前缀和快得多,且数值更稳定。</p>
<p>容易踩的坑集中在权重输入:</p>
<ul>
<li>权重必须是非负浮点数或整数,传负数会触发断言或未定义行为</li>
<li>权重为 0 是合法的,对应概率为 0,但所有权重全为 0 会导致构造失败</li>
<li>别把原始数据指针直接喂给 <code>discrete_distribution</code> 构造函数——它只接受迭代器范围或初始化列表,例如 <code>{w0, w1, w2}</code>
</li>
</ul>
<pre class="brush:php;toolbar:false;">std::vector<double> weights = {0.1, 0.6, 0.3};
std::discrete_distribution<size_t> dist(weights.begin(), weights.end());
for (int i = 0; i <p>真正麻烦的从来不是“怎么写”,而是判断该用哪种采样语义:无放回?有放回?等概?加权?边界条件(空容器、k=0、k>n)有没有被测试覆盖。这些决策点一旦错,后面代码越“高效”越危险。</p></size_t></double>

C++免费学习笔记(深入):立即使用
在学习笔记中,你将探索 C++ 的入门与实战技巧!

相关文章

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

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

下载

相关标签:

c++

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

相关专题

更多
c++和c语言的区别有哪些
c++和c语言的区别有哪些

c++和c语言的区别:1、面向对象编程(OOP)支持不同;2、新增特性不同;3、标准库不同;4、编译方式不同;5、命名空间不同等等。想了解更多c++的相关内容,可以阅读本专题下面的文章。

2024.03.14

2248

9

c++和python学习顺序推荐
c++和python学习顺序推荐

一般建议先学习C++,再学习Python,因为这样可以逐步从较为底层的编程语言向更高级的语言过渡。想了解更多python的相关内容,可以阅读本专题下面的文章。

2024.03.14

1019

6

python和c++学习性价比分析
python和c++学习性价比分析

Python易于学习,广泛应用于Web开发、数据科学和人工智能等领域,但性能较低。C语言性能高,适用于对性能要求较高的场景,如游戏开发和系统编程,但学习曲线陡峭,错误处理复杂。想了解更多python的相关内容,可以阅读本专题下面的文章。

2024.03.14

447

5

c语言和c++一样吗
c语言和c++一样吗

c语言和c++是两种不同的编程语言,虽然有相似之处,但存在显著差异。c语言专注于过程式编程和系统级开发,以简洁、高效著称。c++作为c语言的超集,引入了面向对象编程,增强了代码组织和管理能力,但学习曲线也更陡峭。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.14

327

5

c语言和c++先学哪个好
c语言和c++先学哪个好

初学者选择学习c语言还是c++语言,需要根据个人学习目标、背景以及编程兴趣和预期应用方向来决定。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.14

406

5

c语言和c++的区别和联系
c语言和c++的区别和联系

c语言和c++是计算机科学领域应用广泛的编程语言。虽然它们有着相似的基础,但它们在语言类型、语法功能和内存管理方面存在着显著差异。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.14

580

5

c++软件中文更改教程
c++软件中文更改教程

对于 ide,可通过打开设置,找到语言设置,选择中文,并保存更改。对于非 ide 应用程序,可查找设置或选项,选择语言设置,更改为中文,并保存更改。想了解更多c++的相关内容,可以阅读本专题下面的文章。

2024.03.21

1409

9

python和java和c++学习性价比分析
python和java和c++学习性价比分析

Python以其易学性、丰富的库和活跃的社区而著称,适合数据科学、人工智能和Web开发。Java以其跨平台性、企业级应用开发和Android应用开发而闻名。C++以其底层控制能力、高效性能和游戏开发而著称。选择哪种语言取决于个人兴趣、职业方向和特定需求。想了解更多python和java和c++的相关内容,可以阅读本专题下面的文章。

2024.03.22

1217

7

c++和c语言学习顺序推荐
c++和c语言学习顺序推荐

对于初学者,建议先学习C语言,掌握编程基础后再转入C++,便于理解面向对象编程概念。有编程经验者可直接学习C++,快速接触高级编程技术。想了解更多c++和c语言的相关内容,可以阅读本专题下面的文章。

2024.03.25

1325

9

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
C++ Core Guidelines
C++ Core Guidelines

共0课时 | 0人学习

C++ Reference
C++ Reference

共0课时 | 0人学习

C++ 官方标准说明
C++ 官方标准说明

共0课时 | 0人学习