第八週項目4擴充string類

/*01.
* 程序的版權和版本聲明部分
* Copyright (c)2013, 煙臺大學計算機學院學生
* All rightsreserved.
* 文件名稱：
* 作    者： 紀麗娜
* 完成日期： 2014 年 04月 17 日
* 版本號： v1.0
* 問題描述：
*在P324基礎上，在String類中增加一個
數據成員len表示字符串中字符個數，
然後構造String類的相關運算。這些運算可以包括：
s1 + s2用於兩個字符串的連接；
s1 - s2用於將s1的尾部空格和s2的前導空格去除後的連接；
*/
#include <iostream>
#include <Cstring>
using namespace std;
class String
{
public:
    String( );                         //默認構造函數
    String(const char *s);
    String(String &str);               //構造函數
    ~String();
    void display( );
    friend String operator + (String &s1,String &s2 );
    friend String operator - (String &s1,String &s2 );
private:
    char *p;
    int len;                                       //字符型指針，用於指向字符串
};
String::String()
{
    len=0;
    p = NULL;
}
String::String(const char *s)
{
    len = strlen(s);
    p = new char[len+1];
    strcpy(p,s);
}

String::String(String &str)
{
    len = str.len;
    if(p!=NULL) delete []p;   //當發生在賦值等情形時，原對象可能已經存在，需釋放原有空間
    p = new char[len+1];
    strcpy(p,str.p);
}

String::~String()
{
    if(!p) delete []p;//if(p==NULL)，如果p是無效的釋放p！
}

void String::display( )     //輸出p所指向的字符串
{
    cout<<p<<endl;
}

String operator+(String &s1, String &s2 )
{
    String s;
    s.len =s1.len+s2.len;
    s.p=new char[s.len+1];
    strcpy(s.p,s1.p);
    strcat(s.p,s2.p);
    return s;
}
String operator-(String &s1, String &s2 )
{
    String s;
    //c1爲截去尾部空格的字符串
    char *c1=new char[s1.len+1];
    strcpy(c1,s1.p);
    int i=s1.len-1;
    while(i>=0&&c1[i]==' ') --i;
    c1[i+1]='\0';
    //c2爲去除前導空格的字符串
    char *c2=new char[s2.len+1];
    strcpy(c2,s2.p);
    int j=0,k=1;
    while(k<s2.len+1&&c2[k]!='\0')
    {
        c2[j]=c2[k];
        ++k;
        ++j;
    }
    c2[j]='\0';

    //將這兩部分接起來
    s.len=strlen(c1)+strlen(c2);
    s.p=new char[s.len+1];
    strcpy(s.p,c1);
    strcat(s.p,c2);//連接！！！
    return s;
}

int main( )
{
    String string1(" Hello  "), string2(" World ");
    string1.display();
    string2.display();
    String string3;
    string3=string1+string2;
    string3.display();
    string3=string1-string2;
    string3.display();
    return 0;
}

心得：new  char[   ]
!!!!

if(!p) 如果p是無效的！！