题目:
如下为类型CMyString的声明,请为该类型添加赋值运算符函数。
class CMyString
{
public:
CMyString(char* pData = nullptr);
CMyString(const CMyString& str);
~CMyString(void);
private:
char* m_pData;
};
分析:
- 把返回值的类型声明为该类型的引用,并在函数结束前返回实例自身的引用(*this)。只有返回一个引用,才可以允许连续赋值。否则,如果函数的返回值是void,则应用该赋值运算符将不能进行连续赋值。
- 把传入的参数类型声明为常量应用。如果传入的参数是实例,那么由形参到实参会调用一次复制构造函数,把参数声明为引用可以避免复制操作,提高效率。同事,我们在赋值运算符内不会改变传入的实例的状态,因此要为传入的引用参数加上const关键字。
- 释放实例自身已有的内存,如果在分配新内存之前忘记释放自身已有的空间,则会出现内存泄漏。
- 判断传入的参数和当前实例(*this)是不是同一个实例。如果是同一个实例,直接返回即可。如果不判断就进行赋值,在释放自身内存时候会导致严重的问题:当*this和传入参数是同一个实例时,一旦释放了自身内存,传入的参数内存也被释放了,再也找不到需要赋值的内容了。
虽然说学过C++,不过C++代码写的不多,作为一个Java程序员,对重载运算符并不很熟悉,因为Java里没有运算符重载这一说。对于数值型的运算操作,运算符重载可能会带来很大的便利。运算符重载可以理解成方法的重载。
解法:
经典的解法,适用于初级程序员
CMyString& CMyString::operator=(const CMyString &str)
{
if (this == &str)
{
return *this;
}
delete []m_pData;
m_pData = nullptr;
m_pData = new char[strlen(str.m_pData) + 1];
strcpy(m_pData, str.m_pData);
return *this;
}
考虑异常安全性的解法,高级程序员必备
在前面的函数中,我们在分配内存前行delete释放了实例m_pData的内存。后面有new char,如果此时内存不足将会导致new抛出异常,m_pData将是一个空指针,后面就会导致程序崩溃。
要解决这个问题,有两种方法。一种是先用new分配新的内容,后delete释放已有内存,这样,只有在分配成功后,再用delete释放原有内存,如果分配失败,m_pData的内容不会被修改。另一种是先创建一个临时实例,交换临时实例和原来的实例。
CMyString& CMyString::operator=(const CMyString &str)
{
if (this != &str)
{
CMystring strTemp(str);
char* pTemp = strTemp.m_pData;
strTemp.m_pData = m_pData;
m_pData = pTemp;
}
return *this;
}
先创建一个临时实例strTemp,把strTemp.m_pData和实例自身的m_pData进行交换。由于strTemp是一个局部变量,程序运行到if外就出了该变量的作用域,自动调用strTemp的析构函数,把strTemp.m_pData所指向的内存释放掉,由于strTemp.m_pData指向的内存是实例之前m_pData的内存,这就相当于自动调用了析构函数释放实例的内存。
如果因为内存不足new操作抛出异常,但我们还没有修改原来实例的状态,这就保证了异常安全性。