C++中std::transform的使用

使用std::transform函数需要包含<algorithm>头文件

• 一元操作

std::transform 是 C++ 标准库中的一个算法，用于将一个序列中的每个元素进行转换，并将结果存储在另一个序列中。该算法可以接受两个输入序列，对应位置的元素进行操作，然后将结果存储在输出序列中，也可以接受一个输入序列，对其进行转换后直接覆盖原序列。

以下是 std::transform 函数的通用签名

template <class InputIterator, class OutputIterator, class UnaryOperation>
OutputIterator transform(InputIterator first1, InputIterator last1,
                          OutputIterator result, UnaryOperation op);

std::transform 接受四个参数：

• first1, last1: 表示输入序列的起始和结束迭代器，包括起始位置但不包括结束位置。
• result: 表示输出序列的起始迭代器，用于存储转换后的结果。
• op: 一个一元函数对象（UnaryOperation），用于对输入序列中的每个元素进行转换操作。

以下是一个简单的示例，演示了如何使用 std::transform 对一个向量中的每个元素进行平方操作，并将结果存储在另一个向量中：

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

int main() {
    std::vector<int> input = {1, 2, 3, 4, 5};
    std::vector<int> output(input.size());

    // 对 input 中的每个元素进行平方操作，结果存储在 output 中
    std::transform(input.begin(), input.end(), output.begin(),
                   [](int x) { return x * x; });

    // 输出转换后的结果
    for (int num : output) {
        std::cout << num << " ";
    }
    std::cout << std::endl;

    return 0;
}

在上面的示例中，我们使用 std::transform 对 input 中的每个元素进行平方操作，并将结果存储在 output 中。通过提供一个 lambda 函数 [](int x) { return x * x; } 作为转换操作，我们将每个元素平方后存储在 output 中。

总之，std::transform 是一个非常有用的算法，用于对序列中的元素进行转换操作，并将结果存储在另一个序列中。

• 二元操作

std::transform 函数还可以支持二元操作符，用于对两个输入序列中对应位置的元素进行操作，并将结果存储在输出序列中。以下是 std::transform 函数的另一个通用签名，用于支持二元操作符：

template <class InputIterator1, class InputIterator2, class OutputIterator, class BinaryOperation>
OutputIterator transform(InputIterator1 first1, InputIterator1 last1,
                         InputIterator2 first2, OutputIterator result, BinaryOperation binary_op);

std::transform 函数接受五个参数：

• first1, last1: 表示第一个输入序列的起始和结束迭代器，包括起始位置但不包括结束位置。
• first2: 表示第二个输入序列的起始迭代器。
• result: 表示输出序列的起始迭代器，用于存储两个输入序列对应位置元素进行操作后的结果。
• binary_op: 一个二元函数对象（BinaryOperation），用于对两个输入序列中对应位置的元素进行操作。

以下是一个示例，演示了如何使用二元操作符的 std::transform 对两个向量中对应位置的元素进行相加操作，并将结果存储在另一个向量中：

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

int main() {
    std::vector<int> vec1 = {1, 2, 3, 4, 5};
    std::vector<int> vec2 = {10, 20, 30, 40, 50};
    std::vector<int> result(vec1.size());

    // 对 vec1 和 vec2 中对应位置的元素进行相加操作，结果存储在 result 中
    std::transform(vec1.begin(), vec1.end(), vec2.begin(), result.begin(),
                   [](int x, int y) { return x + y; });

    // 输出相加后的结果
    for (int num : result) {
        std::cout << num << " ";
    }
    std::cout << std::endl;

    return 0;
}

在这个示例中，我们使用二元操作符的 std::transform 对 vec1 和 vec2 中对应位置的元素进行相加操作，并将结果存储在 result 中。通过提供一个 lambda 函数 [](int x, int y) { return x + y; } 作为二元操作符，我们将两个向量中对应位置的元素相加后存储在 result 中。

总之，std::transform 函数不仅支持一元操作符，还支持二元操作符，可以非常方便地对两个序列中对应位置的元素进行操作，并将结果存储在输出序列中。