题目:如下为类型CMyString的声明,请为该类型添加赋值运算符函数。
class CMyString
{
private:
char * m_pData;
public:
CMyString(char *pData = NULL);
CMyString(const CMyString &str);
~CMyString(void);
};
这里要注意一下几点:
- 是否把返回值的类型声明为该类型的引用,并在函数结束前返回实例自身的引用(*this)。只有返回一个引用,才可以允许连续赋值。
- 是否把传入的形参的类型声明为常量引用。如果传入的参数不是引用而是实例,那么从形参到实参会调用一次复制构造函数。
- 是否释放实例自身已有的内存。如果忘记在重新分配内存之前释放自身已有的空间,程序将出现内存泄露。
- 是否判断传入的参数和当前实例(*this)是不是同一个实例。
经典解法:
CMyString& CMyString::operator = (const CMyString *str)
{
if(this == &str)
return *this;
delete []m_pData;
m_pData = NULL;
m_pData = new char[strlen(str.m_pData) + 1];
strcpy(m_pData, str.m_pData);
return *this;
}
考虑异常安全性的解法:
前面的函数中,我们在分配内存之前先用了delete释放了实例m_pData的内存。如果此时内存不足导致new char抛出异常,m_pData将是一个空指针,CMyString的实例不再保持有效的状态,这就违背了异常安全性原则。
办法是先创建一个临时实例,在交换临时实录和原来的实例。
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是一个局部变量,程序运行到if的外面即除了改变量的作用域,就会自动调用strTemp的析构函数,把strTemp.m_pData所指向的内存释放掉。