=运算符重载要注意以下问题
- 是否把返回值的类型声明为该类型的引用,并在函数结束前返回实例自身的引用。
只有返回一个引用,才可以允许连续赋值。例如:str1 = str2 = str3;
- 是否把传入参数的类型设为常量引用。
这样做有两个目的。(1)提高效率,因为以实例作为参数将会调用一次拷贝构造(2)不改变传入实例的状态
- 是否释放实例自身已有的内存,防止内存泄漏
- 判断传入的参数和当前的实例不是同一个实例。防止自赋值。
基于以上几点,可以写出如下代码:
class A
{
private:
char *a;
public:
A(char* ptr)
{
a = ptr;
}
A& operator=(const A& tmp)
{
//防止自赋值
if (this == &tmp)
{
return *this;
}
//防止内存泄漏
delete a;
a = NULL;
a = new char[strlen(tmp.a) + 1];
strcpy(a, tmp.a);
return *this;
}
};
以上写法还有一个安全性的问题没有考虑进去。该代码实在将原有的内存释放掉之后才进行new,如果此时内存不够,new抛出异常,原来的实例中的a指针就会指向NULL。导致没有赋值成功但是还改变了原有的值。对于这个问题有两种解决办法:
1、先new出新空间,再释放原有的空间。
2、先创建一个实例,再交换临时实例和原来的实例。如下:
A& operator=(const A& tmp)
{
if (this != &tmp)
{
A p(tmp);
char *ptr = p.a;
p.a = a;
a = ptr;
}
return *this;
}
//该代码先创建一个临时实例存储形参的值,再将临时实例和当前实例交换,当if作用域结束之后,临时实例会被自动析构掉,不会存在内存泄漏的问题。