string的模拟实现

#pragma warning(disable:4996)
#include<iostream>
using namespace std;
#include<assert.h>
#include<string.h>

class String
{
public:
	typedef char* iterator;
public:
	String(const char* str = "") //构造函数
	{
		_size = strlen(str);
		_capacity = _size;
		_str = new char[_capacity + 1];
		strcpy(_str, str);
	}
	String(const String& str)  //拷贝构造函数
		:_str(nullptr)
		, _capacity(0)
		, _size(0)
	{
		String tmp(str._str);
		this->Swap(tmp);
	}
	//string(const string& s)
	//	: _size(s._size)
	//	, _capacity(s._size)
	//{
	//	_str = new char[_capacity + 1];
	//	strcpy(_str, s._str);
	//}
	~String()  //析构函数
	{
		if (_str)
		{
			delete[] _str;
			_str = nullptr;
		}
	}
	String& operator=(String str)  //赋值运算符重载 
	{
		this->Swap(str);
		return *this;
	}
	iterator begin()  //迭代器
	{
		return _str;
	}
	iterator end()  
	{
		return _str + _size;
	}


//元素修改方法
	void PushBack(char c)  //尾插一个字符c
	{
		if (_size == _capacity)
		{
			Reserve(_capacity * 2);
		}
		_str[_size++] = c;
		_str[_size] = '\0';
	}
	String& operator+=(char c)  //尾插一个字符
	{
		PushBack(c);
		return *this;
	}
	void Append(const char* str)  //尾接一个字符串
	{
		if ((_size + strlen(str)) >= _capacity)
		{
			Reserve(_capacity + strlen(str));
		}
		strcat(_str, str);
		_size += strlen(str);
		_str[_size] = '\0';
	}
	String& operator+=(const char* str)  //尾接一个字符串
	{
		Append(str);
		return *this;
	}
	void Clear()  //清空
	{
		_size = 0;
		_str[_size] = '\0';
	}
	void Swap(String& str)  //交换对象中的资源
	{
		swap(_str, str._str);
		swap(_capacity, str._capacity);
		swap(_size, str._size);
	}
	const char* C_Str()const  //用C风格字符串表示
	{
		return _str;
	}
	String& Reserve(size_t newCapacity)  //扩容
	{
		if (newCapacity > _capacity)
		{
			char* str = new char[newCapacity + 1];
			strcpy(str, _str);

			delete[] _str;
			_str = str;
			_capacity = newCapacity;
		}
		return *this;
	}
	String& Resize(size_t newSize, char c = '\0')  //改变大小，多的空间位用字符c填充
	{
		if (newSize > _size)
		{
			if (newSize > _capacity)
			{
				Reserve(newSize);
			}
			memset(_str + _size, c, newSize - _size);
		}
		_size = newSize;
		_str[newSize] = '\0';
		return *this;
	}

//下标访问运算符重载的方法通常成对出现，一般方法和const方法
	char& operator[](size_t index)  //下标访问运算符重载
	{
		assert(index < _size);
		return _str[_size];
	}
	const char& operator[](size_t index)const
	{
		assert(index < _size);
		return _str[_size];
	}
	
/
//判断运算符重载
	bool operator<(const String& s)
	{
		if (strcmp(_str, s._str) < 0)
		{
			return true;
		}
		return false;
	}
	bool operator<=(const String& s)
	{
		if (strcmp(_str, s._str) <= 0)
		{
			return true;
		}
		return false;
	}
	bool operator>(const String& s)
	{
		if (strcmp(_str, s._str) > 0)
		{
			return true;
		}
		return false;
	}
	bool operator>=(const String& s)
	{
		if (strcmp(_str, s._str) >= 0)
		{
			return true;
		}
		return false;
	}
	bool operator==(const String& s)
	{
		if (strcmp(_str, s._str) == 0)
		{
			return true;
		}
		return false;
	}
	bool operator!=(const String& s)
	{
		if (!(*this == s))
		{
			return true;
		}
		return false;
	}
///
//元素修改方法
	size_t Find(char c, size_t pos = 0)const //返回c在string中第一次出现的位置
	{
		for (size_t i = pos; i < _size; ++i)
		{
			if (_str[i] == c)
			{
				return i;
			}
		}
		return -1;
	}
	size_t Find(const char* s, size_t pos = 0)const //返回子串s在string中第一次出现的位置
	{
		/*
		assert(s != nullptr);
		char* str = strstr(_str, s);
		if (str != nullptr)
		{
			return str - s;
		}
		*/

		size_t len = strlen(s);
		int flag = 1;
		for (size_t i = pos; i < _size; ++i)
		{
			for (size_t j = 0; j < len; ++j)
			{
				if (_str[i + j] != s[j])  //子串遍历中，若有一位的元素未能匹配成功，本次匹配失败，跳出，为下一次匹配做准备
				{
					flag = 0;
					break;
				}
			}
			if (flag == 1)  //经过遍历子串s，子串每一个元素都与目标串相同，直接返回即可
			{
				return i;
			}
			flag = 1;  //本次未能匹配成功，下一次匹配前必须将标志位置为1
		}
		return -1;
	}
	String& Insert(size_t pos, char c) //在pos位插入字符c
	{
		if (_size == _capacity)
		{
			Reserve(_capacity * 2);
		}
		for (size_t i = _size; i > pos; --i) //从结束符'\0'位置开始后移一个位置
		{
			_str[i] = _str[i - 1];
		}
		_str[pos] = c;
		_str[++_size] = '\0';
		return *this;
	}
	String& Insert(size_t pos, const char* str) //在pos位插入字符串str
	{
		size_t len = strlen(str);
		if (_size + len > _capacity) //看是否需要扩容
		{
			Reserve(_size + len);
			_size = _size + len;  //更新_size值
		}
		for (size_t i = _size; i >= pos + len; --i) //从结束符'\0'位置(_str[_size]位置)开始后移len位置
		{
			_str[i] = _str[i - len];
		}
		for (size_t i = pos, j = 0; j < len; ++i, ++j)
		{
			_str[i] = str[j];
		}
		return *this;
	}
	String& Erase(size_t pos, size_t len) //删除pos位置的元素
	{
		for (size_t i = pos; i < _size; ++i)
		{
			_str[i] = _str[i + 1];
		}
		_size--;
		return *this;
	}

private:
	char* _str;  //字符串
	size_t _capacity;  //容量
	size_t _size;  //字符串大小
};

//测试
void TestString1()
{
	String s1;
	String s2("helloWorld");
	String s3(s2);
	String s4("school");
	s1 = s4;
	s2.PushBack('!');
	s3 += '!';
	s3 += " great!";
	s4.Append(" great");
	s1.Clear();
}
void TestString2()
{
	String s1("abcdefabdmn");
	size_t res = s1.Find('a');
	res = s1.Find('a', 3);
	res = s1.Find("abd");
	res = s1.Find("mn");
}
void TestString3()
{
	String s1("abcdefabdmn");
	s1.Insert(1, 'a');
	s1.Insert(3, 'a');
	s1.Insert(1, "abcd");
	s1.Insert(3, "xzy");
}
int main()
{
	//TestString1();
	//TestString2();
	TestString3();
	return 0;
}