c++ - 如何阻止这种内存泄漏

Question

我正在学习 C++ 中的指针，并且正在研究新功能和删除功能。 我有一个在堆上分配内存的本地函数，但是因为我要返回我创建的二维数组，所以我不明白如何堵住这个内存泄漏，我们将不胜感激

主要.cpp

#include<iostream>
#include "integers.h"
using namespace std;
int main()
{
int i[]={1,2,3,4};
int n=sizeof(i)/sizeof(int);
cout<<n<<endl;
printint(genarr(i,n),n);
}

整数.cpp

#include<iostream>
using namespace std;
int** genarr(int* val,int n)
{
int i,j;
int **a=new int*[n];
for(i=0;i<n;i++)
a[i]=new int[n];

for(i=0;i<n;i++)
for(j=0;j<n;j++)
if(i==j)
a[i][j]=val[i];

return a; // The variable that will leak but because i am returning it , how do stop it
}

void printint(int** a,int n){
for(int i=0;i<n;i++)
 {
  for(int j=0;j<n;j++)
   {
    cout<<a[i][j]<<" ";
   }
cout<<endl;
}
return ;
}

整数.h

int** genarr(int*val, int n);
void printint(int **a,int n);

编译者

g++ main.cpp integers.cpp -o integers

我听说过智能指针，并计划在这之后了解它们，但现在我想知道是否有办法解决这个问题，还是我应该直接使用智能指针？

Answer 1

要解决此问题，您需要删除您新的内容。

将 main 中的代码更改为:

int **arr = genarr(i,n);
printint(arr,n);
// we're done using arr; now we need to free it
for(int j=0;j<n;j++)
    delete[] arr[j];
delete[] arr;

---

您还可以扩展 integers.cpp 并添加一个 delarr 函数来补充 genarr:

void delarr(int **a, int n) {
    for (int i = 0; i < n; i++) {
        delete[] a[i];
    }
    delete[] a;
}

然后 main 变得简单:

int **arr = genarr(i,n);
printint(arr,n);
delarr(arr,n);