#include <iostream>
using namespace std;
class CArray {
int size; //数组元素的个数
int *ptr; //指向动态分配的数组
public:
CArray(int s = 0); //s代表数组元素的个数
CArray(CArray&a);
~CArray();
void push_back(int v); //用于在数组尾部添加一个元素v
CArray&operator=(const CArray&a);
//用于数组对象间的赋值
int length() { return size; } //返回数组元素个数
int& CArray::operator[](int i) {
//用以支持根据下标访问数组元素,如n=a[i]和a[i]=4;
return ptr[i]; //非引用的返回值不可以作为左值使用,所以这里用 int& 引用
}
};
CArray::CArray(int s) :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;
}
CArray&CArray::operator=(const CArray&a) {
if (ptr == a.ptr) //防止a=a这样的赋值导致错误
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;
}
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;
}
int main() { //要编写可变长整型数组类,使之能如下使用:
CArray a; //开始的数组是空的
for (int i = 0; i < 5; ++i)
a.push_back(i); //要用动态分配的内存来存放数组元素,需要一个指针成员变量
CArray a2, a3;
a2 = a; //重载运算符“="
for (int i = 0; i < a.length(); ++i)
cout << a2[i] << " ";
a2 = a3; //a2是空的
for (int i = 0; i < a2.length(); ++i)//a2.length()返回()
cout << a2[i] << " "; //重载运算符“[]”
cout << endl;
a[3] = 100;
CArray a4(a);
for (int i = 0; i < a4.length(); i++)
cout << a4[i] << " ";
return 0;
}
No comments:
Post a Comment