赋值运算符函数

2016-04-14  本文已影响92人  913c9536e19a

题目:如下为类型CMyString的声明,请为该类型添加赋值运算符函数


#include "stdafx.h"#includeclass CMyString

{

public:

CMyString(char* pData = NULL);

CMyString(const CMyString& str);

~CMyString(void);

CMyString& operator = (const CMyString& str);

void Print();

private:

char* m_pData;

};

CMyString::CMyString(char *pData)

{

if(pData == NULL)

{

m_pData = new char[1];

m_pData[0] = '\0';

}

else

{

int length = strlen(pData);

m_pData = new char[length + 1];

strcpy(m_pData, pData);

}

}

CMyString::CMyString(const CMyString &str)

{

int length = strlen(str.m_pData);

m_pData = new char[length + 1];

strcpy(m_pData, str.m_pData);

}

CMyString::~CMyString()

{

delete[] m_pData;

}


该类题型一定要注意以下几点:

1.是否把返回值类型声明为该类型的引用。

只有返回一个引用,才可以连续赋值。

2.是否把传入的参数的类型声明为常量引用。

把参数类型声明为常量引用,可以避免从形参到实参所调用构造函数这一无谓的消耗。(别忘了const)

3.是否释放实例已有的内存

如果忘记,则当指针指向其他内存时,会出现内存泄漏

4.是否判断传入的参数和当前的实例是不是同一实例

如果*this和传入的参数是同一实例,则不能继续,因为如果继续的话,先释放实例已有的内存,再申请新的内存,这样会释放自身的内存,传入的参数的内存也被释放了,因此再也找不到需要赋值的内容了。


根据以上几点,写出这样的代码:

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;

}

但是,这样并不完美,因为是先释放的内存,后申请的空间,因此如果当内存不足的时候,实例自身的内存会被释放,同时又没有新的值给它,m_pData将是一个空指针,这样程序的内存容易崩溃。


因此有如下解决方法:一个简单的做法是先用new分配,再释放已有内存,这样只有在分配成功之后再释放原来的内存,更好的方法是借助一个临时变量,再交换临时实例和原来的实例。

CMyString& CMyString::operator = (const CMyString& str)

{

if (this!=&str) {

CMyString strTemp=CMyString(str);

char *pTemp=strTemp.m_pData;

strTemp.m_pData=m_pData;

m_pData=pTemp;

}

return *this;

}

如果对重点标注的代码表示不理解,那么请看下面:

m_pData为实例自身的内存,而strTemp为一个局部变量,当程序出了if之后,就会释放掉strTemp,这样原来m_pData的内存就会释放掉,否则会出现内存泄漏。

上一篇 下一篇

猜你喜欢

热点阅读