剑指Offer系列---(3)赋值运算符函数

1.题目描述：
赋值运算符函数
2.考虑情况：
1）是否把返回值的类型声明为该类型的引用，并在函数结束前返回实例自身的引用（即*this）。只有返回一个引用，才可以允许连续赋值。否则如果函数的返回值是void，应用该赋值运算符将不能做连续赋值。假设有3个CMyString的对象：str1,str2和str3，在程序中语句str1=str2=str3将不能通过编译；
2）是否把传入的参数的类型声明为常量引用。如果传入的参数不是引用而是实例，那么从形参到实参会调用一次复制构造函数，把参数声明为引用可以避免这样的无谓消耗，能提高代码的效率。同时我们在赋值运算符函数内不会改变传入的实例的状态，因此应该为传入的引用参数加上const关键字；
3）是否释放实例自身已有的内存。如果我们忘记在分配新内存之前释放自身已有的空间，程序将出现内存泄露；
4）是否判断传入的参数和当前的实例（*this）是不是同一个实例。如果是同一个，则不进行赋值操作，直接返回。如果事先不判断就进行赋值，那么在释放实例自身的内存的时候就会导致严重的问题：当*this和传入的参数是同一个实例时，那么一旦释放了自身的内存，传入的参数的内存也同时被释放了，因此再也找不到需要赋值的内容了。
3.源代码：

//  Copyright (c) 2015年 skewrain. All rights reserved.
//
#include <iostream>
#include <stdio.h>
using namespace std;

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 = (int)strlen(pData);
        m_pData = new char[length+1];
        strcpy(m_pData,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()
{
    cout<<m_pData<<endl;
}

void test1()//普通赋值
{
    cout<<"test1() begins:"<<endl;
    char *text = "hello world";
    CMyString str1(text);
    CMyString str2;
    str2 = str1;
    
    cout<<"The expected result is:"<<text<<endl;
    cout<<"The actual result is:";
    str2.Print();
  
    cout<<endl;
}


void test2()//自身赋值
{
    cout<<"test2()begins:"<<endl;
    char *text = "hello world";
    CMyString str1(text);
    str1 = str1;
    
    cout<<"The expected result is:"<<text<<endl;
    cout<<"the acutal result is:";
    str1.Print();
    cout<<endl;
}


void test3()//连续赋值
{
    cout<<"test3() begins:"<<endl;
    char *text = "hello world";
    CMyString str1(text);
    CMyString str2,str3;
    str3 = str2 = str1;
    
    cout<<"The expected result is:"<<text<<endl;
    cout<<"the actual result is:";
    str2.Print();
    cout<<endl;
    
    cout<<"The expected result is:"<<text<<endl;
    cout<<"the actual result is:";
    str3.Print();
    cout<<endl;
}

int main(int argc, const char * argv[]) {
    
    test1();
    test2();
    test3();
    return 0;
}