剑指offer面试题之赋值运算符函数

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

class CMyString
{
  public:
     CMyString(char* pData=NULL);

     CMyString(const CMyString& str);

     ~CMyString();

  private:
     char* m_pData;
};

此时应该注意如下几点：

1.返回值类型应该声明为该类型的引用，并在函数结束前返回实例自身的引用(即*this)。只有返回引用，才可以允许连续赋值

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

3. 释放自身已有的内存。如果我们忘记在分配新内存之前释放自身已有的内存，程序将出现内存泄漏。

4. 要判断传入的参数和当前的实例(*this)是不是同一个实例。如果是同一个，则不进行赋值操作，直接返回。如果事先不判断就直接进行赋值，那么在释放实例自身的内存时就会导致严重的问题：当*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;

}

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

要想在赋值运算符中实现异常安全性，我们有两种方法。第一种简单的方法就是我们先用new分配新内容再用delete释放已有的内容。也就是当内存分配失败时我们能确保CMyString的实例不会被修改。

二、解法2：考虑异常安全性，分配内存成功后再释放原来的内容

CMyString& CMyString::operator=(const CMyString &str)
{
    if(this==&str)
    {
       return *this;
    }
    char *temp=new char[strlen(str.m_pData)+1];
    delete []m_pData;
    m_pData=NULL;
    m_pData=temp;
    strcpy(m_pData,str.m_pData);

    return *this;    
}

我们还有一个更好的办法是先创建一个临时实例，在交换临时实例和原来的实例

三、解法3：先创建一个临时实例，然后再交换临时实例和原来的实例

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

在这个函数中，我们先创建一个临时实例strTemp，接着把strTemp.m_pData和实例自身的m_pData做交换。由于strTemp是一个局部变量，但程序运行到if的外面时也就出了该变量的作用域，就会自动调用strTemp的析构函数，把strTemp.m_pData所指向的内存释放掉。由于strTemp.m_pData指向的内存就是实例之前m_pData的内存，这就相当于自动调用析构函数释放实例的内存。

下面是完整的代码以及测试用例：

#include<iostream>
#include<string>
using namespace std;

class CMyString{
public:
      CMyString(char* pData=NULL);
      CMyString(const CMyString &str);
      ~CMyString();
      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;
}

//方法一：经典解法，未考虑内存不足导致new char 抛出异常
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;
}


//方法二：考虑异常安全性，分配内容成功后再释放原来的内容
/*
CMyString& CMyString::operator=(const CMyString &str)
{
   if(this==&str)
   {
       return *this;
   }
   char *pTemp=new char[strlen(str.m_pData)+1];
   delete[] m_pData;
   m_pData=NULL;
   m_pData=pTemp;
   strcpy(m_pData,str.m_pData);
   return *this;
}
*/

//方法三：先创建一个临时实例，再交换临时实例和原来的实例
/*
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;
}
*/

//测试代码

void CMyString::Print()
{
    printf("%s\n",m_pData);
}

void func1()
{
   printf(" test fun1 :\n");

   char* text="My name is CSDN!";
   CMyString str1(text);
   CMyString str2;
   str2=str1;

   printf("The original result is:%s\n",text);
   printf("The actual result is: ");
   str2.Print();
}

//赋值给自己
void fun2()
{
   printf("test func2 :\n");
   char* text="My name is CSDN!";
   CMyString str1(text);
   str1=str1;
   printf("The originalresult is:%s\n",text);
   printf("The actual result is:");
   str1.Print();

}

//连续赋值

void fun3()
{
   printf("test fun3 :\n");
   char* text="My name is CSDN!";
   CMyString str1(text);
   CMyString str2,str3;
   str3=str2=str1;
   printf("The original result is:%s\n",text);
   printf("The actual result is:\n");
   str2.Print();
   str3.Print();
}

int main()
{
   fun1();

   fun2();

   fun3();

   system("pause");

   return 0;
}

程序运行结果：