由返回值为引用而造成的内存泄露

来源:互联网 发布:mac怎么关闭隐藏文件 编辑:程序博客网 时间:2024/06/04 17:50

一个很纠结的问题:
比如两个对象进行操作,比如两个矩阵相乘,在函数中,我们会new一个新的矩阵,并作为引用返回。
如果A = B * C在一个循环里面,那么不停地构造乘积矩阵,最后造成了内存的泄露。这个问题应该怎么解决呢?
CMatrix & CMatrix::operator *(const CMatrix &mat)
{
     CMatrix *pmat;
     pmat = new CMatrix(m, n);
     .....
     return *pmat;
}

CMatrix & CMatrix::operator =(const CMatrix &mat)
{
     if(&mat == this)
     {
          return * this;
     }
     m_row = mat.m_row;
     m_line = mat.m_line;
     m_data = new double [..];
     memcpy(m_data, mat.m_data, ...);
     return *this;
}

例如下面的使用,会造成内存的泄露,在A*B中,构造的矩阵,在=操作时,并没有释放,以后也没有机会释放了。

for(i=0; i<100; i++)
{
      CMatrix C = A * B;
}

在乘法里分配的pmat怎么才能释放掉呢?

OK,换个思路,我们的返回值如果定义为一个临时变量,在=操作以后,就会被释放掉,
CMatrix CMatrix::operator *(const CMatrix &mat)
{
     CMatrix  pmat(m, n);
     .....
     return pmat;
}

原创粉丝点击