面试题：C++中transform自定义操作的基础应用

Question

假设有一个`std::vector<int>`，里面存储了一些整数。请使用`std::transform`算法和自定义操作，将向量中的每个元素都平方，并将结果存储到另一个`std::vector<int>`中。请写出完整的C++代码实现。

Answer 1

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

int square(int num) {
    return num * num;
}

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

    std::transform(original.begin(), original.end(), squared.begin(), square);

    for (int num : squared) {
        std::cout << num << " ";
    }
    std::cout << std::endl;

    return 0;
}