C++ string类的构造、析构、赋值函数

笔试时经常碰到的题目：

请编写STRING类的构造函数、析构函数和赋值函数

分析：

STRING类的私有对象就是一个char*，用来new出空间以存放字符串，而公有对象分别是这几个函数：

构造函数有两个，一个用char *来初始化，一个用另一个STIRNG类的对象实例来初始化

析构函数将new出的空间delete掉

赋值函数重载=操作符，也是将另一个STRING类的对象实例的字符串拷贝过来。

代码（注意注释的部分，另重载<<操作符是为了调试之用）：

#include <string.h>    //strlen()等函数需要这个C的头文件
#include <iostream>
using namespace std;

/*构造函数有两个:
a.参数为字符串,将字符串strcpy到m_data new出的空间中,另字符串为NULL时,strlen会COREDOWN,所以要判断一下
b.参数为String类的对象,将other对象的m_data strcpy到m_data中,在other是本身时,strlen会COREDOWN,所以要if判断一下
析构函数主要是将m_data空间释放,并将指针置为NULL
赋值函数以另一个String对象为参数other,将other.data拷贝到m_data,另要注意自赋值时会把自己delete掉再strcpy,会出错,所以需要if判断*/
class String
{
    
    friend ostream & operator << (ostream &os, const String &ob);    //cout时候要重载操作符<<
public:
    String(const char *str);    //注意加const
    String(const String &other);    //注意加const
    ~String();
    String & operator = (const String &other);    //注意加const
private:
    char *m_data;
};

String::String(const char *str)
{
    int length = 0;

    /*str为NULL时strlen会段错误,所以要分开写*/
    if(str==NULL)
    {
        m_data = new char[1];
        *m_data = '\0';
    }else{
        length = strlen(str);
        m_data = new char[length+1];
        strcpy(m_data, str);
    }
}

String::String(const String &other)
{
    int length = 0;
    /*为本身时strlen会引起段错误*/
    if(this!=&other)
        length = strlen(other.m_data);
    m_data = new char[length+1];
    strcpy(m_data, other.m_data);
}

String::~String()
{
    delete [] m_data;
    m_data=NULL;    /*防止野指针*/
}

String & String::operator=(const String &other)    
{
    int length = strlen(other.m_data);
    /*other为本身时会把自己的m_date delete掉再strcpy,这样是不对的*/
    if(this!=&other)
    {
        delete []m_data;
        m_data = new char[length+1];
        strcpy(m_data, other.m_data);
    }
    return *this;
}

/*另一种写法*/
/*
String & String::operator=(const String &other)
{
    int length = strlen(other.m_data);
    if(this!=&other)
    {
        String tmp(other);    //再造一个String对象,将它的指针给this的m_data,m_data给它,让它在析构的时候释放掉
        char *p=tmp.m_data;
        tmp.m_data=m_data;
        m_data=p;
    }
    return *this;
}
*/

ostream & operator << (ostream &os, const String &ob)        /*<<输出操作符定义*/
{
    os<<ob.m_data;
    return os;
}

int main()
{
    String str1="abc";
    String str2(str1);
    String str3(str3);
    char *p = NULL;
    String str4(p);
    cout<<str1<<endl;
    cout<<str2<<endl;
    cout<<str3<<endl;
    cout<<str4<<endl;
    return 0;
}