Chap 12. String 클래스의 디자인

Chap 12. String 클래스의 디자인

본 챕터에서는 cpp 표준으로 제공하는 <string> 라이브러리의 string class를 String class로 커버하는 것으로 대체한다.

#include <iostream>
#include <cstring>
using namespace std;
 
class String
{
    // 멤버변수
private:
    char * str;
    int len;
    // 멤버함수
public:
    String();
    String(const char * s); // 생성자
    String(const String & s); // 복사생성자
    ~String(); // 소멸자
    String & operator=(const String & s);     // 대입연산자 오버로딩
    String operator+(const String & s);     // +연산자 오버로딩
    String & operator+=(const String & s);     // += 연산자 오버로딩
    bool operator==(const String & s);         // ==연산자 오버로딩
    
    friend ostream & operator<<(ostream & cout, const String & s);
    friend istream & operator >> (istream & cin, String & s);
};
 
// ******************메인 함수******************
void main() {
    String str1 = "I like";
    String str2 = "String class";
    String str3 = str1 + str2;
    
    cout << str3 << endl;
 
    str1 += str2;
    cout << str1 << endl;
 
    if (str1 == str3)
        cout << "str1과 str3는 같습니다" << endl;
}
 
String::String() {
    len = 0;
    str = NULL;
}
 
String::String(const char * s) {
    len = strlen(s)+1;
    str = new char[len];
    strcpy(str, s); // 복사
}
 
String::String(const String & s) {
    len = s.len;
    str = new char[len];
    strcpy(str, s.str); // 복사
}
 
String::~String() { // 소멸자
    if (str != NULL)
        delete[]str; // 해제
}
 
String &  String::operator=(const String & s) {     // 대입연산자 오버로딩
    // 기존 string 해제 후 새 string 할당
    delete[]str; // 해제
    len = s.len;
    str = new char[len];
    strcpy(str, s.str); // 복사
 
    // 객체 참조형 반환
    return *this;
}
String & String::operator+=(const String & s) {     // +=연산자 오버로딩
    char tempstr[100];
    strcpy(tempstr, str);
 
    delete[]str;
    len += s.len; // 길이확장
    str = new char[len];
    
    strcat(tempstr, s.str); // 합친 배열 대입
    strcpy(str, tempstr);
    return *this;
}
 
String String::operator+(const String & s) {     // + 연산자 오버로딩
    char * tempstr;
    tempstr = new char[len+s.len-1];
 
    strcpy(tempstr, str); 
    strcat(tempstr, s.str);
 
    // 객체 만들어 반환
    String temp(tempstr);
    delete[] tempstr;
 
    return temp;
}
bool String::operator==(const String & s) {
    return (strcmp(str, s.str)==0)? true : false;
}
 
ostream & operator<<(ostream & cout, const String & s) {
    cout << s.str;
    return cout;
}
 
istream & operator >> (istream & cin, String & s) {
    char str[100];
    cin >> str;
 
    s = String(str);
 
    return cin;
}