generate_n：别再手写循环填数组了，C++里这个冷门算法真香

上周帮同事看一段性能瓶颈代码，他用 for 循环给 vector 前 1000 个位置塞斐波那契数——逻辑没错，但写了 12 行，还漏了边界检查。我顺手改成一行 generate_n，他盯着屏幕愣了三秒：“这玩意儿……还能这么用？”

是的，std::generate_n 不是教科书里摆着看的装饰品。它解决的是一个非常具体、高频、但常被忽略的问题：按需生成固定数量的元素，并原地填充到指定起始位置之后的连续内存中。不是构造，不是拷贝，是“生成+写入”一步到位。

先说清它不干什么：它不分配内存，不检查容量，不处理越界——这些得你来兜底。它的职责极窄：从某个迭代器开始，调用生成器函数 N 次，把每次返回值赋给对应位置。正因如此，它轻、快、可控。

#include <algorithm>
#include <vector>
#include <iostream>

int main() {
    std::vector<int> v(10); // 明确预留空间
    int a = 0, b = 1;
    auto fib_gen = [&a, &b]() mutable -> int {
        int next = a;
        a = b;
        b = next + b;
        return next;
    };

    std::generate_n(v.begin(), 10, fib_gen);

    for (int x : v) std::cout << x << " "; // 0 1 1 2 3 5 8 13 21 34
}

注意两个关键点：vector 必须提前分配好至少 N 个元素的空间；生成器必须是可调用对象，且返回类型能隐式转换为目标容器的 value_type。漏掉任一，运行时崩溃或编译失败，毫不留情。

有人会问：为什么不用 std::generate？区别就藏在接口里。generate 需要首尾迭代器，意味着你要算好结束位置；而 generate_n 直接告诉你“就填 N 个”，语义更直白，尤其当你只关心数量、不关心容器总长时——比如往一个大 buffer 的前 K 字节写测试数据，或者初始化一个固定长度的临时数组。

更实用的场景是配合自定义分配器或栈上数组：

int stack_arr[64];
std::generate_n(stack_arr, 64, []{ return rand() % 100; });

没有 vector 构造开销，没有堆分配延迟，纯栈操作。做嵌入式或高频数据预热时，这点确定性很珍贵。

生成器也不必是 lambda。函数指针、仿函数类、甚至带状态的闭包都行。常见误区是以为只能生成“无状态序列”。其实只要捕获或持有状态，就能支撑复杂逻辑：

struct PrimeGenerator {
    int candidate = 2;
    bool is_prime(int n) {
        if (n < 2) return false;
        for (int i = 2; i * i <= n; ++i)
            if (n % i == 0) return false;
        return true;
    }
    int operator()() {
        while (!is_prime(candidate)) ++candidate;
        return candidate++;
    }
};

// 生成前 15 个质数
std::vector<int> primes(15);
std::generate_n(primes.begin(), 15, PrimeGenerator{});

这里 PrimeGenerator 封装了查找逻辑和当前状态，generate_n 只管调用 15 次。比起手写 while 循环加下标管理，结构清晰，不易出错。

再提一句线程安全：generate_n 本身不保证线程安全，但如果生成器是无状态的（比如 []{return std::rand();}），且目标内存区域互不重叠，你完全可以并行调用多个 generate_n 分段填充——这是手动循环难以优雅实现的。

最后划重点：用 generate_n 前，请务必确认目标区间已就绪。它不会帮你扩容 vector，也不会在 array 越界时抛异常。它像一把精准的刻刀，只负责雕刻，不管木料够不够厚。

下次当你又准备敲 for (int i = 0; i < N; ++i) 时，停半秒：这个循环，是不是只是在机械地“生成然后塞进去”？如果是，generate_n 很可能就是你少写的那行代码——不炫技，不抽象，就让意图干净地落在纸上。