我是一个初学程序员,我有一个关于一个函数的问题,该函数返回一个指向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倍的内存“?
是否存在这样做的方法(从函数返回指向数组的指针并在循环中使用该函数)??
先谢谢你。
答案 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;
}
比处理new
和delete
更好的解决方案是从std::vector
返回myfunction
。然后,您不必担心管理内存。那是the answer by Barray