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

时间:2015-05-26 02:54:03

标签: c++ arrays function pointers return

我是一个初学程序员,我有一个关于一个函数的问题,该函数返回一个指向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倍的内存“?

是否存在这样做的方法(从函数返回指向数组的指针并在循环中使用该函数)??

先谢谢你。

3 个答案:

答案 0 :(得分:4)

为避免内存泄漏,您需要在获得内容后立即使用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;

    }

答案 1 :(得分:3)

我说更正确的写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
}

答案 2 :(得分:2)

你问:

  

每次我使用“myfunction”(在while循环中)我为变量“s”保留内存,这意味着如果while循环执行5次,那么程序会为内存保留5倍的内存变量“s”?

答案是

您还问:

  

是否存在这样做的方法(从函数返回指向数组的指针并在循环中使用该函数)??

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

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

    // Use s

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

比处理newdelete更好的解决方案是从std::vector返回myfunction。然后,您不必担心管理内存。那是the answer by Barray