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

本题考点：
C++基础语法理解，如运算符函数重载，常量引用等
对内存泄漏的理解，比如当涉及到堆区数据时，要深拷贝，并且new和delete对应
代码异常安全性的理解，比如如果选择先删除原有内存，在开辟新的内存时，如果开辟失败（内存满或其它）就会影响程序运行（指针为空指针）

#include<cstdio>
#include <cstring>
//剑指offer 面试题1 赋值运算符函数
//如下为类型CMyString的声明，为该类型添加赋值运算符函数
class CMyString
{
public:
    CMyString(char* pData = nullptr);
    CMyString(const CMyString& str);
    CMyString& operator = (const CMyString& str);
    void Print();   
    ~CMyString();

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);//strlen返回的长度不包括'\0'，所以内存需要多申请一个
        m_pData = new char[length +1];
        strcpy(m_pData, pData);
    }
    
}
//拷贝构造函数
CMyString::CMyString(const CMyString& str)
{
    int length = strlen(str.m_pData);
    this->m_pData = new char[length + 1];
    strcpy(this->m_pData, str.m_pData);
}

CMyString& CMyString::operator=(const CMyString& str)
{
    if(this != &str)
    {
        // 高级版本 使用局部实例对象自动调用析构函数销毁的策略 将以前申请的内存销毁 
        CMyString strTemp(str);//首先使用拷贝构造创建临时实例strTemp
        char* pTemp = strTemp.m_pData; //交换strTemp.m_pData/str.m_pData与this->m_pData
        strTemp.m_pData = this->m_pData;
        this->m_pData = pTemp;
        
        // //小学生版本 未考虑异常安全性 先释放后申请
        // delete[] this->m_pData;
        // this->m_pData = NULL;
        // this->m_pData = new char[strlen(str.m_pData) + 1];
        // strcpy(this->m_pData,str.m_pData);
        // //考虑了异常安全性 先申请再释放
        // char * temp = new char[strlen(str.m_pData) + 1];
        // delete[] this->m_pData;
        // this->m_pData = NULL;
        // this->m_pData = temp;
   
    }
    return *this;
}

CMyString::~CMyString()
{
    if(this->m_pData != NULL)
    {
        delete [] this->m_pData;
        this->m_pData = NULL;
    }
}

/*********************************Test***********************************/
void CMyString::Print()
{
    printf("%s", m_pData);
}


void Test1()
{
    printf("Test1 begins:\n");

    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();
    printf(".\n");
}

// 赋值给自己
void Test2()
{
    printf("Test2 begins:\n");

    char* text = "Hello world";

    CMyString str1(text);
    // str1 = str1;
    CMyString str2;
    CMyString str3(str2);
    

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

    printf("The actual result is: ");
    str1.Print();
    str3.Print();
    printf(".\n");
}

// 连续赋值
void Test3()
{
    printf("Test3 begins:\n");

    char* text = "Hello world";

    CMyString str1(text);
    CMyString str2, str3;
    str3 = str2 = str1;

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

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

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

    printf("The actual result is: ");
    str3.Print();
    printf(".\n");
}

int main()
{
    Test1();
    Test2();
    Test3();
    return 0;
}