String簡單實現

String類的簡單實現：

一、在拷貝構造函數和賦值運算符重載這兩個函數中有淺拷貝和深拷貝的問題

二、要對輸入輸出函數重載

三、賦值運算符重載有兩種方式。注意在採用交換實現的時候函數的參數只能是臨時變量

class String
{
	friend ostream& operator<<(ostream& os, String& s);
	friend istream& operator>>(istream& is, String& s);
public:
	String(char *str="")
		:_sz(strlen(str))
		, _capacity(strlen(str)+1)
		,_str(new char [strlen(str)+1])
	{
		cout << "String()" << endl;
		strcpy(_str,str);
	}

	String(const String& s)
		:_sz(strlen(s._str))
		, _capacity(strlen(s._str) + 1)
		,_str(new char [strlen(s._str)+1])
	{
		cout << "String(const String&)" << endl;
		strcpy(_str,s._str);
	}

	String& operator=(String s)
	{
		_sz = s._sz;
		_capacity = s._capacity;
		swap(_str,s._str);
		return *this;
	}

	//String& operator=(const String& s)
	//{
	//	
	//	if (_str != s._str)
	//	{
	//		cout << "String& operator=" << endl;
	//		delete[] _str;
	//		_str = new char[strlen(s._str) + 1];
	//		strcpy(_str, s._str);
	//	}
	//	return *this;
	//}

	~String()
	{
		cout << "~String()" << endl;
		delete[] _str;
		_str = NULL;
		_sz = 0;
		_capacity = 0;
	}
public:
	char* C_str()const
	{
		return _str;
	}

	void PushBack(char c)                            //尾插一個字符
	{
		GetCapacity(1);
		_str[_sz++] = c;
		_str[_sz] = '\0';
	}

	String& Insert(int posl, const String& s)          //在指定位置插入一個字符串
	{
		int count = strlen(s._str);
		if (posl<=_sz)                              //若是插入的位置在字符串結束以後，則不插入
		{
			GetCapacity(count);
			int j = _sz + count;
			for (int i = _sz; i >= posl; i--, j--)
			{
				_str[j] = _str[i];
			}
			strncpy(_str + posl, s._str, count);
		}
		_sz += count;
		return *this;
	}

	char& operator[](size_t posl)
	{
		return _str[posl];
	}
private:
	void GetCapacity(int count)                    //獲得擴容後的空間
	{
		if ((_sz+count) >= _capacity)               //若是總的字符數大於或等於容量，則擴容
		{
			int NewCapacity = (2 * _capacity) > (_capacity + count) ?
				(2 * _capacity) : (_capacity + count);
			char *tmp = new char[NewCapacity];
			strcpy(tmp, _str);
			delete[] _str;
			_str = tmp;
			_capacity = NewCapacity;
		}
	}
private:
	char *_str;
	int _sz;                             //標記字符的個數
	int _capacity;                       //標記容量
};

ostream& operator<<(ostream& os, String& s)
{
	os<< s._str;
	return os;
}

istream& operator>>(istream& is, String& s)
{
	is >> s._str;
	return is;
}