北大 C++ 4.4 运算符重载实例

长度可变的数组类型

//code is tested based on Visual Studio 2015
#include <iostream>
#include <string>

using namespace std;

class CArray{
	int size;
	int *ptr;
public:
		CArray(int s = 0);  //构造函数
		CArray(CArray & a); //复制构造函数
		~CArray();//析构函数
		void push_back(int v);
		CArray & operator=(const CArray & a);   //赋值号'='重载


		int length() { return size; } //普通成员函数,是内联的函数
		int & CArray::operator[](int i)    //a[i] = 4;  赋值号重载，是内联的函数。
		{
			return ptr[i];
		}

};

CArray::CArray(int s):size(s) { //默认的参数只能在class内部进行定义。！！！所以
	//不能写成CArray::CArray(int s=0):size(s)
	if (s == 0)
		ptr = NULL;
	else
		ptr = new int[s];
}

CArray::CArray(CArray & a) {
	if (!a.ptr) {
		ptr = NULL;
		size = 0;
		return;
	}
	ptr = new int[a.size];
	memcpy(ptr, a.ptr, sizeof(int)*a.size);
	size = a.size;
}

CArray::~CArray() {
	if (ptr) delete[] ptr;
}


void CArray::push_back(int v) {
	if (ptr) {
		int *tmpPtr = new int[size + 1];
		memcpy(tmpPtr, ptr, sizeof(int)*size);
		delete[] ptr;
		ptr = tmpPtr;
	}
	else //数组本来是空的
		ptr = new int[1];
	ptr[size++] = v;
}

CArray & CArray::operator=(const CArray & a) {
	if (ptr == a.ptr)
		return *this;
	if (a.ptr == NULL) {
		if (ptr) delete[] ptr;
		ptr = NULL;
		size = 0;
		return *this;
	}
	if (size < a.size) {
		if (ptr) {
			delete[] ptr;
		}
		ptr = new int[a.size];
	}
	memcpy(ptr, a.ptr, sizeof(int)*(a.size));
	size = a.size;
	return *this;
}



int main() {
	CArray a,a2;
	for (int i = 0; i < 5; ++i) {
		a.push_back(i);
	}
	a2 = a;
	for (int i = 0; i < a.length(); ++i) {
		cout << a2[i]<< " ";
	}
	cout << endl;
	a[3] = 100;
	CArray a4(a);
	for (int i = 0; i < a4.length(); ++i) {
		cout << a4[i] << " ";
	}
	system("pause");
	return 0;
}