剑指offer：赋值运算符函数

面试题1 ：赋值运算符函数

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

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

思考：

1. 应该把返回值类型声明为该类型的引用，并且在函数结束前返回实例自身的引用（*this)。因为只有返回引用才可以允许连续赋值。
2. 应该把传入参数的类型声明为常量引用。如果传入的参数不是引用而是实例，那么从形参到实参会调用一次复制构造函数。同时，我们在赋值运算符函数内不会改变传入的实例的状态，所以应该加上const关键字。
3. 应该释放实例自身已有的内存。如果我们忘记在分配新内存之前释放自身已有的空间，则程序将出现内存泄漏，因为这个原来的实例指针会变为野指针，导致内存的泄漏。
4. 判断传入的参数和当前实例是不是同一个实例。如果事先不判断就进行赋值，那么在释放实例自身内存的时候就会导致严重的问题：当*this和传入的参数是同一个实例时，一旦释放了自身的内存，传入的参数的内存也同时被释放了，因此再也找不到需要赋值的内容了。

解答：

class CMyString
{
public:
    CMyString(char* pData = nullptr);
    CMyString(const CMyString& str);
    ~CMyString(void);
  	
    CMyString & operator=(const CMyString & C);
private:
    char* m_pData;
};

CMyString::CMyString::CMyString(char *pData){
    if(pData==nullptr){
        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(void){
    delete []m_pData;
}

/*经典解法，没有考虑到异常安全性的解法*/
CMyString::CMyString & operator=(const CMyString & C){
    if(C==*this){
		return *this;
    }
    
    delete []m_pData;
    m_pData = nullptr;
    m_pData = new char[strlen(C.m_pData) + 1];
    strcpy(m_pData,C.m_pData);
    return *this;
}

/*考虑异常安全性的解法*/
CMyString::CMyString & operator=(const CMyString & C){
   if(C!=*this){
       CMyString str(C);
       char* temp = str.m_pData;
       str.m_pData = m_pData;
       m_pData = temp;
   }
    return *this;
}

考虑异常安全性的解法原因：

在前面的函数中，我们在分配内存之前先用delete释放了实例m_pData的内存。如果此时内存不足导致new char抛出异常，则m_pData将是一个空指针，这样非常容易导致程序崩溃。也就是说，一旦在赋值运算符函数内部抛出一个异常，CMyString的实例不再保持有效的状态，这就违背了异常安全性（Exception Safety）原则。

异常安全性解法分析：

当为实例本身时直接返回，当不为实例本身时，先创建一个临时实例str，然后再创建一个临时指针指向str.m_pData，接着把str.m_ pData和实例自身的m_pData进行交换,使其指向的是实例自身的内存，再让实例m_pData指向str.m_pData最开始的数据，达到赋值的目的，当退出if语句的时候，也就出了该变量的作用域，就会自动调用str的析构函数，把str.m_pData所指向的内存释放掉。由于str.m_pData指向的内存就是实例之前m_pData的内存，这就相当于自动调用析构函数释放实例的内存。

测试用例代码：

// ====================测试代码====================
void CMyString::Print()
{
    printf("%s", m_pData);
}

void Test1()
{
    printf("Test1 begins:\n");

    char* text = "Hello world";

    CMyString str1(text);
    CMyString str2;
    str2 = str1;

    printf("The expected result is: %s.\n", text);

    printf("The actual result is: ");
    str2.Print();
    printf(".\n");
}

// 赋值给自己
void Test2()
{
    printf("Test2 begins:\n");

    char* text = "Hello world";

    CMyString str1(text);
    str1 = str1;

    printf("The expected result is: %s.\n", text);

    printf("The actual result is: ");
    str1.Print();
    printf(".\n");
}

// 连续赋值
void Test3()
{
    printf("Test3 begins:\n");

    char* text = "Hello world";

    CMyString str1(text);
    CMyString str2, str3;
    str3 = str2 = str1;

    printf("The expected result is: %s.\n", text);

    printf("The actual result is: ");
    str2.Print();
    printf(".\n");

    printf("The expected result is: %s.\n", text);

    printf("The actual result is: ");
    str3.Print();
    printf(".\n");
}

int main(int argc, char* argv[])
{
    Test1();
    Test2();
    Test3();

    return 0;
}

本题考查：

• 考查应聘者对C++基础语法的理解，如运算符函数、常量引用等。
• 考查应聘者对内存泄漏的理解。
• 对于高级C++程序员，面试官还将考查应聘者对代码异常安全性的理解。