带你深入理解STL之Vector容器

C++内置了数组的类型，在使用数组的时候，必须指定数组的长度，一旦配置了就不能改变了，通常我们的做法是：尽量配置一个大的空间，以免不够用，这样做的缺点是比较浪费空间，预估空间不当会引起很多不便。

STL实现了一个Vector容器，该容器就是来改善数组的缺点。vector是一个动态空间，随着元素的加入，它的内部机制会自行扩充以容纳新元素。因此，vector的运用对于内存的合理利用与运用的灵活性有很大的帮助，再也不必因为害怕空间不足而一开始就配置一个大容量数组了，vector是用多少就分配多少。

要想实现动态分配数组，Vector内部就需要对空间控制做到有效率的掌控，这些机制要如何运作才能高效地实现动态分配呢？本篇博客就从源代码的角度带你领略一下Vector容器内部的构造艺术。

Vector概述

大家知道，初始化一个数组的时候，需要给数组分配一块内存，数组中的数据都是按序存放的。vector也是如此，再初始化的时候给vector容器分配一块内存，用来存放容器中的数据，一旦分配的内存不足以存放新加入的数据时，就需要扩充空间。STLVector的做法是：重新开辟一段新的空间，将原空间的数据迁移过去，然后新加入的数据存放在新空间之后并释放掉原有空间。

在这个过程中，配置新空间->数据移动->释放旧空间会带来一定的时间成本，所以必须尽可能高效的实现，STL的Vector设计中对这一部分做了相当大的优化，使得时间成本尽可能的小。下面就一起去看看这些优秀的设计吧↓。

Vector的数据结构

我们从最简单的开始，Vector的数据结构相当简单，由于需要判断内存是否够用，所以要用到三个指针，分别指向头，目前使用空间的尾，目前可用空间的尾。其源代码如下：

template < class T, class Alloc = alloc>//alloc是STL的空间配置器 class vector { // 这里提供STL标准的allocator接口 typedef simple_alloc<value_type, Alloc> data_allocator; iterator start; // 内存空间起始点 iterator finish; // 当前使用的内存空间结束点 iterator end_of_storage; // 实际分配内存空间的结束点 }

每当初始化一个vector的时候，先分配一段内存，称为目前可用空间，大小为end_of_storage - start + 1，当往vector里面加入数据的时候，finish就往后移，代表目前已使用的空间，这样做的好处是，不用频繁的扩充空间和转移数据，使得时间成本下降。

在上述代码中，我们看到vector采用了STL标准的空间配置其接口，关于空间配置器的知识在带你深入理解STL之空间配置器(思维导图+源码)一文中有讲解，如有疑惑，可以跳转复习一下再来！

vector提供了如下函数来支持获取其数据结构中的相关参数。

//获取指向vector首元素的迭代器 iterator begin() { return start; } //获取指向vector尾元素的迭代器 iterator end() { return finish; } // 返回当前对象个数，即已使用空间的大小 size_type size() const { return size_type(end() - begin()); } // 返回重新分配内存前最多能存储的对象个数，即目前可用空间的大小 size_type capacity() const { return size_type(end_of_storage - begin()); }

Vector的迭代器

既然是STL的容器，必须要满足迭代器的相关要求，如对迭代器有疑惑的，参考带你深入理解STL之迭代器和Traits技法。

vector维护的是一段连续的内存空间，所以不论容器中元素的型别为何，普通指针都可以作为vector的迭代器而满足所有必要的条件。vector支持随机存取，所以vector提供的是Random Access Iterator。

下面来看看vector关于迭代器的源码：

template < class T, class Alloc = alloc> class vector { public: // vector内部是连续内存空间，所以迭代器采用原生指针即可 typedef value_type* iterator; //以下为满足Traits功能定义的内嵌型别 typedef T value_type; typedef value_type* pointer; typedef const value_type* const_pointer; typedef const value_type* const_iterator; typedef value_type& reference; typedef const value_type& const_reference; typedef ptrdiff_t difference_type; typedef size_t size_type; }

vector的构造函数

默认构造函数

在使用vector的时候，我们通常会有如下定义：

#include <vector> vector< int> vec;

在上述定义中，调用了vector的默认构造函数，其默认不分配内存空间，如下：

// vector的默认构造函数默认不分配内存空间 vector() : start( 0), finish( 0), end_of_storage( 0) {}

带参构造函数

通常，vector的初始化可以指定元素个数和初始化类型。如下：

vector< int> vec( 10, 1); // 将vec初始化为10个1

vector提供下面的构造函数以支持上述初始化操作：

带参构造函数

// 构造函数，允许指定vector的元素个数和初值 vector(size_type n, const T& value) { fill_initialize(n, value); } vector( int n, const T& value) { fill_initialize(n, value); } vector( long n, const T& value) { fill_initialize(n, value); } // 需要对象提供默认构造函数 explicit vector(size_type n) { fill_initialize(n, T()); } /** * 填充并予以初始化 */ void fill_initialize(size_type n, const T& value) { start = allocate_and_fill(n, value); finish = start + n; // 设置当前使用内存空间的结束点 //这里不过多的分配内存 end_of_storage = finish; } /** * 配置一块大小为n的内存空间，并予以填充 */ iterator allocate_and_fill(size_type n, const T& x) { // 调用STL的空间配置器配置一块大小为n的内存空间 iterator result = data_allocator::allocate(n); // 调用底层函数uninitialized_fill_n予以填充 uninitialized_fill_n(result, n, x); return result; }

这里面调用了uninitialized_fill_n函数，这个函数是STL的内存基本处理函数，存放在stl_uninitialized.h中，下面来看看它的源码：

// 如果copy construction和operator =等效, 并且destructor is trivial // 那么就可以使用本函数 template < class ForwardIterator, class Size, class T> inline ForwardIterator __uninitialized_fill_n_aux(ForwardIterator first, Size n, const T& x, __true_type) { return fill_n(first, n, x); } // 不是POD类型使用以下函数 template < class ForwardIterator, class Size, class T> ForwardIterator __uninitialized_fill_n_aux(ForwardIterator first, Size n, const T& x, __false_type) { ForwardIterator cur = first; for ( ; n > 0; --n, ++cur) construct(&*cur, x); return cur; } // 利用type_traits来判断是否是POD类型 template < class ForwardIterator, class Size, class T, class T1> inline ForwardIterator __uninitialized_fill_n(ForwardIterator first, Size n, const T& x, T1*) { typedef typename __type_traits<T1>::is_POD_type is_POD; return __uninitialized_fill_n_aux(first, n, x, is_POD()); } // 利用Iterator_traits来萃取出其值类型 template < class ForwardIterator, class Size, class T> inline ForwardIterator uninitialized_fill_n(ForwardIterator first, Size n, const T& x) { return __uninitialized_fill_n(first, n, x, value_type(first)); }

vector的元素操作函数

push_back()

push_back()函数将新元素插入于vector的尾部，该函数再完成这一操作的时候，先检查是否还有备用空间，如果有直接再备用空间上构造函数；如果没有就扩充空间，通过重新配置一块大空间，移动数据，释放原空间的操作来完成push_back操作。其源代码如下：

// 向容器尾追加一个元素, 可能导致内存重新分配 // push_back(const T& x) // | // |---------------- 容量已满? // | // ---------------------------- // No | | Yes // | | // ↓ ↓ // construct(finish, x); insert_aux(end(), x); // ++finish; | // |------ 内存不足, 重新分配 // | 大小为原来的2倍 // new_finish = data_allocator::allocate(len); <stl_alloc.h> // uninitialized_copy(start, position, new_start); <stl_uninitialized.h> // construct(new_finish, x); <stl_construct.h> // ++new_finish; // uninitialized_copy(position, finish, new_finish); <stl_uninitialized.h> void push_back(const T& x) { // 内存满足条件则直接追加元素, 否则需要重新分配内存空间 if (finish != end_of_storage) { construct(finish, x); ++finish; } else insert_aux(end(), x); } // 提供插入操作 // insert_aux(iterator position, const T& x) // | // |---------------- 容量是否足够? // ↓ // ----------------------------------------- // Yes | | No // | | // ↓ | // 从opsition开始, 整体向后移动一个位置 | // construct(finish, *(finish - 1)); | // ++finish; | // T x_copy = x; | // copy_backward(position, finish - 2, finish - 1); | // *position = x_copy; | // ↓ // data_allocator::allocate(len); // uninitialized_copy(start, position, new_start); // construct(new_finish, x); // ++new_finish; // uninitialized_copy(position, finish, new_finish); // destroy(begin(), end()); // deallocate(); template < class T, class Alloc> void vector<T, Alloc>::insert_aux(iterator position, const T& x) { if (finish != end_of_storage) { // 还有剩余内存 construct(finish, *(finish - 1)); ++finish; T x_copy = x; copy_backward(position, finish - 2, finish - 1); *position = x_copy; } else { // 内存不足, 需要重新分配 const size_type old_size = size(); //配置原则：如果原大小为0，就配置1个元素大小 // 如果原大小不为0，就配置原大小的两倍 // 前半段用来放置原数据，后半段用来放置新数据 const size_type len = old_size != 0 ? 2 * old_size : 1; iterator new_start = data_allocator::allocate(len); iterator new_finish = new_start; // 将内存重新配置 __STL_TRY { // 将原vector的内容拷贝到新vector new_finish = uninitialized_copy(start, position, new_start); // 构造新元素并赋值为x construct(new_finish, x); // 调整finish的位置 ++new_finish; // 将安插点的原内容也拷贝过来 new_finish = uninitialized_copy(position, finish, new_finish); } // 分配失败则抛出异常 catch (...) { destroy(new_start, new_finish); data_allocator::deallocate(new_start, len); throw; } // 析构原容器中的对象 destroy(begin(), end()); // 释放原容器分配的内存 deallocate(); // 调整内存指针状态 start = new_start; finish = new_finish; end_of_storage = new_start + len; } }

pop_back()函数

pop_back函数弹出当前尾端元素。其源代码比较简单，如下：

void pop_back() { //调整finish --finish; //释放调弹出的元素 destroy(finish); }

erase()函数

erase函数支持两个版本：

• 清除某个位置上的元素

iterator erase(iterator position) { if (position + 1 != end()) copy(position + 1, finish, position); //将[position+1,finish]移到[position,finish] --finish; destroy(finish); return position; //返回删除点的迭代器 }

• 清除某个区间上的所有函数

iterator erase(iterator first, iterator last) { iterator i = copy(last, finish, first); //关于copy函数的源码分析在以后的博文中会提到 // 析构掉需要析构的元素 destroy(i, finish); finish = finish - (last - first); return first; }

这里放上两张《STL源码剖析》中的图，便于理解这一过程：

erase函数

有上述erase函数，可以衍生出一个函数，用来清除迭代器中所有的元素

void clear() { erase(begin(), end()); }

insert()函数

insert函数实现的功能是：从position开始，插入n个元素，元素的初值均为x。其源码如下：

// 在指定位置插入n个元素 // insert(iterator position, size_type n, const T& x) // | // |---------------- 插入元素个数是否为0? // ↓ // ----------------------------------------- // No | | Yes // | | // | ↓ // | return; // |----------- 内存是否足够? // | // ------------------------------------------------- // Yes | | No // | | // |------ (finish - position) > n? | // | 分别调整指针 | // ↓ | // ---------------------------- | // No | | Yes | // | | | // ↓ ↓ | // 插入操作, 调整指针 插入操作, 调整指针 | // ↓ // data_allocator::allocate(len); // new_finish = uninitialized_copy(start, position, new_start); // new_finish = uninitialized_fill_n(new_finish, n, x); // new_finish = uninitialized_copy(position, finish, new_finish); // destroy(start, finish); // deallocate(); template < class T, class Alloc> void vector<T, Alloc>::insert(iterator position, size_type n, const T& x) { // 如果n为0则不进行任何操作 if (n != 0) { if (size_type(end_of_storage - finish) >= n) { // 剩下的内存够分配 T x_copy = x; const size_type elems_after = finish - position; // 计算插入点之后的现有元素个数 iterator old_finish = finish; if (elems_after > n) { // 插入点之后的现有元素个数大于新增元素个数，见下图1 // 先复制尾部n个元素到尾部 uninitialized_copy(finish - n, finish, finish); finish += n; // 调整新的finish // 从后往前复制剩余的旧元素 copy_backward(position, old_finish - n, old_finish); // 从position开始填充新元素 fill(position, position + n, x_copy); } else { // 插入点之后的现有元素个数小于新增元素个数，见下图2 // 先在尾部填充n - elems_after个新增元素 uninitialized_fill_n(finish, n - elems_after, x_copy); // 调整新的finish finish += n - elems_after; // 复制[position,old_finish]区间的数到新的finish之后 uninitialized_copy(position, old_finish, finish); // 调整finish finish += elems_after; // 从position开始填充新增元素 fill(position, old_finish, x_copy); } } else { // 剩下的内存不够分配, 需要重新分配 const size_type old_size = size(); const size_type len = old_size + max(old_size, n); iterator new_start = data_allocator::allocate(len); iterator new_finish = new_start; __STL_TRY { // 将旧的vector中插入点之前的元素复制到新空间，见下图3 new_finish = uninitialized_copy(start, position, new_start); // 将新增元素复制到新空间 new_finish = uninitialized_fill_n(new_finish, n, x); // 将插入点之后的元素复制到新空间 new_finish = uninitialized_copy(position, finish, new_finish); } catch (...) { destroy(new_start, new_finish); data_allocator::deallocate(new_start, len); throw; } // 清除并释放原有vector destroy(start, finish); deallocate(); // 调整新的start和finish start = new_start; finish = new_finish; end_of_storage = new_start + len; } } }

上述操作可以使插入操作达到最高的效率。配合以下图解更容易理解：

• 插入点之后的现有元素个数大于新增元素个数的情况

  
  第一种情况

• 插入点之后的现有元素个数小于新增元素个数的情况

  
  第二种情况

• 剩下的内存不够分配，重新配置的情况

  
  第三种情况

后记

STL的Vector容器到此就介绍完毕了。其中对改善效率作了不少优化，很多设计点都值得学习！若有疑惑可以在博文下方留言，我看到会及时帮大家解答！

参考：

• C++ STL源码剖析
• STL源码剖析——迭代器
• 侯捷先生的《STL源码剖析》 华中科技大学出版社

文章作者: Zeech

文章链接: https://zcheng.ren/2016/08/23/STLVector/

版权声明: 本博客所有文章除特别声明外，均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 「 Zeech 」！