C++入门——实现String类

注意事项

1. 注意内存泄漏问题
2. 左移符号的重载
3. 熟练使用const关键字

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

class String
{
public:
    String(const char* str = NULL);             //通用构造函数
    String(const String &another);              //拷贝构造函数
    ~String();                                  //析构函数
    String& operator=(const String &rhs);       //赋值函数
    friend ostream& operator<<(ostream &os, String &str); //输出
private:
    char *m_data;
};

String::String(const char* str)
{
    if (str == NULL)
    {
        m_data = new char[1];
        m_data[0] = '\0';
    }
    else
    {
        m_data = new char[strlen(str) + 1];
        strcpy(m_data, str);
    }
}

String::String(const String &another)
{
    m_data = new char[strlen(another.m_data) + 1];
    strcpy(m_data, another.m_data);
}

String::~String()
{
    delete[] m_data;
}

String& String::operator=(const String &rhs)
{
    if (this == &rhs)
    {
        return *this;
    }
    delete[] m_data;
    m_data = new char[strlen(rhs.m_data) + 1];
    strcpy(m_data, rhs.m_data);
    return *this;
}

ostream& operator<<(ostream &os, String &str)
{
    os << str.m_data;
    return os;
}