c++ transform函数用法_c++容器元素批量转换处理

std::transform是C++标准库中用于对一个或两个输入范围元素执行一元或二元操作并写入目标区间的算法,要求目标区间预先分配足够空间或使用插入迭代器。

transform 是什么,它能做什么

std::transform 是 C++ 标准库中定义在 头文件里的算法函数,用于对一个或两个输入范围的元素执行一元或二元操作,并将结果写入目标迭代器指向的位置。它不修改原容器结构,只做“转换后写入”,因此必须确保目标区间有足够空间(或使用插入迭代器)。

一元 transform:单容器元素批量映射

最常见用法:把 vector 里每个数平方,存进另一个 vector

std::vector src = {1, 2, 3, 4};
std::vector dst(src.size()); // 必须预先分配空间
std::transform(src.begin(), src.end(), dst.begin(),
               [](int x) { return x * x; });

关键点:

  • dst 容器必须已有足够容量,否则写入越界(不会自动扩容)
  • 不能用 std::back_inserter(dst) 直接传给第三个参数再配 dst.begin() —— 这是常见误写
  • 若想边转换边追加,第三个参数应为 std::back_inserter(dst),且 dst 可为空
  • lambda 捕获、函数对象、普通函数指针均可作为一元操作符,但签名必须是 auto operator()(const T&) 形式

二元 transform:双容器逐元素运算

例如把两个 vector 对应位置相加:

std::vector a = {1.1, 2.2, 3.3};
std::vector b = {0.1, 0.2, 0.3};
std::vector res(a.size());
std::transform(a.begin(), a.end(), b.begin(), res.begin(),
               std::plus<>{});

注意:

  • 第二个输入范围(b.begin())长度至少要 ≥ 第一个范围,否则行为未定义
  • std::plus 是 C++14 起支持的透明函数对象,自动推导参数类型;用 std::plus{} 也可,但不够泛化
  • 两输入范围可来自同一容器(如 v.begin()v.begin() + 1),实现滑动窗口类操作

常见错误和绕不开的坑

最容易出问题的是目标迭代器的合法性:

  • std::vector.begin() 作输出,但没调 resize()reserve() → 写入野指针,运行时崩溃或静默数据损坏
  • 误以为 transform 能像 Python 的 map() 那样返回新容器 → 它不构造也不返回容器,只写内存
  • std::array 时传错迭代器范围(比如漏写 .end()),导致部分元素未处理
  • std::string 使用时,忘记 std::stringdata() 不一定以 \0 结尾,但 transform 不关心这个,只要迭代器合法就 OK

真正要小心的从来不是语法,而是你有没有亲手确认过目标区间的可写长度。