template <class T>
class vector
{
public:
typedef T* iterator;
typedef const T* const_iterator;
iterator begin()
{
return _start;
}
iterator end()
{
return _finish;
}
const_iterator begin() const
{
return _start;
}
const_iterator end() const
{
return _finish;
}
private:
iterator _start;
iterator _finish;
iterator _endofstorage;
};
size_t size() const
{
return _finish - _start;
}
size_t capacity() const
{
return _endofstorage - _start;
}
T& operator[](size_t pos)
{
assert(pos < size());
return _start[pos];
}
const T& operator[](size_t pos) const
{
assert(pos < size());
return _start[pos];
}
bool empty() const
{
return _start == _finish;
}
vector()
{
_start = nullptr;
_finish = nullptr;
_endofstorage = nullptr;
}
~vector()
{
if (_start)
{
delete[] _start;
}
_start = _finish = _endofstorage;
}
template <class InputIterator>
vector(InputIterator first, InputIterator end)
:_start(nullptr)
, _finish(nullptr)
, _endofstorage(nullptr)
{
while (first != end)
{
push_back(*first);
first++;
}
}
vector当容量不足时需要扩容,reserve的重要性不言而喻,后面的insert,push_back,拷贝构造都需要用到,但是编写代码时却很容易出错
大家应该都能想到,先判断传入的参数是否大于目前容量,大于则进行扩容,即申请一块新的空间,将旧的空间数据拷贝过去,然后释放旧空间,最后更新3个成员变量的指向
最容易错的地方就是将旧空间的数据拷贝到新空间的这一步
我们通常会想到memcpy,方便快捷,还是内存拷贝,不用担心数据类型,似乎毫无问题,但其实有着重大的隐患
如果vector中存的是内置类型,比如int,char那么没有什么问题,但是要是存的是string,问题就来了
memcpy直接拷贝了string的_str,_size,_capacity,而_str指向了一片堆上的空间,这就变成了浅拷贝,完成拷贝后,我们释放旧空间,编译器也会调用string的析构函数释放其指向的空间
那我们新空间的_str就成了野指针,指向了一片未知空间,最后vector使用完毕调用析构函数时,就会再次释放_str指向的空间,而此时这片空间已经被释放过了,不再是堆上分配来的空间,我们无权释放,硬性释放会导致我们的程序崩溃
解决方法就是不使用memcpy,而是遍历使用=操作符进行赋值,对于string,编译器会调用string的operator=完成深拷贝,就可以避免上述问题
void reserve(size_t n)
{
size_t len = size();
if (n > capacity())
{
T* tmp = new T[n];
if (_start)//_start不为空指针
{
//禁止使用memcpy,对于需要深拷贝的对象会出问题
//比如vector
for (int i = 0; i < len; i++)
{
tmp[i] = _start[i];//string的operator=
}
delete[] _start;
}
_start = tmp;
_finish = _start + len;
_endofstorage = _start + n;
}
}
const T& val = T()
,利用T类型的自己的构造函数构造一个匿名对象作为缺省值void resize(size_t n, const T& val = T())
{
if (n <= size())
{
_finish = _start + n;
}
else
{
if (n > capacity())
{
reserve(n);
}
while (_finish < _start + n)
{
*_finish = val;
_finish++;
}
}
}
vector(const vector<T>& v)
:_start(nullptr)
, _finish(nullptr)
, _endofstorage(nullptr)
{
reserve(v.capacity()); //直接push_back的话capacity可能不一样大
for (auto e : v)
{
push_back(e);
}
/*_start = new T[v.capacity()];
memcpy(_start, v._start, sizeof(T) * v.size());
_endofstorage = _start + v.capacity();
_finish = _start + v.size();*/
}
void swap(vector<T>& v)
{
std::swap(_start, v._start);
std::swap(_finish, v._finish);
std::swap(_endofstorage, v._endofstorage);
}
vector<T>& operator=(vector<T> v)
{
swap(v);
return *this;
}
void push_back(const T& x)
{
if (_finish == _endofstorage)
{
size_t newcapacity = capacity() == 0 ? 4 : capacity() * 2;
reserve(newcapacity);
}
*_finish = x;
_finish++;
}
void pop_back()
{
assert(!empty());
_finish--;
}
iterator insert(iterator pos, const T& x)
{
assert(pos < _finish);
if (_finish == _endofstorage)
{
size_t len = pos - _start;
size_t newcapacity = capacity() == 0 ? 4 : capacity() * 2;
reserve(newcapacity);
pos = _start + len;
}
iterator end = _finish - 1;
while (end >= pos)
{
*(end + 1) = *end;
end--;
}
*pos = x;
_finish++;
return pos;
}
iterator erase(iterator pos)
{
assert(pos < _finish);
iterator it = pos + 1;
while (it != _finish)
{
*(it - 1) = *it;
it++;
}
_finish--;
return pos;
}