在复习C++模板中,整理了一个数组类模板,注释都在代码上。
#include <iostream>
#include <cassert>
using namespace std;
//数组类模板定义
template <class T>
class Array
{
private:
T *list; //T类型指针,用于存放动态内存分配的数组内存首地址
int size; //数组大小(数组个数)
public:
Array(int sz = 50); //构造函数
Array(const Array<T>&a); //复制构造函数
~Array(); //析构函数
Array<T>& operator=(const Array<T>&rhs); //重载"="使数组对象可以整体赋值
T & operator [](int i); //重载"[]",使Array对象可以起到C++普通数组的作用
const T & operator [](int i) const; //"[]"运算符的const版本
operator T * (); //重载到 T* 类型的转换,使Array对象可以起到C++普通数组的作用
operator T * () const; //重载到 T* 类型的转换的const版本
int getSize() const; //取数组的大小
void resize(int sz); //修改数组的大小
};
//构造函数
template <class T>
Array<T>::Array(int sz)
{
assert(sz>=0); //sz为数组大小(元素个数),应当为负
size = sz; //将元素个数赋值给变量size
list = new T [size]; //动态分配size个T类型的元素空间
}
//析构函数
template <class T>
Array<T>::~Array()
{
delete [] list;
}
//复制构造函数
template <class T>
Array<T>::Array(const Array<T>&a)
{
//从对象 X 取得数组大小,并赋值给当前对象的成员
size = a.size;
//为对象申请内存并进行出错检查
list = new T[size]; //动态内存分配n个T类型的元素空间
//从对象 X 复制数组元素到本对象
for(int i=0; i<size; i++)
{
list[i] = a.list[i];
}
}
//重载"="运算符,将对象 rhs 赋值给本对象, 实现对象之间的整体复制
template <class T>
Array<T>& Array<T>::operator=(const Array<T>&rhs)
{
if(&rhs != this)
{
//如果本对象中数组大小与rhs不同,则删除数组原有内存,然后重新分配
if(size != rhs.size)
{
delete [] list; //删除数组原有内存
size = rhs.size; //设置本对象的数组大小
list = new T[size]; //重新分配n个元素的内存
}
//从对象 X 复制数组元素到本对象
for(int i=0; i<size; i++)
{
list[i] = rhs.list[i];
}
}
return *this;
}
//重载下标运算符,实现与普通数组一样通过下标访问元素,并且具有越界检查功能
template <class T>
T &Array<T>::operator[](int i)
{
assert(i>=0 && i<size); //检查下标是否越界
return list[i]; //返回下标为n的数组元素
}
template <class T>
const T &Array<T>::operator[](int i) const
{
assert(i>=0 && i<size); //检查下标是否越界
return list[i]; //返回下标为n的数组元素
}
//重载指针转换运算符,将Array类的对象名转换为T类型的指针,指向当前对象中的私有数组
//因而可以像使用普通数组首地址一样使用Array类的对象名
template <class T>
Array<T>::operator T * ()
{
return list; //返回当前对象中私有数组的首地址
}
template <class T>
Array<T>::operator T * () const
{
return list; //返回当前对象中私有数组的首地址
}
//取当前数组的大小
template <class T>
int Array<T>::getSize() const
{
return size;
}
//将数组大小修改为sz
template <class T>
void Array<T>::resize(int sz)
{
assert(sz>=0); //检查sz是否非负
if(sz == size)return; //如果指定的大小和原有大小一样,什么也不做
T* newList = new T [sz]; //申请新的数组内存
int n = (sz < size) ? sz : size; //将sz与size中较小的一个赋值给n
//将原有数组中前n个元素复制到新数组中
for(int i=0; i<n; i++)
{
newList[i] = list[i];
}
delete[] list; //删除原数组
list = newList; //使list指向新数组
size = sz; //更新size
}
int main()
{
Array<int> a(10);
a[0] = 5;
cout<<a[0];
return 0;
}
就到这里啦,拜拜~