賦值運算符函數

題目：如下為類型CMyString的聲明，請為該類型添加賦值運算符函數。

class CMyString
{
public:
CMyString(char *pData=NULL);//構造函數
CMyString(const CMyString& str);//拷貝構造函數
~CMyString();//析構函數
private:
char* m_pData;//數據域，字符指針
};

關注如下幾點:是否把返回值類型聲明為該類型的引用，並在函數結束前返回實例的自身引用。是否把傳入的參數的類型聲明為常量引用。是否釋放實例自身的內存。是否判斷傳入的參數和當前的實例是不是同一個實例。

#include "stdafx.h"
#include<string>
class 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;
}

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;
}

void CMyString::Print()
{
printf("%s\n", m_pData);
}

int main()
{
char* text = "Hello world";
CMyString str1(text);
CMyString str2;
str2 = str1;

printf("The expected result is: %s.\n", text);

printf("The actual result is: ");
str2.Print();

return 0;
}