#include<iostream>
#include<iomanip>
using namespace std;
class String {
friend ostream& operator<< (ostream&, String&);//重载<<运算符
friend istream& operator>> (istream&, String&);//重载>>运算符
public:
String(const char* str = NULL); //赋值构造兼默认构造函数(char)
String(const String &other); //赋值构造函数(String)
String& operator=(const String& other); //operator=
String operator+(const String &other)const; //operator+
bool operator==(const String&); //operator==
char& operator[](unsigned int); //operator[]
size_t size() { return strlen(m_data); };
~String(void) { delete[] m_data; m_data = NULL; }
char * operator->()
{
return m_data;
}
private:
char *m_data; // 用于保存字符串
};
inline String::String(const char* str)
{
if (!str)m_data = 0; //声明为inline函数,则该函数在程序中被执行时是语句直接替换,而不是被调用
else {
m_data = new char[strlen(str) + 1];
strcpy_s(m_data, strlen(str) + 1, str);
}
}
inline String::String(const String &other)
{
if (!other.m_data)m_data = 0;//在类的成员函数内可以访问同种对象的私有成员(同种类则是友元关系)
else
{
m_data = new char[strlen(other.m_data) + 1];
strcpy_s(m_data, strlen(other.m_data) + 1, other.m_data);
}
}
inline String& String::operator=(const String& other)
{
if (this != &other)
{
delete[] m_data;
if (!other.m_data) m_data = 0;
else
{
m_data = new char[strlen(other.m_data) + 1];
strcpy_s(m_data, strlen(other.m_data) + 1, other.m_data);
}
}
return *this;
}
inline String String::operator+(const String &other)const
{
String newString;
if (!other.m_data)
newString = *this;
else if (!m_data)
newString = other;
else
{
newString.m_data = new char[strlen(m_data) + strlen(other.m_data) + 1];
strcpy_s(newString.m_data, strlen(m_data)+1, m_data);
strcat_s(newString.m_data, strlen(m_data) + strlen(other.m_data) + 1,other.m_data);
}
return newString;
}
inline bool String::operator==(const String &s)
{
if (strlen(s.m_data) != strlen(m_data))
return false;
return strcmp(m_data, s.m_data) ? false : true;
}
inline char& String::operator[](unsigned int e)
{
if (e >= 0 && e <= strlen(m_data))
return m_data[e];
}
ostream& operator<<(ostream& os, String& str)
{
os << str.m_data;
return os;
}
istream &operator>>(istream &input, String &s)
{
char temp[255]; //用于存储输入流
input >> temp;
s = temp; //使用赋值运算符
return input; //使用return可以支持连续使用>>运算符
}
int main()
{
string str;
String str1 = "Aha!";
String str2 = "My friend";
String str3 = str1 + str2;
cout << str3 <<endl << str3.size() << endl;
return 0;
}
有个有趣的编译错误:
error C3551: 如果使用尾随返回类型,则主要返回类型应该是单个类型说明符“自动”(而不是 "重载函数类型")
error C3613: “->”后缺少返回类型(假定为“int”)
error C2059: 语法错误:“)”
error C2334: “{”的前面有意外标记;跳过明显的函数体
这是在工作中,用sourceinsight 编程 ,做运算符重载时报的错误,当时看不出来啥毛病,网上也没搜索到,所以在这里说一下,其实是个非常低级的错误,是因为operator拼写错误。。。。比如opeartor,这就会造成以上的编译错误。。