C++用類別範本實現可變長陣列

2020-07-16 10:04:21
為加深對類別範本應用的理解,下面的程式給出一個可變長陣列類別範本的實現。

可變長陣列類別範本:
#include <iostream>
#include <cstring>
using namespace std;
template <class T>
class CArray
{
    int size; //陣列元素的個數
    T *ptr; //指向動態分配的陣列
public:
    CArray(int s = 0);  //s代表陣列元素的個數
    CArray(CArray & a);
    ~CArray();
    void push_back(const T & v); //用於在陣列尾部新增一個元素v
    CArray & operator=(const CArray & a); //用於陣列物件間的賦值
    T length() { return size; }
    T & operator[](int i)
    {//用以支援根據下標存取陣列元素,如a[i] = 4;和n = a[i]這樣的語句
        return ptr[i];
    }
};
template<class T>
CArray<T>::CArray(int s):size(s)
{
     if(s == 0)
         ptr = NULL;
    else
        ptr = new T[s];
}
template<class T>
CArray<T>::CArray(CArray & a)
{
    if(!a.ptr) {
        ptr = NULL;
        size = 0;
        return;
    }
    ptr = new T[a.size];
    memcpy(ptr, a.ptr, sizeof(T ) * a.size);
    size = a.size;
}
template <class T>
CArray<T>::~CArray()
{
     if(ptr) delete [] ptr;
}
template <class T>
CArray<T> & CArray<T>::operator=(const CArray & a)
{ //賦值號的作用是使"="左邊物件裡存放的陣列,大小和內容都和右邊的物件一樣
    if(this == & a) //防止a=a這樣的賦值導致出錯
    return * this;
    if(a.ptr == NULL) {  //如果a裡面的陣列是空的
        if( ptr )
            delete [] ptr;
        ptr = NULL;
        size = 0;
        return * this;
    }
     if(size < a.size) { //如果原有空間夠大,就不用分配新的空間
         if(ptr)
            delete [] ptr;
        ptr = new T[a.size];
    }
    memcpy(ptr,a.ptr,sizeof(T)*a.size);   
    size = a.size;
     return *this;
}
template <class T>
void CArray<T>::push_back(const T & v)
{  //在陣列尾部新增一個元素
    if(ptr) {
        T *tmpPtr = new T[size+1]; //重新分配空間
    memcpy(tmpPtr,ptr,sizeof(T)*size); //拷貝原陣列內容
    delete []ptr;
    ptr = tmpPtr;
}
    else  //陣列本來是空的
        ptr = new T[1];
    ptr[size++] = v; //加入新的陣列元素
}
int main()
{
    CArray<int> a;
    for(int i = 0;i < 5;++i)
        a.push_back(i);
    for(int i = 0; i < a.length(); ++i)
        cout << a[i] << " ";   
    return 0;
}