📜  向量之和 c++ (1)

📅  最后修改于: 2023-12-03 15:22:58.630000             🧑  作者: Mango

向量之和

在C++中,可以使用STL中的vector来存储一系列的元素。在有时候需要计算向量之和时,可以使用vector来实现。

实现方法

以下是一种简单的calcSum函数实现向量之和的方法:

#include <iostream>
#include <vector>

using namespace std;
vector<int> calcSum(vector<int> &v1, vector<int> &v2)
{
    vector<int> res;
    if (v1.size() != v2.size())
        return res;
    int n = v1.size();
    res.resize(n);
    for (int i = 0; i < n; i++)
        res[i] = v1[i] + v2[i];
    return res;
}

这个函数的参数是两个vector类型的变量v1和v2,分别表示两个向量。首先检查两个向量的长度是否相等,如果不等则直接返回一个空的向量。否则,计算它们的和,并将结果存储在一个新的vector类型变量res中,最后返回这个res向量。

示例

下面是一段简单的程序,展示了如何使用calcSum函数计算两个向量的和:

#include <iostream>
#include <vector>

using namespace std;

vector<int> calcSum(vector<int> &v1, vector<int> &v2)
{
    vector<int> res;
    if (v1.size() != v2.size())
        return res;
    int n = v1.size();
    res.resize(n);
    for (int i = 0; i < n; i++)
        res[i] = v1[i] + v2[i];
    return res;
}

int main()
{
    vector<int> v1 = {1, 2, 3};
    vector<int> v2 = {4, 5, 6};
    vector<int> res = calcSum(v1, v2);
    for (auto i : res)
        cout << i << " ";
    cout << endl;
    return 0;
}

在这个程序中,我们定义了两个vector类型的变量v1和v2,分别初始化为{1,2,3}和{4,5,6},然后调用calcSum函数计算它们的和,并将结果存储在变量res中。最后,我们使用一个循环打印输出res中的元素,结果为{5,7,9}。

总结

在C++中,我们可以使用vector来表示向量,并使用简单的代码计算向量之和。这个过程中需要注意检查向量的长度是否相等,否则无法计算其和。