天天看点

【C++项目实战】赋值操作符重载

1 知识点

  • 构建类
  • 传递字符数组
  • 使用cstring中的函数strlen和strcpy
  • 赋值操作符重载

2 代码

#include <iostream>
#include <cstring>
using namespace std;
class CMyString{
public:
    CMyString(char* pData = nullptr): m_pData(pData){} // 构造函数,列表初始化
    CMyString(const CMyString& str): m_pData(str.m_pData){}// 重载构造函数,列表初始化
    CMyString& operator=(const CMyString &str);// 赋值操作符重载声明
    void getData(){ // 字符数组打印内容
        for(int i=0; *(m_pData+i) != 0; ++i){
            cout << *(m_pData+i);
        }
        cout << endl;
    }
    ~CMyString(){};
private:
    char* m_pData;
};

inline CMyString& CMyString::operator=(const CMyString &str){ // 使用引用返回量
    if (this == &str){ // 判断是否自身,如果是则直接返回*this
        return *this;
    }
    m_pData = new char[strlen(str.m_pData)+1]; // 根据右值创建空间
    strcpy(m_pData, str.m_pData); // 拷贝
    return *this;
}

int main(){
    CMyString s1 = (char*)"Hello";
    CMyString s2 = (char*)"World";
    s2 = s1; 
    s2.getData();
    return 0;
}
           

3 输出

Hello