剑指offer面试题1:赋值运算符函数（C++版）

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

考察点：

1、是否把返回值的类型声明为该类型的实例，并在函数结束前返回实例自身的引用(*this)：以便连续赋值操作。

2、是否把传入的参数的类型声明为常量引用。原因见代码注释。

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

4、是否判断传入的参数和当前的实例（*this）是不是同一个实例。

#include <iostream>
#include <string>
#include <cstring>
using namespace std;

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

private:
	char* m_pData;
};

CMyString::CMyString(char* pData)
{
	if(NULL == pData){
		m_pData = NULL;	
		return;	
	}
	else{
		unsigned int len = strlen(pData) + 1;
		m_pData = new char[len];
		for(int i = 0; i < len; i ++){//此处也可以直接用strcpy(m_pData, pData)
			m_pData[i] = pData[i];
		}
	}
}
CMyString::CMyString(const CMyString& str)
{
	unsigned int len = strlen(str.m_pData) + 1;
	m_pData = new char[len];
	for(int i = 0; i < len; i ++){//此处也可以直接用strcpy(m_pData, pData)
		m_pData[i] = str.m_pData[i];
	}
}

CMyString::~CMyString()
{
	delete []m_pData;
}
/*需要考虑的点：
1、返回值的类型为引用：可以连续赋值，比如str1 = str2 = str3，若返回类型为void，不支持连续赋值。
2、传入的参数设置为const引用：如果传入的参数是实例，从形参到实参会调用一次复制构造函数，声明为引用可以避免这样无谓的浪费，可提高代码效率;
同时我们在赋值运算符函数内不会改变传入的实例的状态，因此应将引用参数加上const关键字，以防意外修改。
3、释放实例自身已有的内存：赋值之前应释放实例已有的内存，再分配适合赋值的大小的内存。不释放，程序将出现内存泄露。
4、判断传入的参数和当前的实例（*this）是不是同一个实例：如果是，不需要进行赋值操作，直接返回。如果不判断，在释放实例自身内存时会出现严重问题，
因为参数和*this是同一个实例，一旦释放了自身内存，传入的参数的内存也被释放了。
*/

CMyString& CMyString::operator=(const CMyString &str)
{
	if(this != &str){//this:指向当前对象本身
		char* tmp = new char[strlen(str.m_pData) + 1];//strlen()不计入最后'\0'，所以+1
		strcpy(tmp, str.m_pData);//strcpy()把从源串从起始地址开始到'\0'复制到目的串
		delete []m_pData;//释放原来占的内存，注意m_pData的类型为char*,相当于数组
		m_pData = tmp;
	}
	return *this;
}

/*
或者直接创建一个str的实例strTemp作为临时交换实例，然后把strTemp.m_pData和当前对象本身的m_pData做交换。由于strTemp是局部变量，程序运行到if外面
的时候也就出了改变量的作用域，会自动调用strTemp的析构函数，把strTemp.m_pData所指的内存释放掉，也就是当前对象的m_pData之前指向的内存。
*/

CMyString& CMyString:: operator=(const CMyString &str)
{
	if(this != &str){
		CMyString strTemp(str);
		char *pTemp = strTemp.m_pData;
		strTemp.m_pData = m_pData;
		m_pData = pTemp;
	}
	return *this;
}
void CMyString::print()
{
	cout << m_pData << endl;
}
void test()
{
	char test[] = "hello world";
	CMyString myStr1(test);
	CMyString myStr2;
	myStr2 = myStr1;
	cout << test << endl;
	myStr1.print();
	myStr2.print();

	CMyString myStr3(test);
	myStr1 = myStr3;
	myStr1.print();

	CMyString myStr4;
	myStr4 = myStr2 = myStr1;
	myStr4.print();
}
int main()
{
	test();

	return 0;
}

测试用例：

把一个CMyString的实例赋给另外一个实例；

把一个CMyString的实例赋值给它自己。

连续赋值。