[互联网面试笔试汇总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;

}