免费覆盖前代码，如何修复内存泄漏？

Question

我有一个使用2D数组的两个副本并对它们进行一些数学运算的函数，但我可以在覆盖之前释放。代码是这样的：

int **function(int **A)
{
    int **B = (int **)malloc(N*sizeof(int *));
    for(i = 0; i < N; i++)
         B[i] = (int *)malloc(M*sizeof(int));
    //the operations i need to do

   return B;
}

在main中，调用就像这样（在int **中的A）：

A = function(A);

有没有办法在不改变代码的情况下摆脱泄漏？