剑指offer——（2.1）面试题1：赋值运算符函数

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

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

private:
	char* m_pData;
};

面试官注意的点：

（1）是否把返回值的类型声明为该类型的引用，并在函数结束前返回实例自身的引用。（即*this）只有返回一个引用，才可以允许连续赋值。

（2）是否把传入的参数的类型声明为常量引用。为避免无谓消耗，提高代码效率。

（3）是否释放实例自身已有的内存。

（ 4）是否判断传入的参数和当前的实例（*this）是不是同一个实例。如果是，直接返回。当*this和传入的参数是同一个实例时，那么一旦释放了自己的内存，传入的参数的内存也同时被释放了，这样就再找不到需要赋值的内容了。

经典解法，初级程序员：

//1.返回值是该类型的引用
CMyString& CMyString::operator =(const CMyString &str)//2.传入的参数为常引用
{
	if(this == &str)//4.判断是否为同一个实例
		return *this;

	delete []m_pData;//3.释放内存
	m_pData = NULL;
	m_pData = new char[strlen(str.m_pData) + 1];
	strcpy(m_pData,str.m_pData);

	return *this;
}

         如果内存不够，new char 抛出异常，m_pData是一个空指针，会程序崩溃。所以先创建一个临时实例，再交换临时实例和原来的实例。代码如下：

CMyString& CMyString::operator =(const CMyString &str)//2.传入的参数为常引用
{
	if(this != &str)
	{
		CMyString strTemp(str);
        
		//strTemp.m_pData与m_pData 交换，由于strTemp是局部变量，运行完if就会被释放，而他所指向就是之前m_pData的内存，所以不需要再专门释放内存了
		char* pTem = strTemp.m_pData;
		strTemp.m_pData = m_pData;
		m_pData = pTemp;
	}

	return *this;
}