stl vector 模拟实现修改版

1. // Iterator.cpp : Defines the entry point for the console application.
2. //
4. #include "stdafx.h"
6. #include <iostream>
7. #include <new>
9. using namespace std;
11. template <class T>
12. class Iterator
13. {
14. public:
15.     Iterator()
16.         : m_pData(NULL)//, m_nPos(0)
17.         , m_nSize(0), m_nLen(0)
18.     {
19.     }
21.     virtual ~Iterator()
22.     {
23.         if (NULL != m_pData)
24.         {
25.             delete[] m_pData;
26.             m_pData = NULL;
27.         }
28.     }
30. public:
31.     virtual T* begin() = 0;
32.     virtual T* end() = 0;
34. protected:
35.     // int  m_nPos;
36.     int m_nSize;
37.     int m_nLen;
38.     T *m_pData;
39. };
41. template <class T>
42. class CMyVector : public Iterator<T>
43. {
44. public:
45.     typedef T* iterator;
47. public:
48.     CMyVector(int nSize = 10)
49.     {
50.         if (nSize <= 0)
51.         {
52.             nSize = 10;
53.         }
55.         m_nSize = nSize;
56.         m_pData = new T[m_nSize];
57.     }
59.     ~CMyVector()
60.     {
61.     }
63. public:
64.     int Length() const
65.     {
66.         return m_nLen;
67.     }
69.     bool push_back(T obj)
70.     {
71.         if (m_nLen >= m_nSize)
72.         {
73.             //get more memory
74.             int nSize = m_nSize * 2 + 1;
76.             T *pTemp = new(nothrow) T[nSize];
77.             if (!pTemp)
78.             {
79.                 return false;
80.             }
82.             // memset(pTemp, 0, sizeof(T) * m_nSize);
83.             memcpy(pTemp, m_pData, sizeof(T) * m_nLen);         
85.             delete []m_pData;
86.             m_pData = pTemp;
87.             m_nSize = nSize;
88.         }
90.         memcpy(m_pData + m_nLen, &obj, sizeof(obj));
91.         m_nLen++;
93.         return true;
94.     }
96. public:
97.     virtual T* end()
98.     {
99.         return m_pData + m_nLen;
100.     }
102.     virtual T* begin()
103.     {
104.         // m_nPos = 0;
105.         return m_pData;
106.     }
107. };
109. int _tmain(int argc, _TCHAR* argv[])
110. {
111.     CMyVector<int> vtData;
112.     CMyVector<int>::iterator it;
114.     for (int i = 0; i < 30; i++)
115.     {
116.         vtData.push_back(i);
117.         //cout << "nLen = " << vtData.Length() << endl;
118.     }
120.     cout << "vector data: " << endl;
121.     for (it = vtData.begin(); it != vtData.end(); it++)
122.     {
123.         cout << *it << "/t";
124.     }
125.     cout << endl;
127.     return 0;
128. }