11、【常见算法】编写CString类

编写一个完整的CString类

//编写CString类
class CString
{
public:
    //带参构造函数
    CString(const char *pStr = NULL)
    {
        if(pStr == NULL)
        {
            pData = new char;
            *pData = '\0';
        }
        else
        {
            pData = new char[strlen(pStr)+1];
            assert(pData != NULL);
            strcpy(pData, pStr);
        }
    }

    //拷贝构造函数
    CString(const CString &other)
    {
        pData = new char[strlen(other.pData)+1];
        assert(pData != NULL);
        strcpy(pData, other.pData);
    }

    //操作符重载
    CString& operator=(const CString &other)
    {
        if(&other == this)
        {
            CString strTemp(other);

            char *pTemp = strTemp.pData;
            strTemp.pData = pData;
            pData = pTemp;
        }
        return *this;
    }

    //析构函数
    ~CString ()
    {
        if(!pData)
            delete[] pData;
    }

private:
    char *pData;
};