C++ string类的实现

学了C++的重载机制之后，忍不住去探究一下C++中string类的内部实现过程。于是仿照这string类的功能，写了一个自己的string类练练手。可能还有些东西并未完善，后面再不断去完善。

参考代码如下所示：

Main.cpp

 
   [cpp]  
   view plaincopy
/***************************************************** 
Copyright (C)： 2017-2018   
File name    ： Main.cpp 
Author       ： Zhengqijun 
Date         ： 2017年02月10日 星期五 09时56分07秒 
Description  ：  
Funcion List ：  
*****************************************************/  
  
#include <iostream>  
#include "String.h"  
  
using namespace std;  
  
int main()  
{  
    String s1("hello");  
    s1.Display();  
  
    String s2(s1);  
    //String s2;  
    //s2.Display();  
#if 0  
    String s3;  
    s3 = s1;  
    //s3 = "hello";  
    s3.Display();  
//#endif  
  
    s1[1] = 'E';  
    s1.Display();  
#endif  
  
    String s3("world");  
#if 0  
    s3 += s1;  
    s3.Display();  
  
    cin >> s2;  
    cout << s2 << endl;  
#endif  
  
    s3 = s1 + s2;  
    s3.Display();  
  
    return 0;  
}  

String.h

 
   [cpp]  
   view plaincopy
/***************************************************** 
Copyright (C)： 2017-2018   
File name    ： String.h 
Author       ： Zhengqijun 
Date         ： 2017年02月10日 星期五 09时48分13秒 
Description  ：  
Funcion List ：  
*****************************************************/  
  
#ifndef _STRING_H_  
#define _STRING_H_  
  
#include <iostream>  
  
using namespace std;  
  
class String  
{  
public:  
    String();  
    String(char *str);  
    String(const String& other);  
    ~String();  
  
    String& operator=(char *str);  
    String& operator=(const String& other);  
  
    char& operator[](unsigned int index);  
    const char& operator[](unsigned int index) const;  
  
    friend String operator+(const String& s1, const String& s2);  
  
    String& operator+=(const String& s);  
  
    friend ostream& operator<<(ostream& out, const String& s);  
    friend istream& operator>>(istream& in, String& s);  
  
    void Display();  
  
private:  
    char *str_;  
};  
  
#endif  

String.cpp

 
   [cpp]  
   view plaincopy
/***************************************************** 
Copyright (C)： 2017-2018   
File name    ： String.cpp 
Author       ： Zhengqijun 
Date         ： 2017年02月10日 星期五 09时50分05秒 
Description  ：  
Funcion List ：  
*****************************************************/  
  
#include <iostream>  
#include <string.h>  
#include "String.h"  
  
using namespace std;  
  
String::String()  
{  
    str_ = new char['\0'];  
    cout << "default construct String!" << endl;  
}  
  
String::String(char *str)  
{  
    cout << "construct String!" << endl;  
  
    int len = strlen(str) + 1;  
    str_ = new char[len];  
    memset(str_, 0, len);  
    strcpy(str_, str)  
}  
  
String::String(const String& other)  
{  
    int len = strlen(other.str_) + 1;  
    str_ = new char[len];  
    memset(str_, 0, len);  
    strcpy(str_, other.str_);  
}  
  
String& String::operator=(const String& other)  
{  
    if(this == &other)  
    {  
        return *this;  
    }  
      
    int len = strlen(other.str_) + 1;  
    delete [] str_;  
    str_ = new char[len];  
    memset(str_, 0, len);  
    strcpy(str_, other.str_);  
  
    return *this;  
}  
  
String& String::operator=(char *str)  
{  
    delete [] str_;  
    int len = strlen(str) + 1;  
    str_ = new char[len];  
    memset(str_, 0, len);  
    strcpy(str_, str);  
  
    return *this;  
}  
  
char& String::operator[](unsigned int index)  
{  
    //return str_[index];  
    return const_cast<char&>(static_cast<const String&>(*this)[index]);  
}  
  
const char& String::operator[](unsigned int index) const  
{  
    return str_[index];  
}  
  
String operator+(const String& s1, const String& s2)  
{  
#if 0  
    int len = strlen(s1.str_) + strlen(s2.str_) + 1;  
    char *newptr = new char[len];  
    memset(newptr, 0, len);  
    strcpy(newptr, s1.str_);  
    strcat(newptr, s2.str_);  
    String tmp(newptr);  
#endif  
  
    String tmp(s1);  
  
    tmp += s2;  
  
    return tmp;  
}  
  
String& String::operator+=(const String& s)  
{  
    int len = strlen(s.str_) + strlen(str_) + 1;  
  
    char *newptr = new char[len];  
    memset(newptr, 0, len);  
    strcpy(newptr, str_);  
    strcat(newptr, s.str_);  
    String tmp(newptr);  
      
    delete [] str_;  
    str_ = new char[len];  
    strcpy(str_, newptr);  
  
    return *this;  
}  
  
ostream& operator<<(ostream& out, const String& s)  
{  
    out << s.str_;  
    return out;  
}  
  
istream& operator>>(istream& in, String& s)  
{  
    char buffer[4096];  
  
    in >> buffer;  
    s.str_ = buffer;  
  
    return in;  
}  
  
String::~String()  
{  
    cout << "destroy String!" << endl;  
}  
  
void String::Display()  
{  
    cout << "str = " << str_ << endl;  
}  

C++ string类的实现

悦读