std::transform 用于转换容器元素,支持一元和二元操作,需确保输出空间足够。

在C++中,std::transform 是一个非常实用的标准库算法,定义在 algorithm 头文件中。它用于将一个容器中的元素通过指定操作转换后,写入另一个容器或原地更新。相比手动遍历,使用 std::transform 更加简洁、安全且易于理解。
基本语法与参数说明
std::transform 有两种常用形式:
- std::transform(input_begin, input_end, output_begin, unary_op)
 - std::transform(first1_beg, first1_end, first2_beg, output_begin, binary_op)
 
其中:
- 前两个参数是输入范围的起始和结束迭代器。
 - 第三个参数是输出容器的起始位置(需确保空间足够)。
 - 第四个参数是函数对象、Lambda 表达式或函数指针,用于定义转换规则。
 - 第二个版本支持两个输入序列,进行二元操作。
 
对每个元素进行一元操作
常见场景是将一个 vector 中的所有元素翻倍或取平方:
立即学习“C++免费学习笔记(深入)”;
#include <algorithm> #include <vector> #include <iostream> <p>int main() { std::vector<int> nums = {1, 2, 3, 4, 5}; std::vector<int> result(nums.size()); // 预分配空间</p><pre class='brush:php;toolbar:false;'>std::transform(nums.begin(), nums.end(), result.begin(), [](int x) { return x * x; }); for (int n : result) std::cout << n << " "; // 输出: 1 4 9 16 25
}
这里使用 lambda 将每个元素平方,并写入 result 容器。注意:输出容器必须有足够空间,否则会引发未定义行为。
使用二元操作合并两个容器
当需要对两个容器对应位置的元素进行操作时,例如向量加法:
 std::vector<int> a = {1, 2, 3}; std::vector<int> b = {10, 20, 30}; std::vector<int> sum(3); <p>std::transform(a.begin(), a.end(), b.begin(), sum.begin(), [](int x, int y) { return x + y; });</p>
这段代码将 a 和 b 对应元素相加,结果存入 sum。注意第二个输入序列从 b.begin() 开始,长度应不小于第一个范围。
原地修改容器元素
若想直接修改原容器,可将目标位置设为源容器的起始迭代器:
 std::transform(nums.begin(), nums.end(), nums.begin(),                [](int x) { return x * 2; }); 
这会把 nums 中每个元素乘以 2。只要操作不依赖后续值,这种原地转换是安全的。
基本上就这些。掌握 std::transform 能让你写出更清晰、函数式风格的 C++ 代码。关键是传对迭代器,提供合适的转换函数,并保证输出空间足够。不复杂但容易忽略细节。


