[互联网面试笔试汇总C/C++-9] 实现赋值运算符函数-剑指offer

题目：如下为类型CMyString的声明，请为该类型添加赋值运算符函数。

class CMyString
{
public:
CMyString(char* pData = NULL);
CMyString(const CMyString & str);
~CMyString(void);
private:
char* m_pData;
};

分析：

赋值运算符要注意以下四点：

1.把返回值的类型声明为该类型的引用。

2.把传入的参数的类型声明为常量引用。

3.释放实例自身已有的内存。

4.判断传入的参数和当前的实例是否是同一个实例。

根据以上几点要求，我们可以写出如下赋值运算符。

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;
}

对于初级C++程序员能写出上述函数就OK了，但是如果要作为一个高级C++程序员的话，则需要考虑更多的安全性问题。

例如上面代码中的new部分要是内存不足则会抛出异常，此时实例的m_pData已经被删除，而新的内存又没申请到，就会导致数据丢失，所以我们可以采用下面的代码来保证代码的安全性。

CMyString & CMyString::operator = (const CMyString & str)
{
if(this != & str)
{
CMyString tmp(str);
char* pTmp = tmp.m_pData;
tmp.m_pData = m_pData;
m_pData = pTmp;
}
return *this;
}