类的赋值运算符的重载函数

时间:2022-07-14 01:40:19

题目:类CMyString的声明如下,请实现其赋值运算符的重载函数,要求异常安全,即当对一个对象进行赋值时发生异常,对象的状态不能改变。

class CMyString
{
public:
CMyString(
char* pData = NULL);
CMyString(
const CMyString& str);
CMyString
& operator = (const CMyString& str);
~CMyString();

private:
char* m_pData;
};

答:

//1、可能有异常
CMyString& CMyString::operator = (const CMyString& str)
{
if (this != &str)
{
delete [] m_pData;
m_pData
= NULL;
m_pData
= new char[strlen(str.m_pData) + 1]; //这里内存分配不成功,之前的数据已经释放,不安全
strcpy(m_pData, str.m_pData);
}
return *this;
}

//2、异常安全
CMyString& CMyString::operator = (const CMyString& str)
{
if (this != &str)
{
CMyString tmp(str);
char *pTmp = m_pData;
m_pData
= tmp.m_pData;
tmp.m_pData
= pTmp;
}
return *this;
}