赋值运算符函数

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

---

#include "stdafx.h"#includeclass CMyString

{

public:

CMyString(char* pData = NULL);

CMyString(const CMyString& str);

~CMyString(void);

CMyString& operator = (const CMyString& str);

void Print();

private:

char* m_pData;

};

CMyString::CMyString(char *pData)

{

if(pData == NULL)

{

m_pData = new char[1];

m_pData[0] = '\0';

}

else

{

int length = strlen(pData);

m_pData = new char[length + 1];

strcpy(m_pData, pData);

}

}

CMyString::CMyString(const CMyString &str)

{

int length = strlen(str.m_pData);

m_pData = new char[length + 1];

strcpy(m_pData, str.m_pData);

}

CMyString::~CMyString()

{

delete[] m_pData;

}

---

该类题型一定要注意以下几点：

1.是否把返回值类型声明为该类型的引用。

只有返回一个引用，才可以连续赋值。

2.是否把传入的参数的类型声明为常量引用。

把参数类型声明为常量引用，可以避免从形参到实参所调用构造函数这一无谓的消耗。(别忘了const)

3.是否释放实例已有的内存

如果忘记，则当指针指向其他内存时，会出现内存泄漏

4.是否判断传入的参数和当前的实例是不是同一实例

如果*this和传入的参数是同一实例，则不能继续，因为如果继续的话，先释放实例已有的内存，再申请新的内存，这样会释放自身的内存，传入的参数的内存也被释放了，因此再也找不到需要赋值的内容了。

---

根据以上几点，写出这样的代码:

CMyString& CMyString::operator = (const CMyString& str)

{

if(this == &str)

return *this;

delete []m_pData;

m_pData = NULL;

m_pData = new char[strlen(str.m_pData) + 1];

strcpy(m_pData, str.m_pData);

return *this;

}

但是，这样并不完美，因为是先释放的内存，后申请的空间，因此如果当内存不足的时候，实例自身的内存会被释放，同时又没有新的值给它，m_pData将是一个空指针，这样程序的内存容易崩溃。

---

因此有如下解决方法:一个简单的做法是先用new分配，再释放已有内存，这样只有在分配成功之后再释放原来的内存，更好的方法是借助一个临时变量，再交换临时实例和原来的实例。

CMyString& CMyString::operator = (const CMyString& str)

{

if (this!=&str) {

CMyString strTemp=CMyString(str);

char *pTemp=strTemp.m_pData;

strTemp.m_pData=m_pData;

m_pData=pTemp;

}

return *this;

}

如果对重点标注的代码表示不理解，那么请看下面:

m_pData为实例自身的内存，而strTemp为一个局部变量，当程序出了if之后，就会释放掉strTemp，这样原来m_pData的内存就会释放掉，否则会出现内存泄漏。