C++用类模板实现可变长数组

为加深对类模板应用的理解,下面的程序给出一个可变长数组类模板的实现。

可变长数组类模板:
  1. #include <iostream>
  2. #include <cstring>
  3. using namespace std;
  4. template <class T>
  5. class CArray
  6. {
  7. int size; //数组元素的个数
  8. T *ptr; //指向动态分配的数组
  9. public:
  10. CArray(int s = 0); //s代表数组元素的个数
  11. CArray(CArray & a);
  12. ~CArray();
  13. void push_back(const T & v); //用于在数组尾部添加一个元素v
  14. CArray & operator=(const CArray & a); //用于数组对象间的赋值
  15. T length() { return size; }
  16. T & operator[](int i)
  17. {//用以支持根据下标访问数组元素,如a[i] = 4;和n = a[i]这样的语句
  18. return ptr[i];
  19. }
  20. };
  21. template<class T>
  22. CArray<T>::CArray(int s):size(s)
  23. {
  24. if(s == 0)
  25. ptr = NULL;
  26. else
  27. ptr = new T[s];
  28. }
  29. template<class T>
  30. CArray<T>::CArray(CArray & a)
  31. {
  32. if(!a.ptr) {
  33. ptr = NULL;
  34. size = 0;
  35. return;
  36. }
  37. ptr = new T[a.size];
  38. memcpy(ptr, a.ptr, sizeof(T ) * a.size);
  39. size = a.size;
  40. }
  41. template <class T>
  42. CArray<T>::~CArray()
  43. {
  44. if(ptr) delete [] ptr;
  45. }
  46. template <class T>
  47. CArray<T> & CArray<T>::operator=(const CArray & a)
  48. { //赋值号的作用是使"="左边对象里存放的数组,大小和内容都和右边的对象一样
  49. if(this == & a) //防止a=a这样的赋值导致出错
  50. return * this;
  51. if(a.ptr == NULL) { //如果a里面的数组是空的
  52. if( ptr )
  53. delete [] ptr;
  54. ptr = NULL;
  55. size = 0;
  56. return * this;
  57. }
  58. if(size < a.size) { //如果原有空间够大,就不用分配新的空间
  59. if(ptr)
  60. delete [] ptr;
  61. ptr = new T[a.size];
  62. }
  63. memcpy(ptr,a.ptr,sizeof(T)*a.size);
  64. size = a.size;
  65. return *this;
  66. }
  67. template <class T>
  68. void CArray<T>::push_back(const T & v)
  69. { //在数组尾部添加一个元素
  70. if(ptr) {
  71. T *tmpPtr = new T[size+1]; //重新分配空间
  72. memcpy(tmpPtr,ptr,sizeof(T)*size); //拷贝原数组内容
  73. delete []ptr;
  74. ptr = tmpPtr;
  75. }
  76. else //数组本来是空的
  77. ptr = new T[1];
  78. ptr[size++] = v; //加入新的数组元素
  79. }
  80. int main()
  81. {
  82. CArray<int> a;
  83. for(int i = 0;i < 5;++i)
  84. a.push_back(i);
  85. for(int i = 0; i < a.length(); ++i)
  86. cout << a[i] << " ";
  87. return 0;
  88. }