题目:
如下为类型CMyString的声明,请为该类型添加赋值运算符函数。
class CMyString
{
public:
CMyString(char* pData = NULL);
CMyString(const CMyString& str);
~CMyString();
private:
char* m_pData;
};
想法:
1.要知道运算符重载是什么东西,
2.要考虑是否要实现连等的形式,如str1=str2=str3,
3.要考虑传入的参数和当前的实例是不是同一个实例,
4.m_pData是指针,要考虑指针的内存问题,
5.......
思路:
1.运算符重载形式有两种,重载为类的成员函数和重载为类的友元函数。
运算符重载为类的成员函数的一般语法形式为:
函数类型 operator 运算符(形参表)
{
函数体;
}
运算符重载为类的友元函数的一般语法形式为:
friend 函数类型 operator 运算符(形参表)
{
函数体;
}
3.如果传入参数和当前实例是同一个实例,则不进行操作,直接返回
4.在忘记分配新内存之前释放自身已有的空间,程序将出现内存泄露
代码:
class CMyString
{
public:
CMyString(char* pData = NULL);
CMyString(const CMyString& str);
~CMyString();
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;
}
private:
char* m_pData;
};
这里把传入的参数的类型声明为
常量引用
,是因为如果传入的参数如果不是引用而是实例,会调用一次复制构造函数,但是如果把参数声明为引用可以避免无所谓的内存消耗,提高代码效率。
另外在引用参数加上
const关键字
,是因为这个函数不会改变传入的实例的状态。
附加:
在上面的代码中,在new char这个部分,可能会因为内存不足抛出异常,m_pData将是一个空指针,这样非常容易导致程序崩溃,这里涉及到异常安全性原则,在赋值运算符函数内部抛出异常,CMyString的实例不在保持有效状态。在这个函数中实现异常安全性原则,我们有两个方法:一个就是先用new分配新内容在用delete释放已有内容,二是先创建一个临时实例,再交换临时实例和原来的实例。下面是第二种思路的参考代码:
class CMyString
{
public:
CMyString(char* pData = NULL);
CMyString(const CMyString& str);
~CMyString();
CMyString& CMyString::operator = (const CMyString& str)
{
if (this != &str)
{
CMyString strTemp(str);
char* pTemp = strTemp.m_pData;
strTemp.m_pData = m_pData;
m_pData = pTemp;
}
return *this;
}
private:
char* m_pData;
};