C++ 友元 常成员函数 运算符重载

 

在昨天作业的基础上，将能进行重载的运算符全部重载

 

#include <iostream>
#include <cstring>

using namespace std;

class myString
{
private:
    char *str;          //记录c风格的字符串
    int size;            //记录字符串的实际长度
public:
    //无参构造
    myString():size(10)
    {
        str = new char[size];         //构造出一个长度为10的字符串
        strcpy(str,"");
    }
    //有参构造
    myString(const char *s)
    {
        size = strlen(s);
        str = new char[size+1];
        strcpy(str, s);
    }

    //拷贝构造
    myString(myString &other):str(new char(*(other.str)))
    {
        strcpy(str,other.str);
    }

    //析构函数
    ~myString()
    {
        delete str;
        cout<<"析构成功"<<endl;
    }
    //判空函数
    bool empty()
    {
        size = strlen(str);
        if(size == 0)
        {
            return 0;
        }else
        {
            return 1;
        }

    }
    //size函数
    int len()
    {
        return size;
    }
    //c_str函数
    char *c_str()
    {
        return str;
    }
    //at函数
    char &at(int pos)
    {
        size = strlen(str);
        if(pos >= size || pos < 0)
            cout<<"越界访问"<<endl;
        else
            return str[pos];
    }

    //赋值运算符重载
    myString & operator=(const myString &s)
    {
        strcpy(this->str,s.str);
        return *this;
    }

    //重载中括号运算符
    char &operator[](int index)
    {
        if(index < 0 && index >= size)
        {
            cout<<"越界"<<endl;
        }
        else
        {
            return str[index];
        }
    }

    //重载加号
    myString &operator+(const myString &s)
    {
        strcat(this->str,s.str);
        return *this;
    }

    //重载==
    bool operator==(const myString &s)const
    {
        if(0 == strcmp(this->str,s.str))
            return 1;
        else
            return 0;
    }

    //重载>
    bool operator>(const myString &s)const
    {
        if(0 < strcmp(this->str,s.str))
            return 1;
        else if(0 > strcmp(this->str,s.str))
            return 0;
    }

    //重载<
    bool operator<(const myString &s)const
    {
        if(0 > strcmp(this->str,s.str))
            return 1;
        else if(0 < strcmp(this->str,s.str))
            return 0;
    }

    //重载<<设置为友元
    friend ostream &operator<<(ostream&out,const myString &s);
    //>>重载设置为友元
    friend istream &operator>>(istream &in,const myString &s);
};

//全局实现<<重载
ostream &operator<<(ostream&out,const myString &s)
{
    out<<s.str;
    return out;
}

//全局实现>>重载
istream &operator>>(istream &in,const myString &s)
{
    in>>s.str;
    return in;
}

int main()
{
    //有参构造
    myString s1("zxcvb");
    cout<<s1.c_str()<<endl;

    //拷贝构造
    myString s2(s1);
    cout<<s2.c_str()<<endl;

    //判空
    myString s3;
    cout<<s3.empty()<<endl;

    //size
    cout<<s2.len()<<endl;

    //at
    cout<<s2.at(3)<<endl;

    //赋值运算符重载
    myString s4;
    s4 = s1;
    cout<<s4.c_str()<<endl;

    //中括号重载
    cout<<s4[2]<<endl;

    //加号重载
    myString s5;
    s5 = s4 + s1;
    cout<<s5.c_str()<<endl;

    //==重载
    if(s5 == s1)
    {
        cout<<"yes"<<endl;
    }
    else
    {
        cout<<"no"<<endl;
    }

    //>重载
    if((s5 > s1) > 0)
    {
        cout<<"yes"<<endl;
    }
    else
    {
        cout<<"no"<<endl;
    }

    //<重载
    if((s5 < s1) > 0)
    {
        cout<<"yes"<<endl;
    }
    else
    {
        cout<<"no"<<endl;
    }

    //重载<<
    cout<<s1<<endl;

    //重载>>
    myString s6;
    cin>>s6;
    cout<<"s6 = "<<s6<<endl;
    return 0;
}