C++中STL中簡單的Vector的實現

該vector只能容納標准庫中string類，

直接上代碼了，StrVec.h文件內容為:

#ifndef STRVEC_H
#define STRVEC_H

#include<iostream>
#include<string>
#include<memory>
using namespace std;
class StrVec{
public:
    // 默認構造函數
    StrVec():elements(nullptr),first_free(nullptr),cap(nullptr)
    {
        
    }
    //拷貝構造函數
    StrVec(const StrVec& );
    //拷貝賦值運算符
    StrVec& operator=(const StrVec&);
    ~StrVec();
    void push_back(const string&);
    size_t size() const { return first_free - elements ; }
    size_t capacity() const {  return cap - elements  ;}
    string* begin() const { return elements ;}
    string* end() const { return first_free ;}
    
    
    
private:
static allocator<string> alloc;  // 分配元素
 // 被添加的元素的函數使用
 void chk_n_alloc()
 {
     if( size() == capacity() )
            reallocate();

 }
// 工具函數，被拷貝構造函數、賦值運算符和析構函數所使用
pair<string* ,string*> alloc_n_copy(const string* ,const string* );

  
void free();
void reallocate();
string* elements; // 指向數組首元素的指針
string* first_free; // 指向數組第一個空閑元素的指針
string* cap; // 指向數組尾后的指針

    
};

#endif

StrVec.cpp文件內容為:

#include "StrVec.h"

std::allocator<string> StrVec::alloc;

void StrVec::push_back( const string& s )
{
    chk_n_alloc(); // 確保有空間容納新元素
    // first_free指向的元素中構造s的副本
    alloc.construct(first_free++,s)  ; 
        
    
}
pair<string* ,string* > StrVec::alloc_n_copy(const string* b, const string* e )
{
    auto data = alloc.allocate(e-b);
    return make_pair(data , uninitialized_copy(b,e,data)  );
    
    
}
void StrVec::free()
{
    // 不能傳遞給deallcoate一個空指針
    if( elements)
    {
        for(auto p = first_free ; p != elements ; )
            alloc.destroy(--p);
        alloc.deallocate(elements,cap-elements) ;
    }
    
    
}
StrVec& StrVec::operator=( const StrVec& rhs )
{
    auto data = alloc_n_copy(rhs.begin() , rhs.end() );
    free() ;
    elements = data.first ;
    first_free= cap=data.second;
    
    return *this ;
    
}
void StrVec::reallocate()
{
    // 我們將分配當前大小兩倍的內存空間;
    auto newcapacity  = size() ? 2*size() :1 ;
    //分配新內存
    auto newdata = alloc.allocate( newcapacity ) ;
    
    auto dest=newdata ;
    auto elem = elements ;
    for(size_t i=0; i != size() ; i++)
        alloc.construct(dest++,std::move( *elem++) );
    free();
    elements = newdata;
    first_free = dest ;
    cap = elements+ newcapacity ;
    
    
}
StrVec::~StrVec()
{
    free();
}

測試代碼為maintest.cpp

#include "StrVec.h"

int main()
{
    StrVec vec1;
    vec1.push_back("ok1");
    vec1.push_back("ok2");

    auto begin = vec1.begin();
    auto end= vec1.end();

    while( begin != end )
    {
      cout<<*begin<<endl;
      // cout<<endl;
      begin++;

    }

 return 0;
}