STL源码剖析---vector

vector容器概述

      vector的数据安排以及操作方式,与array非常相似。两者的唯一区别在于空间的运用的灵活性。array是静态空间,一旦配置了就不能改变;要换个大(或小)一点的房子,可以,一切琐细都得由客户端自己来:首先配置一块新空间,然后将元素从旧址一一搬往新址,再把原来的空间释还给系统。vector是动态空间,随着元素的加入,它的内部机制会自行扩充空间以容纳新元素。因此,vector的运用对于内存的合理利用与运用的灵活性有很大的帮助,我们再也不必因为害怕空间不足而一开始要求一个大块头的array了,我们可以安心使用array,吃多少用多少。

      vector的实现技术,关键在于其对大小的控制以及重新配置时的数据移动效率。一旦vector的旧有空间满载,如果客户端每新增一个元素,vector的内部只是扩充一个元素的空间,实为不智。因为所谓扩充空间(不论多大),一如稍早所说,是”配置新空间/数据移动/释还旧空间“的大工程,时间成本很高,应该加入某种未雨绸缪的考虑。稍后我们便可看到SGI vector的空间配置策略了。

      另外,由于vector维护的是一个连续线性空间,所以vector支持随机存取。

      注意:vector动态增加大小时,并不是在原空间之后持续新空间(因为无法保证原空间之后尚有可供配置的空间),而是以原大小的两倍另外配置一块较大的空间,然后将原内容拷贝过来,然后才开始在原内容之后构造新元素,并释放原空间。因此,对vector的任何操作,一旦引起空间重新配置,指向原vector的所有迭代器就都失效了。这是程序员易犯的一个错误,务需小心。

以下是vector定义的源代码摘录:

[cpp] view plain copy

#include<iostream>  

using namespace std;  

#include<memory.h>    

  

// allocSGI STL的空间配置器  

template <class T, class Alloc = alloc>  

class vector  

{  

public:  

    // vector的嵌套类型定义,typedefs用于提供iterator_traits<I>支持  

    typedef T value_type;  

    typedef value_type* pointer;  

    typedef value_type* iterator;  

    typedef value_type& reference;  

    typedef size_t size_type;  

    typedef ptrdiff_t difference_type;  

protected:  

    // 这个提供STL标准的allocator接口  

    typedef simple_alloc <value_type, Alloc> data_allocator;  

  

    iterator start;               // 表示目前使用空间的头  

    iterator finish;              // 表示目前使用空间的尾  

    iterator end_of_storage;      // 表示实际分配内存空间的尾  

  

    void insert_aux(iterator position, const T& x);  

  

    // 释放分配的内存空间  

    void deallocate()  

    {  

        // 由于使用的是data_allocator进行内存空间的分配,  

        // 所以需要同样使用data_allocator::deallocate()进行释放  

        // 如果直接释放对于data_allocator内部使用内存池的版本  

        // 就会发生错误  

        if (start)  

            data_allocator::deallocate(start, end_of_storage - start);  

    }  

  

    void fill_initialize(size_type n, const T& value)  

    {  

        start = allocate_and_fill(n, value);  

        finish = start + n;                         // 设置当前使用内存空间的结束点  

        // 构造阶段此实作不多分配内存,  

        // 所以要设置内存空间结束点和已经使用的内存空间结束点相同  

        end_of_storage = finish;  

    }  

  

public:  

    // 获取几种迭代器  

    iterator begin() { return start; }  

    iterator end() { return finish; }  

  

    // 返回当前对象个数  

    size_type size() const { return size_type(end() - begin()); }  

    size_type max_size() const { return size_type(-1) / sizeof(T); }  

    // 返回重新分配内存前最多能存储的对象个数  

    size_type capacity() const { return size_type(end_of_storage - begin()); }  

    bool empty() const { return begin() == end(); }  

    reference operator[](size_type n) { return *(begin() + n); }  

  

    // 本实作中默认构造出的vector不分配内存空间  

    vector() : start(0), finish(0), end_of_storage(0) {}  

  

  

    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()); }  

  

    vector(const vector<T, Alloc>& x)  

    {  

        start = allocate_and_copy(x.end() - x.begin(), x.begin(), x.end());  

        finish = start + (x.end() - x.begin());  

        end_of_storage = finish;  

    }  

  

    ~vector()  

    {  

        // 析构对象  

        destroy(start, finish);  

        // 释放内存  

        deallocate();  

    }  

  

    vector<T, Alloc>& operator=(const vector<T, Alloc>& x);  

  

    // 提供访问函数  

    reference front() { return *begin(); }  

    reference back() { return *(end() - 1); }  

  

      

    // 向容器尾追加一个元素可能导致内存重新分配  

      

    //                          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(iterator position, const T& x)  

    //                                   |  

    //                                   |------------ 容量是否足够 && 是否是end()?  

    //                                   |  

    //               -------------------------------------------  

    //            No |                                         | Yes  

    //               |                                         |  

    //               ↓                                         ↓  

    //    insert_aux(position, x);                  construct(finish, x);  

    //               |                              ++finish;  

    //               |-------- 容量是否够用?  

    //               |  

    //        --------------------------------------------------  

    //    Yes |                                                | No  

    //        |                                                |  

    //        ↓                                                |  

    // construct(finish, *(finish - 1));                       |  

    // ++finish;                                               |  

    // T x_copy = x;                                           |  

    // copy_backward(position, finish - 2, finish - 1);        |  

    // *position = x_copy;                                     |  

    //                                                         ↓  

    // 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>  

    // destroy(begin(), end());                             <stl_construct.h>  

    // deallocate();  

      

  

    iterator insert(iterator position, const T& x)  

    {  

        size_type n = position - begin();  

        if (finish != end_of_storage && position == end())  

        {  

            construct(finish, x);  

            ++finish;  

        }  

        else  

            insert_aux(position, x);  

        return begin() + n;  

    }  

  

    iterator insert(iterator position) { return insert(position, T()); }  

  

    void pop_back()  

    {  

        --finish;  

        destroy(finish);  

    }  

  

    iterator erase(iterator position)  

    {  

        if (position + 1 != end())  

            copy(position + 1, finish, position);  

        --finish;  

        destroy(finish);  

        return position;  

    }  

  

  

    iterator erase(iterator first, iterator last)  

    {  

        iterator i = copy(last, finish, first);  

        // 析构掉需要析构的元素  

        destroy(i, finish);  

        finish = finish - (last - first);  

        return first;  

    }  

  

    // 调整size, 但是并不会重新分配内存空间  

    void resize(size_type new_size, const T& x)  

    {  

        if (new_size < size())  

            erase(begin() + new_size, end());  

        else  

            insert(end(), new_size - size(), x);  

    }  

    void resize(size_type new_size) { resize(new_size, T()); }  

  

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

  

protected:  

    // 分配空间并且复制对象到分配的空间处  

    iterator allocate_and_fill(size_type n, const T& x)  

    {  

        iterator result = data_allocator::allocate(n);  

        uninitialized_fill_n(result, n, x);  

        return result;  

    }  

  

    // 提供插入操作  

      

    //                 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 insert_aux(iterator position, const T& x)  

    {  

        if (finish != end_of_storage)    // 还有备用空间  

        {  

            // 在备用空间起始处构造一个元素,并以vector最后一个元素值为其初值  

            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();  

            const size_type len = old_size != 0 ? 2 * old_size : 1;  

            // 以上配置元素:如果大小为0,则配置1(个元素大小)  

            // 如果大小不为0,则配置原来大小的两倍  

            // 前半段用来放置原数据,后半段准备用来放置新数据  

  

            iterator new_start = data_allocator::allocate(len);  // 实际配置  

            iterator new_finish = new_start;  

            // 将内存重新配置  

            try  

            {  

                // 将原vector的安插点以前的内容拷贝到新vector  

                new_finish = uninitialized_copy(start, position, new_start);  

                // 为新元素设定初值 x  

                construct(new_finish, x);  

                // 调整水位  

                ++new_finish;  

                // 将安插点以后的原内容也拷贝过来  

                new_finish = uninitialized_copy(position, finish, new_finish);  

            }  

            catch(...)  

            {  

                // 回滚操作  

                destroy(new_start, new_finish);  

                data_allocator::deallocate(new_start, len);  

                throw;  

            }  

            // 析构并释放原vector  

            destroy(begin(), end());  

            deallocate();  

  

            // 调整迭代器,指向新vector  

            start = new_start;  

            finish = new_finish;  

            end_of_storage = new_start + len;  

        }  

    }  

  

      

    // 在指定位置插入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 insert(iterator position, size_type n, const T& x)  

    {  

        // 如果n0则不进行任何操作  

        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)  

                {  

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

                    uninitialized_copy(finish - n, finish, finish);  

                    finish += n;    // vector 尾端标记后移  

                    copy_backward(position, old_finish - n, old_finish);  

                    fill(position, position + n, x_copy); // 从插入点开始填入新值  

                }  

                else  

                {  

                    // 插入点之后的现有元素个数 小于等于 新增元素个数  

                    uninitialized_fill_n(finish, n - elems_after, x_copy);  

                    finish += n - elems_after;  

                    uninitialized_copy(position, old_finish, finish);  

                    finish += elems_after;  

                    fill(position, old_finish, x_copy);  

                }  

            }  

            else  

            {   // 剩下的备用空间小于“新增元素个数”(那就必须配置额外的内存)  

                // 首先决定新长度:就长度的两倍 , 或旧长度+新增元素个数  

                const size_type old_size = size();  

                const size_type len = old_size + max(old_size, n);  

                // 以下配置新的vector空间  

                iterator new_start = data_allocator::allocate(len);  

                iterator new_finish = new_start;  

                __STL_TRY  

                {  

                    // 以下首先将旧的vector的插入点之前的元素复制到新空间  

                    new_finish = uninitialized_copy(start, position, new_start);  

                    // 以下再将新增元素(初值皆为n)填入新空间  

                    new_finish = uninitialized_fill_n(new_finish, n, x);  

                    // 以下再将旧vector的插入点之后的元素复制到新空间  

                    new_finish = uninitialized_copy(position, finish, new_finish);  

                }  

#         ifdef  __STL_USE_EXCEPTIONS  

                catch(...)  

                {  

                    destroy(new_start, new_finish);  

                    data_allocator::deallocate(new_start, len);  

                    throw;  

                }  

#         endif /* __STL_USE_EXCEPTIONS */  

                destroy(start, finish);  

                deallocate();  

                start = new_start;  

                finish = new_finish;  

                end_of_storage = new_start + len;  

            }  

        }  

    }  

};  

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值