从 C++ 中的函数返回指向数组的指针？

Question

我是一名初级程序员，我有一个关于在 C++ 中返回指向双精度数组指针的函数的问题。该函数采用两个数组并将每个元素相加，就像向量的总和一样。

我认为正确的做法是......

double *myfunction(double *x, double *y, int n){
  double *r = new double[n];
  for(int i=0;i<n;i++){
    r[i] = x[i]+y[i];
  }
  return r;
}

问题是我在这样的主函数的while循环中使用该函数

int main(){
  double *x, *y, *s;
  x = new double[2];
  y = new double[2];
  x = {1,1};
  y = {2,2};

  while(/*some condition */){
    /*some process...*/

    s = myfunction(x,y, 2);

    /*some process...*/
  }

  delete[] x;
  delete[] y;
  delete[] s;
}

我的问题是内存泄漏怎么办？每次我使用“myfunction”（在while循环内）我为变量“s”保留内存，这意味着如果while循环执行5次，那么程序为变量“s”保留5倍的内存“？

有没有办法做到这一点（从函数返回指向数组的指针并在循环中使用该函数）？？

先谢谢你了。

Answer 1

你问：

每次我使用“myfunction”（在while循环中）时，我都会为变量“s”保留内存，这意味着如果while循环执行5次，那么程序会为变量“s”？

答案是是的。

你还问过：

有没有办法做到这一点（从函数返回指向数组的指针并在循环中使用该函数）？？

答案是是的。需要添加代码删除返回的内存，

  while( /*some condition */){
    s = myfunction(x,y, 2);

    // Use s

    // Now delete s.
    delete[] s;
  }

比处理new 和delete 更好的解决方案是从myfunction 返回std::vector。然后，您无需担心管理内存。那是the answer by Barray

Answer 2

我想说写myfunction 更正确的方式是：

std::vector<double> myfunction(double *x, double *y, int n){
  std::vector<double> r;
  r.reserve(n);
  for(int i=0;i<n;i++){
    r.push_back(x[i]+y[i]);
  }
  return r;
}

这样，您不必担心内存泄漏，而您的 while 循环可以是：

while (/* some condition*/) {
    std::vector<double> s = myfunction(x, y, 2);
    // whatever
}

Answer 3

为避免内存泄漏，您需要在获取后立即使用s，并在完成后将其删除。

 int main(){
      double *x, *y, *s;
      x = new double[2];
      y = new double[2];
      x = {1,1};
      y = {2,2};

      while(//some condition ){
        s = myfunction(x,y, 2);
        //do some process here
        delete[] s;
      }
      delete[] x;
      delete[] y;

    }