Question

我有一个返回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];

Answer 1

当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;
    }
}

Answer 2

从函数返回数组的另一种方法是将其嵌入到结构中，因为这样就会复制内容：

typedef struct
{
  double vec[3];
} vecret;


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

将双*分配给V [3]

2 个答案: