自己实现的stirng类,注意自我复制的异常安全性,采用copy and swap技术。
#include <iostream>
#include <cstring>
using namespace std;
class CString{
public:
CString(const char* str=NULL);
~CString();
CString(const CString& str);
void swap(CString& rhs);
CString& operator= (const CString& str);
int length() const { return len; }
char* c_str() { return m_data; }
private:
char* m_data;
int len;
};
CString::CString(const char* str)
{
if(NULL == str)
{
len = 0;
m_data = new char[1];
*m_data = '\0';
}
else
{
len = strlen(str);
m_data = new char[len + 1];
strcpy(m_data, str);
}
}
CString::~CString()
{
if(NULL != m_data)
{
delete []m_data;
m_data = NULL;
}
}
CString::CString(const CString& other)
{
len =strlen(other.m_data);
m_data = new char[len+1];
strcpy(m_data, other.m_data);
}
void CString::swap(CString& rhs)
{
std::swap(m_data, rhs.m_data);
std::swap(len, rhs.len);
}
CString& CString::operator=(const CString& other)
{
//cout << "operator =:" << len << endl;
CString temp(other);
swap(temp);
return *this;
}
int main()
{
CString str1("hello");
CString str2("world!");
cout<< str2.c_str() << " " << str2.length() << endl;
CString str3(str2);
cout<< str3.c_str() << " " << str3.length() << endl;
str2 = str1;
str3.swap(str1);
cout<< str1.c_str() << " " << str1.length() << endl;
cout<< str2.c_str() << " " << str2.length() << endl;
cout<< str3.c_str() << " " << str3.length() << endl;
return 0;
}