题目:实现一个string类,为该类型添加赋值运算符函数。
关注点:
- 返回值的类型应声明为该类型的引用,这样才可连续赋值
- 传入参数的类型应声明为常量引用,提高灵活性
- 释放实例自身已有内存
- 判断传入参数是否为本身
mystring & mystring::operator=(const mystring &o)
{
// 判断是否为本身
if (this == &o)
return *this;
// 释放原有内存
delete [] m_data;
int length = strlen(o.m_data);
// 申请新内存
m_data = new char[length + 1];
if (m_data != nullptr)
{
// 复制数据并返回引用
strcpy_s(m_data, (length + 1), o.m_data);
return *this;
}
}
参见姊妹问题:编程实现一个string类
《剑指offer》书上说:写出上面代码只是应届毕业生或者初级程序员的水平。还有更高的要求。
潜在的问题:分配内存前首先delete当前实例的内存,如果内存不足导致new抛出异常,则m_data将是一个空指针,很容易导致程序崩溃。一旦在赋值运算符函数内部抛出一个异常,则该实例也丢了,违背了异常安全Exception Safety的原则。
两个解决思路:
- 先用new分配新内容,成功后再释放。当分配失败时能确保原对象在。
- 更好思路:先创建一个临时对象,交换临时对象和原对象数据指针。退出时临时对象销毁,因其指向原对象数据,则可保证销毁。
mystring & mystring::operator=(const mystring &o)
{
if(this != &o)
{
mystring strtmp(o);
// 二者交换
char * ptmp = strtmp.m_data;
strtmp.m_data = m_data;
m_data = ptmp;
}
return *this;
// 返回后临时变量销毁,原对象对应的内存回收
// 原对象指向新的内存,起到赋值的作用
}
strtmp.m_data指向的内存就是对象之前的内存,返回时调用strtmp的析构函数释放老内存。该对象指向的新内存则由该对象接管。
上述代码中把申请新空间放在构造函数中用new分配。若内存不足抛出异常,则原对象不受影响,保证了“异常安全”。