剑指offer之赋值操作符

题目描述：

个人觉得，类中四少（构造函数、拷贝构造函数、赋值操作符、析构函数）都有一些注意点，于是都实现了下，注意点均有标注（注意下^_^）

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

class CMyString{
public:

	/*
		赋值语句：（四部曲）
			1：检查自赋值
			2：释放原有空间
			3：开辟新空间并赋值
			4：return *this
	*/

	CMyString& operator=(const CMyString &str){
		if (this != &str){
			delete[] m_pData;//1
			m_pData = NULL;
			int len = strlen(str.m_pData) + 1;
			m_pData = new char[len];//2
			strcpy_s(m_pData, len, str.m_pData);//3
		}
		return *this;//4
	}
	explicit CMyString(const char *pData = NULL){//explicit防止类型的隐式转换：如CMyString str = "hello";
		if (pData == NULL){//为空的情况要单独考虑(strlen(NULL)会导致程序崩溃)
			m_pData = new char[1]; //即使开辟一个空间也使用new[]---->与析构函数delete[]相对应
			m_pData[0] = '\0';
		}
		else{
			int len = strlen(pData) + 1;
			m_pData = new char[len];
			strcpy_s(m_pData, len, pData);
		}
	}

	CMyString(const CMyString &str){
		int len = strlen(str.m_pData)+1;
		m_pData = new char[len];
		strcpy_s(m_pData, len, str.m_pData);
	}

	virtual ~CMyString(void){//析构函数一般声明为虚函数（继承时保证析构函数的调用顺序，防止内存泄露）
		delete[] m_pData;
		m_pData = NULL;
	}
private:
	char *m_pData;
};

赋值操作符函数：

版本一：

//版本一：先释放原有空间->开辟新空间 如new开辟空间失败，那么原有的空间(内容)也不复存在（未考虑异常安全）
	CMyString& operator=(const CMyString &str){
		if (this != &str){
			delete[] m_pData;//1
			m_pData = NULL;
			int len = strlen(str.m_pData) + 1;
			m_pData = new char[len];//2
			strcpy_s(m_pData, len, str.m_pData);//3
		}
		return *this;//4
	}

版本二：

//版本二：保证空间开辟失败，原有的内容依旧存在
CMyString& operator=(const CMyString &str){
	if (this != &str){
		int len = strlen(str.m_pData) + 1;
		char *tmp = new char[len];//new 失败时会发出异常,不会执行到下一步delete[] m_pData
		delete[] m_pData;         //若执行到下一步delete[] m_pData,说明空间开辟成功
		m_pData = tmp;
		strcpy_s(m_pData, len, str.m_pData);
	}
	return *this;
}

版本三：（推荐）

//版本三:利用局部临时对象的生存期
CMyString& operator=(const CMyString &str){
	if (this != &str){
		CMyString tmp(str.m_pData);
		swap(m_pData, tmp.m_pData);//交换各自m_pData所指的空间，临时对象出了作用域if，调用析构函数释放m_pData
								   //所指向的空间
	}
	return *this;
}

测试用例：

int main()
{
	CMyString str("hello");
	str = str;//自赋值( if(this != &str) )

	CMyString str1("xxxxxx");//一般赋值
	str1 = str;

	CMyString str3;
	str3 = str = str1;//连续赋值（返回CMystring &的重要性）
	system("pause");
	return 0;
}

注：

1、#include<vld.h> 是一款用来检测内存泄露的工具(vld)-->因为operator=中涉及众多内存释放开辟

（养成良好编程习惯）点击打开链接（vld使用教程）

2、开发平台：VS2013（所以使用了strcpy_s函数）