题目:如下为类型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;
}