c++ - 将双*分配给V [3]

我有一个返回double *的函数

double* addVectors( double* vec1, double* vec2 )
{
    double vec[3];
    for( int i = 0; i < 3; i++ )
        vec[i] = vec1[i] + vec2[i];

    return vec;
}

如何将addVectors的结果分配给数组？我做了以下工作，但我敢肯定有一种专业的方法可以做到这一点。

double A[3] = {1, 2, 3};
double B[3] = {4, 5, 6};
double C[3];

double *val = addVectors( A, B );
C[0] = val[0];
C[1] = val[1];
C[2] = val[2];

最佳答案

addVectors返回时，addVectors返回指向本地数组的指针val成为垃圾数据。

您可以将第三个参数传递给addVectors，但是您需要确保所有vec至少具有3个元素:

void addVectors( double* vec1, double* vec2, double* vec)
{
    for( int i = 0; i < 3; i++ )
        vec[i] = vec1[i] + vec2[i];
}

double A[3] = {1, 2, 3};
double B[3] = {4, 5, 6};
double C[3] = {};         // initialize all C elements to 0.0

addVectors(A,B,C);

我的建议是使用STL容器，例如std::vector，std::list etc，或者如果启用C++ 11则使用st::array。

使用STL容器可以存档您的意图:

std::vector<double> addVectors(const std::vector<double>& v1,
                               const std::vector<double>& v2)
{
    std::vector<double> v;

    for( int i = 0; i < 3; i++ )
    {
        v.push_back(v1[i] + v2[i]);
    }

    return v;
}


int main()
{

    std::vector<double> A = {1, 2, 3};
    std::vector<double> B = {4, 5, 6};

    std::vector<double> C(3, 0.0);

    std::vector<double> D = addVectors(addVectors(A,B),C);

    for(auto& e : D)
    {
        cout << e << endl;
    }
}