剑指offer_1_赋值运算符函数

/*
 * 1.返回类型声明为该类型的引用才可以连续赋值；
 * 2.传入的参数类型声明为该类型的常量引用，避免形参到实参的传递调用复制构造函数；
 * 3.在分配内存之前先释放内存避免内存泄漏；
 * 4.判断传入的参数和当前实例（*this）是否相同，如果相同则直接返回，否则释放内存操作作用于本身。
*/
#include <iostream>
#include <cstring>
#include <cstdio>
class CMyString
{
public:
    CMyString(char* pData = nullptr); //构造函数
    CMyString(const CMyString& str); //复制构造函数
    ~CMyString(void);

    CMyString& operator = (const CMyString& str); //赋值运算符函数
public:
    char* m_pData;
};

CMyString::CMyString(char *pData) {
    if(pData == nullptr){
        m_pData = new char[1];
        m_pData[0] = '\0';
    }
    else{
        m_pData = new char[strlen(pData) + 1];
        strcpy(m_pData, pData);
    }
}
CMyString::CMyString(const CMyString &str) {
    m_pData = new char[strlen(str.m_pData) + 1];
    strcpy(m_pData, str.m_pData);
}

CMyString::~CMyString() {
    delete[] m_pData;
}
//解法1:
CMyString& CMyString::operator=(const CMyString &str) {
    if(this == &str) return *this;
    delete []m_pData;
    m_pData = nullptr;
    m_pData = new char[strlen(str.m_pData) + 1];
    strcpy(m_pData, str.m_pData);
    return *this;
}
//解法2：在解法1中的new可能会出现内存不足的情况，而抛出异常。2中则不会出现。如果内存不足，原来的数据并没有被delete掉。
CMyString& CMyString::operator=(const CMyString &str){
    if(this != &str){
        CMyString strTemp(str);
        char* pTemp = strTemp.m_pData;
        strTemp.m_pData = m_pData;
        m_pData = pTemp;
    }
}

int main(){
    char* text = "Hello world";
    CMyString str1(text);
    CMyString str2;
    str2 = str1;
    printf("str2: %s", str2.m_pData);
    return 0;
}