面试算法（一）赋值运算符函数

1、题目：如下为类型CMyString的声明，请为该类型添加赋值运算符函数。

class CMyString
{
public:
CMyString( char* pData = NULL);
CMyString( const CMyString& str);
~CMyString( );
private:
char *m_pData;
};

当面试官要求定义一个赋值运算符函数时，会考查如下几点：

1）是否把返回值的类型声明为该类型的引用，并且在函数结束前返回实例自身的引用（即*this）。只有返回一个引用，才可以允许连续赋值。否则如果函数的返回值是void，应用该赋值运算符将不能做连续赋值，比如str1=str2=str3不能通过编译。

2）是否把传入的参数的类型声明为常量引用。如果传入的参数不是引用而是实例，那么从形参到实参会调用一次复制构造函数。把参数声明为引用可以避免这样的无谓消耗，能提高代码的效率。同时，我们在赋值运算符函数内不会改变传入的实例的状态，因此应该为传入的引用参数加上const。

3）是否释放实例自身已有的内存。如果我们忘记在分配新内存之前释放自身已有的空间，将出现内存泄露。

4）是否判断传入的参数和当前的实例（*this）是不是同一个实例。如果是同一个，则不进行赋值操作，直接返回。如果事先不判断就进行赋值，那么在释放实例自身的内存时就会出错：当*this和传入的参数是同一个实例时，那么一旦释放了自身的内存，传入的参数的内存也同时被释放了，因此再也找不到需要赋值的内容了。

2、基于上面4点考虑，写出基本程序：

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;
}

3、考虑异常安全性的程序：

在上面函数中，分配内存之前先用delete释放了实例m_pData的内存，如果此时内存不足导致new char抛出异常，m_pData将是一个空指针，这样容易导致程序崩溃。

此时有两种方法可以避免，一是先用new分配新内容再用delete释放已有的内容。二是先创建一个临时实例，再交换临时实例和原来的实例。如下代码：

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;
}