题目:如下为类型CMyString的声明,请为该类型添加赋值运算符函数。
class CMyString { public: CMyString(char* pData = NULL); CMyString(const CMyString & str); ~CMyString(void); private: char* m_pData; };
分析:
赋值运算符要注意以下四点:
1.把返回值的类型声明为该类型的引用。
2.把传入的参数的类型声明为常量引用。
3.释放实例自身已有的内存。
4.判断传入的参数和当前的实例是否是同一个实例。
根据以上几点要求,我们可以写出如下赋值运算符。
CMyString & CMyString::operator = (const CMyString & str) { if(this == &str) return *this; delete [] m_pData; m_pData = NULL; m_pData = new char[strlen(str,m_pData) + 1]; strcpy(m_pData, str.m_pData); return * this; }
对于初级C++程序员能写出上述函数就OK了,但是如果要作为一个高级C++程序员的话,则需要考虑更多的安全性问题。
例如上面代码中的new部分要是内存不足则会抛出异常,此时实例的m_pData已经被删除,而新的内存又没申请到,就会导致数据丢失,所以我们可以采用下面的代码来保证代码的安全性。
CMyString & CMyString::operator = (const CMyString & str) { if(this != & str) { CMyString tmp(str); char* pTmp = tmp.m_pData; tmp.m_pData = m_pData; m_pData = pTmp; } return *this; }