北大 C++ 4.2 赋值运算符重载

有道云笔记有一份听课笔记

#include <iostream>
#include <string>
using namespace std;

class String {
private:
	char *str;
public:
	String() :str(NULL) { }//构造函数，初始化str为NULL
	String(String & s); //复制构造函数
	const char* c_str() { return str; }
	char* operator=(const char* s);
	String & operator=(const String & s);
	~String();
};

String::String(String & s) {
	if (s.str) {
		str = new char[strlen(s.str) + 1];
		strcpy(str,s.str);
	}
	else
		str = NULL;
}

char* String::operator=(const char *s) {
	if (str) delete[] str;
	if (s) { //s != NULL才会执行深拷贝
		str = new char[strlen(s) + 1];
		strcpy(str, s);
	}
	else
		str = NULL;
	return str;
}

String & String::operator=(const String & s) {
	if (str == s.str) return *this;
	if (str) delete[] str;
	if (s.str) { //s.str!=NULL才会执行拷贝
		str = new char[strlen(s.str) + 1];
		strcpy(str, s.str);
	}
	else {
		str = NULL;
	}
	return *this;
}

String::~String() {
	if (str) delete[] str;
}

int main() {
	String s;
	s = "Good Luck"; //s.operator=("Go");
	cout << s.c_str() << endl;
	system("pause");
	return 0;
}