手写 C++ Vector 容器底层原理与实现
Vector 作为 C++ STL 中最常用的序列容器,其核心在于对动态内存的高效管理。在实现层面,我们主要依赖三个关键指针来追踪数据状态:指向起始位置的 _start,指向有效数据末尾的 _finish,以及指向已分配存储空间末尾的 _endofstorage。
核心成员变量与初始化
这三个指针构成了 Vector 的骨架。构造函数负责将它们初始化为空状态,确保对象创建时处于安全可用状态。
#pragma once
#include <iostream>
#include <assert.h>
using namespace std;
namespace Solution
{
template<class T>
class vector
{
public:
typedef T* iterator;
typedef const T* const_iterator;
vector()
:_start(nullptr)
,_finish(nullptr)
,_endofstorage(nullptr)
{}
//实现拷贝构造,先开一样的空间,再拷贝
vector(const vector<T>& v)
:_start(nullptr)
,_finish(nullptr)
,_endofstorage(nullptr)
{
reserve(v.capacity());
//因为 memcpy 是浅拷贝,故不考虑,用深拷贝赋值拷贝
if (_start)
{
for (size_t i = 0; i < v.size(); ++i)
{
*(_start + i) = *(v._start + i);
}
}
_finish = _start + v.size();
_endofstorage = _start + v.capacity();
}
~vector()
{
delete[] _start;
_start = _finish = _endofstorage = nullptr;
}
//尾插
void push_back(const T& val)
{
if (_finish == _endofstorage)
{
size_t capacity1 = capacity();
size_t newcapacity = capacity1 == 0 ? 4 : 2 * capacity1;
reserve(newcapacity);
}//实现扩容
*_finish = val;
++_finish;
//插入数据
}
//尾删
void pop_back()
{
assert(_start != _finish);
_finish--;
}
//插入
void insert(iterator pos, const T& val)
{
assert(pos >= _start && pos <= _finish);
if (_finish == _endofstorage)
{
size_t capacity1 = capacity();
size_t newcapacity = capacity1 == 0 ? 4 : 2 * capacity1;
reserve(newcapacity);
}//扩容
iterator end1 = end() - 1;
while (end1 >= pos)
{
*(end1+1) = *end1;
--end1;
}
*pos = val;
_finish++;
}
void erase(iterator pos)
{
assert(pos >= _start && pos < _finish);
while (pos < _finish-1)
{
*pos = *(pos + 1);
++pos;
}
--_finish;
}
void resize(size_t n, const T& val = T())
{
if (n < size())
{
_finish = _start + n;
}
else
{
reserve(n);
while (_finish < _start + n)
{
*_finish = val;
++_finish;
}
}
}
void reserve(size_t n)
{
if (n > capacity())
{
size_t sz = size();
T* tmp = new T[n];
//memcpy(tmp, _start, sz);//思考是否有问题
for (size_t i = 0; i < sz; ++i)
{
*(tmp + i) = *(_start + i);
}
delete[] _start;
_start = tmp;
_finish = tmp + sz;
_endofstorage = tmp + n;
}
}
//-----------------
iterator begin()const
{
return _start;
}
iterator end()const
{
return _finish;
}
//--------------------
T& operator[](size_t n)
{
assert(n < size());
return *(_start + n);
}
vector<T>& operator=(const vector<T>& v) // 将一个数据拷贝到另一个数据中
{
if (this != &v)
{
vector<T> v1(v);
Swap(v1);
}
return *this;
}
void Swap(vector <T>& v)
{
swap(_start, v._start);
swap(_finish, v._finish);
swap(_endofstorage, v._endofstorage);
}
size_t size()const
{
return _finish - _start;
}
size_t capacity()const
{
return _endofstorage - _start;
}
private:
T* _start;
T* _finish;
T* _endofstorage;
};
}
内存管理与扩容策略
当空间不足时,我们需要重新分配更大的内存块。这里有一个关键点:对于非 POD 类型,直接使用 memcpy 进行内存复制可能会跳过对象的构造函数或析构函数,导致资源泄漏或未定义行为。因此,我们在 reserve 和拷贝构造中采用了逐元素赋值的方式,确保每个对象都能正确构造。
迭代器与运算符重载
为了提供类似原生数组的体验,我们实现了下标运算符 operator[] 以及迭代器接口。begin() 和 end() 返回的指针可以直接用于遍历,配合 insert 和 erase 操作,基本覆盖了动态数组的核心功能。
赋值运算符重载采用了经典的'拷贝 - 交换'(Copy-and-Swap)惯用法,这不仅保证了异常安全性,还避免了自赋值的问题。通过临时对象 v1 完成深拷贝,然后与当前对象交换内部指针,最后让临时对象在析构时释放旧内存。

