剑指offer：面试题1 赋值运算符函数

<pre name="code" class="cpp">//定义一个字符串类 重载赋值运算符（剑指offer：面试题一 赋值运算符函数）

#include <string>
class MyString
{
public:
//在类内部声明带参构造函数
MyString(char* pData = NULL);
//在类内部声明赋值构造函数
MyString(const MyString& str);
//在类内部声明析构函数
~MyString();

//重载等号运算符  即赋值函数
MyString& operator = (const MyString &str);
//在类内部申明打印函数
void print();
private:
char* mData;

};

//在类外定义带参构造函数
MyString::MyString(char* pData)
{
if (pData == NULL)
{
mData = new char[1];
mData[0] = '\0';
}
else{

int length = strlen(pData);
mData = new char[length + 1];
strcpy_s(mData, strlen(pData) + 1, pData);
}
}

//在类外定义复制构造函数
MyString::MyString(const MyString& str)
{
int length = strlen(str.mData);
mData = new char[length + 1];
strcpy_s(mData, strlen(str.mData) + 1, str.mData);
}
//在类外定义析构函数
MyString::~MyString(){
delete mData;
}
/*
//在类外定义赋值函数
MyString& MyString::operator = (const MyString& str)
{
if (this == &str)
return *this;
delete mData;
mData = NULL;
int length = strlen(str.mData);
mData = new char[length + 1];
strcpy_s(mData, strlen(str.mData)+1,str.mData);
return *this;
}*/
/*
上面的赋值函数存在一些程序崩溃的隐患，书中也给出了一个解决方法：
先创建一个临时实例，再交换临时实例和原来的实例。
实现代码如下：
*/
MyString& MyString::operator =(const MyString &str){
if(this != &str){
MyString Tempstr(str);
char* temp = Tempstr.mData;
Tempstr.mData = mData;
mData = temp;
}
return *this;
}

//在类外定义打印函数
void MyString::print(){
printf("%s \n",mData);
}

/**********************测试代码*****************/
void test1()
{
char* text = "hello world!";
MyString str1(text);
MyString str2;
//调用赋值函数
str2 = str1;
str2.print();
}
void test2()
{
char* text = "hello world!";
MyString str1(text);
//调用赋值构造函数
MyString str2(str1);
str2.print();
}
void test3()
{
char* text = "hello world!";
MyString str1(text);
MyString str2;
MyString str3;
//调用赋值函数
str3= str2 = str1;
str2.print();
str3.print();
}
void main()
{
//test1();
//test2();
test3();
}