=运算符重载

=运算符重载要注意以下问题

  • 是否把返回值的类型声明为该类型的引用,并在函数结束前返回实例自身的引用。

只有返回一个引用,才可以允许连续赋值。例如: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作用域结束之后,临时实例会被自动析构掉,不会存在内存泄漏的问题。

猜你喜欢

转载自blog.csdn.net/qq_41727218/article/details/86094876