剑指Offer - 面试题1：赋值运算符函数

题目

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

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

private:
	char* m_pData;
};

分析

该代码并不难，但是有几个值得注意的点。
1、返回值声明为当前类型的引用，且返回自身的引用(*this)。目的是为了可以连续赋值.如：
CMyString str1,str2,str3;
str1 = str2 = str3;

2、传入参数的类型是否是常量引用。引用可以避免调用一次复制构造函数，能提高代码的效率，常量可以避免在赋值过程中改变传入的值。

3、是否释放实例自身已有的内存。如果不释放，然后赋值就会导致之前申请的那块内存彻底找不到了，造成内存泄漏。

4、(忽略的点)判断传入的参数和当前的实例(*this)是否为同一个实例(虽然这种情况少)。如果是就直接返回当前实例(*this)。如果我们不加以判断，先释放当前实例的内存，那么传入的参数指向的内存也就被释放了(因为俩者指向同一块内存)，就找不到之前的值了。

代码如下

 CMyString& CMyString::operator=(const CMyString& str)
{
	//判断是否指向同一块地址
	if (this == &str)
	{
		return *this;
	}
	
	//释放自身
	delete[] m_pData;
	m_pData;

	//赋值
	int len = strlen(str.m_pData);
	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)
{
	//判断是否指向同一块地址
	if (this != &str)
	{//交换str和strTemp的m_pData值
		CMyString strTemp(str);			//通过拷贝构造初始化strTemp
		char* pTemp = strTemp.m_pData;	//创建指针保存传入实例值的地址
		strTemp.m_pData = m_pData;		//将当前实例值的地址给到strTemp
	}//出了strTemp的作用后,自动调用strTemp的析构函数。

	return *this;
}

总结

越是简单的题，越要注意细节。考察的一些细微的处理。

本章完