设计 operator = ,需要满足连续赋值以及自我赋值
类的设计者有责任设计operator = 运算符函数,因为只有类的设计者才知道对于这个操作背后的具体动作。
直接上代码:
#include<cstring> #include<cstdio> class CMYString { private: /* data */ char* m_pData; public: CMYString(char* m_pData = nullptr); CMYString(const CMYString& str); ~CMYString(); CMYString& operator = (const CMYString& str); void Print(); }; 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]; strcpy(m_pData,pData); } } CMYString::~CMYString() { delete[] m_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::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; }
测试用例设计:
1.连续赋值 str3 = str2 = str1
2.自我赋值 str1 = str1
3.正常赋值 str2 = str1