String Class

//String.h

#ifndef STRING_H_H

#define STRING_H_H

#include <cstring>

#include <iostream>

using std::ostream;

using std::istream;

using std::cout;

using std::cin;



class String

{

    friend ostream& operator<<(ostream&, String&);

　　friend istream& operator>>(istream&, String&);

public:

    String(const char* psz=NULL);

    String(const String&);//直接初始化

    ~String();



    String& operator=(const String&);//复制初始化

    String operator+(const String&);

    int GetLength() const;

    int Compare(const String&) const;

private:

    char *data;

　　int length;

};

#endif







//String.cpp

#include "String.h"



String::String(const char *psz)

{

   length = strlen(psz);

　　data = new char[length + 1];

　　for (int i=0; i<length; i++)

　　　　data[i] = psz[i];

　　data[length] = '\0';

}



String::String(const String& rhs)

{

　　length = strlen(rhs.data);

　　data = new char[length];

　　char * tmp = rhs.data;

　　while (*data++ = *tmp++);

　　data[length] = '\0';

}



String::~String()

{    

　　delete [] data;

}



String& String::operator=(const String& rhs)

{

    if (this == &rhs)

　　　　return *this;

　　

　　delete [] data;

　　length = rhs.length;

　　data = new char[length];　

　　for (int i=0; i<length; i++)

　　　　data[i] = rhs.data[i];

　　data[length] = '\0';



　　return *this;　

}



int String::GetLength() const

{

    return length;

}



String String::operator+(const String& rhs)

{

    int new_len = length + rhs.length;

　　char * str = new char[new_len-1];



　　int count = 0;

　　for (int i=0; i<length; i++) {

　　　　str[count] = data[i];

　　　　count++;

　　}

　　for (int i=0; i<rhs.length; i++) {

　　　　str[count] = rhs.data[i];

　　　　count++;

　　}

　　str[new_len] = '\0';



　　String tmp(str);

　　*this = tmp;

　　return *this;　　　　

}



int String::Compare(const String& rhs) const

{

    return strcmp(data, rhs.data);

}



ostream& operator<<(ostream& os, String& str)

{

    os<<str.data;

    return os;

}



istream& operator>>(istream& is, String & str)

{

　　is >> str.data;

　　str.length = strlen(str.data);

　　return is;

}







//main.cpp

#include "String.h"



int main()

{

    String str1("hello ");

    String str2("world!");

    String str3 = str1 + str2;

    cout<<str3;

    return 0;

}