菜鸟被虐记之《剑指offer》——(题 1)

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

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

该题需要注意的地方：

1. 返回值必须返回对象的引用，否则不能连续赋值
2. 传入的形参应该设置为常量引用，防止被修改
3. 拷贝构造之前，先判断对象是否有额外开辟的内存，有的话释放掉
4. 判断自赋值的情况。

方法一思路：（书上所说的菜鸡法）

1. 进入赋值运算符重载函数首先判断是不是自赋值情况，如果是的话直接返回*this。
2. 如果不是自赋值的情况，先判断被赋值的对象的m_pData是否为空；否，释放m_pData原有的内存。
3. 给被赋值对象的m_pData开辟内存，并拷贝字符串。

程序代码：

#include <iostream>
#include <string.h>
using namespace std;

class CMyString
{
public:
	CMyString();    //默认构造函数
	CMyString(char *pData);    //构造函数
	CMyString &operator=(const CMyString &str);   //赋值运算符重载函数
	~CMyString();    //析构函数
	void Show();     //打印函数
private:
	char *m_pData;
};

CMyString :: CMyString()
	{
		m_pData = NULL;
	}
CMyString :: CMyString(char *pData)  //带参数的构造函数，构造步骤：①开辟空间 ②赋值
{
	if(m_pData != NULL)
	{
		m_pData = new char[strlen(pData)+1];
		strcpy(m_pData, pData);
	}
}

CMyString &CMyString :: operator=(const CMyString &str)   
{
	if(this == &str)   //自赋值情况,直接返回
	{
		return *this;
	}
	if(m_pData != NULL)
	{
		delete []m_pData;
		m_pData = NULL;  //置空防止m_pData成为野指针
	}
	m_pData = new char[strlen(str.m_pData) + 1];
	strcpy(m_pData,str.m_pData);
	
	return *this;     //返回 *this 方便连续赋值
}
CMyString :: ~CMyString()    //析构函数
{
	delete []m_pData;
	m_pData = NULL;
}

void CMyString :: Show()     //打印函数
{
	cout << m_pData << endl;
}

int main()
{
	CMyString str1(" 我是小菜鸡 ");
	cout << "str1.m_PData = ";
	str1.Show();

	CMyString str2; //默认构造str2
	CMyString str3;
	str3 = str2 = str1;    //str2调用赋值运算符重载函数

	cout << "str2.m_PData = ";
	str2.Show();
	cout << "str3.m_PData = ";
	str3.Show();
	return 0;
}

程序运行截图

方法二思路：（考虑到异常安全性的解法）

1. 先调用拷贝构造函数构造临时量pTemp，构造失败，抛出异常
2. pTemp构造成功，将其m_pData和被赋值对象的m_pData指向的地址交换
3. 析构pTemp，返回 *this。

注：这里所说的安全性，是指如果使用方法一的办法，万一此时内存不够了，则会提示用户内存申请失败；但由于之前被赋值对象原有的内存也被释放了；被赋值对象不在保持有效状态，因此不安全。
（打个比方：你在投股，你先把你的房子卖了，然后股市崩了，你变得一无所有；方法二的思想就是无论如何得保证股市崩了之后，你的房子还在）

程序代码

CMyString &CMyString :: operator=(const CMyString &str)   
{
	if(this!= &str)                                     //自赋值情况,直接返回
	{
		CMyString strTemp(str);               /*拷贝构造函数定义临时量；如果构造失败， 则直接抛出内存不足的异常，
											下面的交换代码将不会执行，因此，保留了被赋值对象的有效状态*/
		
		char *pTemp = strTemp.m_pData;
		strTemp.m_pData = m_pData;     //将被赋值对象之前的m_pData内存保存在临时量中
		m_pData = pTemp;                       //更新被赋值对象的m_pData值
	}
	
	/* pTemp 是if语句中定义的临时量，出了if作用域，自动调用析构，
	因为此时pTemp指向的是被赋值对象原来的内存，因此释放了被赋值对象原来的内存*/

	return *this;     //返回 *this 方便连续赋值
}