自己实现简单的string类
1.前言
最近看了下《C++Primer》,觉得受益匪浅。不过纸上得来终觉浅,觉知此事须躬行。今天看了类类型,书中简单实现了String类,自己以前也学过C++,不过说来惭愧,以前都是用C来写程序,学的C++基本都忘记了,也说明自己以前对C++的理解不够深入。基于这些,觉得有必要动手来写写C++的一些程序了,毕竟C++有很多的功能是C所不具备的。正好看了课本中String类的简单实现,而且string类在C++中的使用频率也很高,了解其内部的实现是很有必要的。所以今天打算写个string类,就当做练手吧。
2.String类的设计
写类首先是定义类的名字,我们实现的string就叫String吧,以防和标准库中的string类冲突。其次是类中的内部数据,既然是字符串类,必须要把字符串类所代表的字符串保存起来,所以必须定义一个存放这些字符的字符数组或者一个指向字符数组的指针。我是用后者实现,因为这样更为的灵活,不会因为提前定义数组的大小而限制了能存放到数组中的字符个数。为了方便,可以定义字符串的长度,当然也可以不定义,可以在字符数组的末尾存放0来表示字符串的结束,不过每次得到字符串的长度比较麻烦。最后是类对外的接口,根据我们平时使用string类的情况,我们一般会用到接口size(), c_str(),还有就是+,=,>>,<<,+=,[],==等操作符。
根据前面的分析,可以得到我们要设计的String类中的成员如下所示:
1 class String 2 { 3 public: 4 String(); 5 String(const char *); 6 String(const String &); 7 String(String &&); //新加的move构造函数 8 9 ~String(); 10 11 String& operator=(const char *); 12 String& operator=(const String &); 13 14 bool operator==(const char *); 15 bool operator==(const String &); 16 17 char &operator[](int); 18 19 String operator+(const char *); 20 String operator+(const char); 21 String operator+(const String &); 22 23 String &operator +=(const char *); 24 String &operator +=(const char); 25 String &operator +=(const String &); 26 27 int size(){return _size;} 28 char *c_str(){return _string;} 29 30 friend istream &operator>>(istream &cin, String &str); 31 private: 32 int _size; 33 char *_string; 34 };
3.String类的主要实现
我们先来说一下构造函数的实现。由于我想实现的String类中存放字符的数组大小是可以根据实际需要存放字符的个数来动态调整的(这样对于存放字符的个数就没有限制,除非内存不够用了),所以必须根据实际存放的字符个数来动态的申请内存空间,代码可以用如下的方式实现:
1 String::String(const char *str) 2 { 3 if (!str) 4 { 5 _size = 0; 6 _string = NULL; 7 } 8 else 9 { 10 _size = strlen(str); 11 _string = new char[_size + 1]; 12 strcpy(_string, str); 13 _string[_size] = 0; 14 } 15 }
由于构造函数动态申请了内存,所以必须定义析构函数来释放我们申请的内存空间。
1 String::~String() 2 { 3 if (_string) 4 delete _string; 5 6 cout << "~String() call" << endl; 7 }
同样,由于我们每个String类都会动态的申请内存空间,所以必须定义拷贝构造函数,否则默认的拷贝构造函数会导致多个String对象共享相同内存空间的问题(深拷贝与浅拷贝的的问题)。代码和前面的构造函数差不多。
1 String::String(const String &str) 2 { 3 if (!str._size) 4 { 5 _size = 0; 6 _string = NULL; 7 } 8 else 9 { 10 _size = str._size; 11 _string = new char[_size + 1]; 12 strcpy(_string, str._string); 13 _string[_size] = 0; 14 } 15 16 }
其它的函数就是对+,=,+=操作符的重载,原理都是一样的,需要重新分配内存空间来适应新的字符串个数的需求,不同的是+的返回类型不需要使用引用。具体代码如下(用+=实现的):
String String::operator+(const String &str) { assert(_string && str._string); String str_temp(*this); str_temp += str; return std::move(str_temp); }
由于函数+会返回对象,编译器会默认产生一个临时的对象,这个对象完全是返回对象的拷贝,而返回对象马上就会析构掉,所以如果我们能把返回对象中的字符数组地址拷贝到临时对象中,那么临时对象就可以不用再申请内存了,这样可以提供效率。所以这里我实现了一个move函数( tangzhnju的提醒):
String::String(String && str) { _size = str._size; _string = str._string; str._string = NULL; }
还有我们经常用的是直接对string类进行输出,所以我们必须重载<<,>>操作符,由于输入需要用到String类中的私有成员变量_string,所以应该把输入函数设置为String类的友元函数。而输出可以通过调用c_str函数获得字符数组地址来输出。
istream &operator>>(istream &cin, String &str) { const int limit_string_size = 4096; str._string = new char[limit_string_size]; cin >> setw(limit_string_size) >> str._string; str._size = strlen(str._string); return cin; } ostream &operator<<(ostream &cout, String &str) { return cout << str.c_str(); }
编写这个String类遇到了两个小的问题,一个就是在写+=操作符重载函数的时候,忘记在函数的前面写String::,导致编译器总是包各种莫名的错误,什么参数不对,不能访问内部成员,这个问题完全是自己粗心导致的。另一个问题就是写+操作符重载函数,开始写+函数的时候直接修改了当前的String对象,后来测试发现这是有问题的,因为+应该重新返回一个新的String对象,这个新String对象是当前String对象与传入参数的字符串的和,这个问题完全是自己没有想清楚+与+=的区别,+=才是要修改当前String对象的。