文章目录
- vector的底层结构
- 迭代器
- 容量操作
- size()
- capacity()
- reserve()
- resize()
- 默认成员函数
- 构造
- 无参构造函数
- 带参构造函数
- 析构
- 拷贝构造
- 赋值重载
- operator[ ]
- 插入删除操作
- insert()任意位置插入
- erase()任意位置删除
- push_back()尾插
- pop_back()尾删
vector的底层结构
我们的目的不是去模拟实现vector,而是更深入地理解vector的底层原理,更好地提升自己。本篇将简单地模拟实现vector,更好地理解它的构造和原理。参考:vector使用说明
在C++的STL中,vector是最常用的容器之一,底层是一段连续的线性内存空间(泛型的动态类型顺序表),可以支持随机访问。vector可以存储各种类型,int、char、string等,所以它是一种类模板,可以套用各种类型。
STL标准库中vector的核心是这样定义的,这里的alloc涉及到内存池的知识,我们可以先不用管。
vector的底层会用三个指针,利用三个指针相减来实现动态存储。
我们自己定义一个vector类
template<class T>
class vector
{
public:typedef T* iterator;//迭代器typedef const T* const_iterator;//常量迭代器
private:iterator _start;iterator _finish;iterator _end_of_storage;
}
迭代器
vector的迭代器是一个原生指针,他的迭代器和string相同都是操作指针来遍历数据。
迭代器返回的是存储数据的起始位置和末尾的下一个位置,区间是左开右闭的[_start, _finish)
实现了迭代器,我们在代码测试时就可以使用范围for了。
iterator begin()
{return _start;
}
iterator end()
{return _finish;
}
const_iterator begin() const
{return _start;
}
const_iterator end() const
{return _finish;
}
容量操作
size()
size_t size() const
{return _finish - _start;
}
capacity()
size_t capacity() const
{return _end_of_storage - _start;
}
reserve()
这个函数十分重要,因为vector许多地方都会用reserve()去扩容,并且还有几个非常容易搞错的地方。
reserve只能扩容,不能增容,因此要进行判断是否需要扩容,缩容就不进行操作。
扩容的步骤是:申请一块更大的新空间,再将旧空间数据移动到新空间中,最后释放旧空间。
下面这种写法有什么问题?
void reserve(size_t n)
{if (n > capacity()){//申请新空间T* tmp = new T[n];if (_start){memcpy(tmp, _start, sizeof(T) * old_size);//释放旧空间delete[] _start;}_start = tmp;_finish = tmp + size();_end_of_storage = _start + n;}
}
错误一:倒数第二行的_finish没有发生变化
看这段代码的最后三行,_start先指向新空间的起始位置,_finish再调用size()的话,此刻的size()已经不是当初的size()了。size()的返回值是_finish - _start,而原本的_start已经改变成了tmp,此时_finish的值 = _start + _finish - _start = _finish;所以_finish没有发生变化。
解决方法有两种:
1._start和_finish赋值的顺序调换一下,先改变_finish,再改变_start。
2.挪动数据前先保留原本的size();
我们这里采用第二种写法。void reserve(size_t n) {if (n > capacity()){T* tmp = new T[n];size_t old_size = size();//保留之前的sizeif (_start){memcpy(tmp, _start, sizeof(T) * old_size);delete[] _start;}_start = tmp;_finish = tmp + old_size;_end_of_storage = _start + n;} }
2.不能用memcpy去拷贝内容,而用赋值去拷贝内容。
对于int、char等内置类型,可以使用memcpy不会出问题。对于自定义类型一般也没有问题,但是对于动态申请资源的自定义类型,memcpy就会发生浅拷贝,导致一块空间析构两次。
比如vector<string>
类型,此时的T是string类型。上一篇我们了解过string的底层原理,string底层用的是char*指针_str来存储字符串的地址,因此需要动态申请空间。
所以我们是在申请的空间(_start)上面又申请了一块空间(_str),如果我们使用memcpy去拷贝_start中的内容到tmp中;就会把申请的_str指向的地址拷贝给tmp,这样_start和tmp中的_str就会指向同一块空间,再执行
delete[] _start;
的时候就会执行string的析构函数把这块空间释放。
所以不能用memcpy浅拷贝,解决方法:
一个个遍历用=赋值,对于string这种深拷贝的类,调用的是string的赋值重载完成深拷贝。
注意:这里的string使用的是STL库中的string类。void reserve(size_t n) {if (n > capacity()){T* tmp = new T[n];size_t old_size = size();//保留之前的sizeif (_start){for (size_t i = 0; i < old_size; i++){tmp[i] = _start[i];}delete[] _start;}_start = tmp;_finish = tmp + old_size;_end_of_storage = _start + n;} }
resize()
resize函数用于改变vector的大小,调整vector中元素的数量。
n > size():多余空间添加元素(第二个参数)
n <= size():删除n后面的元素。
void resize(size_t n, const T& val = T())//匿名对象
{if (n <= size()){_finish = _start + n;}else{reserve(n);while (_finish != _start + n){*_finish++ = val;}}
}
默认成员函数
构造
无参构造函数
vector():_start(nullptr),_finish(nullptr),_end_of_storage(nullptr)
{}
使用
vector<int> v1;
vector<string> s2;
带参构造函数
vector(size_t n, const T& val = T())
{resize(n, val);
}
使用
vector<int> v1(10);//开辟5个int大小空间,默认初始化为0
vector<int> v2(10, 1);//开辟10个int大小空间,并初始化为1
vector<string> s2(10,"abc");//开辟10个string大小空间,并初始化为"abc"
析构
~vector()
{if (_start){delete[] _start;_start = _finish = _end_of_storage = nullptr;}
}
拷贝构造
写法一:尾插法
vector(const vector<T>& v): _start(nullptr), _finish(nullptr), _end_of_storage(nullptr)
{reserve(v.size());for (auto& e : v)//引用防止调用拷贝构造{push_back(e);}
}
写法二:常规法
vector(const vector<T>& v): _start(nullptr), _finish(nullptr), _end_of_storage(nullptr)
{_start = new T[v.size()];for (size_t i = 0; i < v.size(); i++){_start[i] = v._start[i];}_finish = _start + v.size();_end_of_storage = _start + v.capacity();
}
赋值重载
和string一样,我们直接复用标准库的swap函数。
注意:用swap的话,拷贝构造的参数要用传值传递(不能改变实参),且不能用const修饰(发生交换)
void swap(vector<T>& v)
{std::swap(_start, v._start);std::swap(_finish, v._finish);std::swap(_end_of_storage, v._end_of_storage);
}
vector<T>& operator=(vector<T> v)
{swap(v);return *this;
}
operator[ ]
两个重载版本,一个普通对象调用,一个const对象调用。
T& operator[](size_t pos)
{assert(pos < size());return _start[pos];
}
const T& operator[](size_t pos) const
{assert(pos < size());return _start[pos];
}
插入删除操作
insert()任意位置插入
插入之后,如果原始空间发生扩容,则pos指针仍指向原来空间的位置,迭代器就会发生失效。所以我们需要在扩容之前保存pos的相对位置,然后重新指向正确位置。
//insert后 迭代器可能会失效,扩容就会引起失效
iterator insert(iterator pos, const T& val)
{assert(pos >= _start && pos <= _finish);if (_finish == _end_of_storage){size_t len = pos - _start;//保存pos相对位置,防止失效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 = val;_finish++;return pos;
}
但是,这样只保证了能正确插入数据。如果在使用时,我们insert()一个元素后,迭代器还会可能失效,因为我们使用的传值传递,形参不会影响实参。因此insert()函数返回pos这个位置的迭代器,我们在外部使用的时候更新迭代器即可。
vector<int> v(5, 1);
vector<int>::iterator p = v.begin() + 1;
p = v.insert(p, 10);//更新迭代器
erase()任意位置删除
iterator erase(iterator pos)
{assert(pos >= _start && pos < _finish);iterator end = pos;while (end < _finish - 1) {*end = *(end + 1);end++;}_finish--;return pos;
}
删除虽然不会扩容(不用保存pos指针的相对位置),但是删除一个元素后,后面的元素都会向前移动,此时迭代器指向空间虽然不变,但内容变成了下一个元素,我们在使用的时候要注意这个问题。
int main()
{vector<int> v1;v1.push_back(1);v1.push_back(2);v1.push_back(2);v1.push_back(6);auto it = v1.begin();//法一:边使用边更新while (it != v1.end()){if (*it % 2 == 0){it = v1.erase(it);}else{it++;}}//法二:使用完将迭代器减一,防止++出现走两步的情况while (it != v1.end()){if (*it % 2 == 0){v1.erase(it);it--;}it++;}
}
push_back()尾插
实现insert()函数后,我们就可以直接复用。
void push_back(const T& val)
{/*if (_finish == _end_of_storage){size_t newcapacity = capacity() == 0 ? 4 : capacity() * 2;reserve(newcapacity);}*_finish++ = val;*/insert(end(), val);
}
pop_back()尾删
同样直接复用erase(),让erase()自己判断合法性。
void pop_back()
{/*assert(size() > 0);_finish--;*/erase(_finish - 1);
}
vector模拟实现代码