学习笔记1——赋值运算符函数

当定义一个赋值运算符函数时，应注意以下四点：

1. 为了使运算符能够进行连续赋值，应把返回值的类型声明为该类型的引用，并在函数结束前返回实例自身的引用（*this）。
2. 是否把传入的参数的类型声明为常量引用。因为如果传入的参数是实例的话，从形参到实参会调用一次复制构造函数，把参数声明为引用可以避免此种情况。
3. 为了防止内存泄漏，应在分配新内存之前释放实例自身已有的内存。
4. 事先判断传入的参数和当前的实例（*this）是不是同一个实例。

示例代码如下：

// 如下为类型CMyString的声明，请为该类型添加赋值运算符函数。

#include <cstring>
#include <cstdio>
#include <iostream>
class CMyString
{
public:
    CMyString(char* pData = nullptr);
    CMyString(const CMyString& str);
    ~CMyString(void);

    CMyString& operator = (const CMyString& str);

    void Print();
      
private:
    char* m_pData;
};

CMyString::CMyString(char *pData)
{
    if(pData == nullptr)
    {
        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 = nullptr;

    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 Test()
{
    char* text = "Hello world";

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

    cout << "The actual result is: " << endl;
    str2.Print();
    str3.Print();
}

int main(){
    Test(); 
    return 0;
}

在VS下如果使用strcpy有时会报**error C4996: ‘strcpy’: This function or variable may be unsafe. Consider using strcpy_s instead.**错误，解决方法是找到【项目属性】，点击【C++】里的【预处理器】，对【预处理器】进行编辑，在里面加入一段代码：_CRT_SECURE_NO_WARNINGS。