C++提高：String 类实现

题目

设计并实现一个 string 类，并重载<<、>>、[]、==、+、=等运算符

代码

#include<iostream>
#include<iomanip>

using namespace std;

class String
{
	friend ostream& operator<< (ostream&,String&);//重载<<运算符 
	friend istream& operator>> (istream&, String&);//重载>>运算符 
public:
	String(const char* str=NULL);	// 赋值默认构造函数
	String(const String &other); //赋值构造函数(String) 
	String& operator=(const String& other); //operator= 
	String operator+(const String &other)const; //operator+ 
	bool operator==(const String&); //operator== 
	char& operator[](unsigned int); //operator[] 
	size_t size() { return strlen(m_data); };
	~String(void) { delete[] m_data; }
private:
	char *m_data; // 用于保存字符串 
};

inline String::String(const char* str)
{
	// 如果初始化的字符串为空 那么初始为0 否则将传入的字符串赋值给当前对象的m_data
	if (!str)
		m_data = 0;
	else
	{
		m_data = new char[strlen(str)+1];
		strcpy(m_data, str);
	}
}

inline String::String(const String &other)
{
	if (!other.m_data)m_data = 0;//在类的成员函数内可以访问同种对象的私有成员（同种类则是友元关系）
	else
	{
		m_data = new char[strlen(other.m_data) + 1];
		strcpy(m_data, other.m_data);
	}
}

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


inline String String::operator+(const String &other)const
{
	String newString;
	if (!other.m_data)
		newString = *this;
	else if (!m_data)
		newString = other;
	else
	{
		newString.m_data = new char[strlen(m_data) + strlen(other.m_data) + 1];
		strcpy(newString.m_data, m_data);
		strcat(newString.m_data, other.m_data);
	}
	return newString;
}

inline bool String::operator==(const String &s)
{
	if (strlen(s.m_data) != strlen(m_data))
		return false;
	return strcmp(m_data, s.m_data) ? false : true;
}

inline char& String::operator[](unsigned int e)
{
	if (e >= 0 && e <= strlen(m_data))
		return m_data[e];
}


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

istream &operator>>(istream &input, String &s)
{
	char temp[255]; //用于存储输入流 
	input >> setw(255) >> temp;
	s = temp; //使用赋值运算符 
	return input; //使用 return 可以支持连续使用>>运算符 
}

int main()
{
	String str1 = "Yang!";
	String str2 = "My friend";
	String str3 = str1 + str2;
	cout << str3 << "\n" << str3.size() << endl;
	return 0;
}