【C++】实现STL模板类string

#include<String>
#include <iostream>
using namespace std;
class String
{
public:
String()
{
_str = new char[1];
_str[0] = '\0';
_size = 0;
_capacity = 1;
}

String(const char* str)
{
_size = strlen(str);
_capacity = strlen(str)+1;
_str = new char[strlen(str)+1];
strcpy(_str,str);

}

~String()
{
if(_str)
{
cout<<"析构"<<endl;
delete []_str;
}
}
String(const String& str)
{
_str = new char [strlen(str._str) + 1];
strcpy(_str,str._str);
_size = str._size ;
_capacity = str._capacity ;
}
String& operator=(const String& str)
{
if(this != &str)
{
delete _str;
_str = new char[strlen(str._str) + 1];
strcpy(_str,str._str);
_size = str._size ;
_capacity = str._capacity ;
}
return *this;
}

public:
const char* GetStr()
{
return _str;
}

void _CheckCapacity(size_t capacity)
{
if(_capacity < (capacity +_capacity))
{

_capacity *= 2;
char* tmp = new char[_capacity];
strcpy(tmp,_str);
delete []_str;
_str = tmp;
}
}

void PushBack(char ch)
{
_CheckCapacity(1);
_str[_size] = ch;
_size++;
_str[_size] = '\0';
}

void PopBack()
{
if(_str)
{
_str[_size - 1] = '\0';
_size--;
}
else
{
cout<<"您要删除的节点不存在"<<endl;
}
}

void Insert(size_t pos, const char ch)
{
int tmp2 = _capacity;
int tmp = _size;
_CheckCapacity(1);
while(tmp2 != pos)
{
_str[tmp] = _str[tmp-1];
tmp--;
tmp2--;
}

_str[_size + 1] = '\0';
_size++;
_str[ pos -1] = ch;
}

void Insert(int pos, const char* str)
{

int len = strlen(str);

_CheckCapacity(_size + len + 1);

int i = _size;
int j = _size + len;
for(; i>=pos ;)
{
_str[j--] = _str[i--];
}
while(*str)
{
_str[pos++] = *str++;
}
_size += len;
_str[_size] = '\0';
}

int Find(char ch)
{
char* tmp = _str;
while(*tmp)
{
if(*tmp == ch)
{
return 1;
}
tmp++;
}
return 0;
}
int Find(const char* str)
{
const char* src = _str;
const char* sub = str;
int srclen = strlen(src);
int sublen = strlen(sub);

if(sublen > srclen)
{
return -1;
}
int srcIndex = 0;
while(srcIndex <= srclen - sublen)
{
int i = srcIndex, j = 0;
while(i < srclen  && j<sublen && src[i]==sub[j])
{
++i;
++j;
}
if(j == sublen)
{
return srcIndex;
}
++srcIndex;
}
return -1;
}

public:
// 不能用C库的函数
bool operator<(const String & s)
{
const char* str1 = _str;
const char* str2 = s._str;
while(*str1 && *str2)
{
if(*str1 < *str2)
{
return true;
}
else if(*str1 > *str2)
{
return false;
}
else
{
str1++;
str2++;
}
}
if(*str1)
{
return false;
}
if(*str2)
{
return true;
}
}
bool operator>(const String & s)
{
return !(this<s._str ||this == s._str);
}
bool operator<=(const String & s)
{
return !(this>s._str);
}
bool operator>=(const String & s)
{
return !(this<s._str);
}
bool operator==(const String & s)
{
const char* str1 = _str;
const char* str2 = s._str;
while(*str1 && *str2)
{
if(*str1++ != *str2++)
{
return false;
}
}
if(*str1 == *str2)//比较'\0'
{
return true;
}
else
return false;
}

String operator+(const String& s)
{
/*	String tmp(_str);
char* str = s._str;
while (*str)
{
tmp.PushBack(*str++);
}*/

String tmp(_str);
tmp.Insert(_size, s._str);

return tmp;
}

String operator+=(const String & s)
{
Insert(_size,s._str);
return *this;
}

friend ostream& operator<<(ostream& os, const String& s)
{
os<<s._str;
return os;
}
friend istream& operator>>(istream& is, String& s);
private:
size_t _size;		// 大小
size_t _capacity;	// 容量

char* _str;			// 指向字符串的指针
};

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

return os;
}

istream& operator>>(istream& is, String& s);

void Test1()
{
String s1;
String s2("abcd");
/*String s3("efgh");
s3 = s2;
s3.PushBack('x');
s3.PopBack();*/
/*s2.Insert(2,'x');
cout<<s2<<endl;*/
s2.Insert(2,"efg");
cout<<s2<<endl;
}

void Test2()
{
String s1("abcd");
int ret = s1.Find('e');
cout<<ret<<endl;
}

int main()
{
Test1();
//Test2();
getchar();
return 0;
}