如何在Rust中高效生成堆上存储的随机字节序列

你提到的vec![0; 50_000_000]确实会先将所有字节初始化为0，随即又被随机值覆盖，这会产生不必要的内存写入开销，尤其是大内存场景下影响更明显。下面提供两种更高效的方案：

方案一：手动分配未初始化内存并填充（unsafe） #

通过Vec::with_capacity只分配内存不做初始化，再通过unsafe操作标记内存可用，最后直接填充随机字节：

use rand::RngCore;

let size = 50_000_000;
// 仅分配指定容量的堆内存，不初始化
let mut v = Vec::with_capacity(size);
// unsafe操作：告知Vec内存已可用（必须确保后续完全覆盖所有字节）
unsafe {
    v.set_len(size);
}
// 直接填充随机字节到整个内存区域
let mut rng = rand::thread_rng();
rng.fill_bytes(&mut v);

注意：set_len是unsafe操作，必须保证后续代码会完全覆盖所有未初始化的内存，否则读取未初始化值会触发未定义行为（UB）。

方案二：利用rand迭代器直接生成（安全且简洁） #

借助rand库的迭代器功能，直接生成随机字节并收集到Vec，全程无需unsafe，且不会产生初始化开销：

use rand::Rng;

let size = 50_000_000;
let mut rng = rand::thread_rng();
// 生成指定数量的随机u8并收集到Vec，Vec会预分配容量且不初始化
let v: Vec<u8> = rng.sample_iter(rand::distributions::Standard)
    .take(size)
    .collect();

这个方案的优势是安全、可读性强，collect方法会根据迭代器的长度预分配足够容量，然后直接将随机字节写入内存，跳过了初始化为0的步骤，性能和方案一接近。

总结 #

• 追求极致性能且能保证unsafe代码安全时，可选择方案一；
• 优先推荐方案二，兼顾性能、安全性和代码简洁性。

内容的提问来源于stack exchange，提问作者RBF06