詳解C++?STL模擬實(shí)現(xiàn)vector
vector 概述
vector 的數(shù)據(jù)結(jié)構(gòu)安排及操作方式,與原生數(shù)組十分相似,兩者唯一的差別在于空間運(yùn)用的靈活性。原生數(shù)組是靜態(tài)空間,一旦配置了就不能改變大小;vector 是動(dòng)態(tài)空間,可以在插入過(guò)程中動(dòng)態(tài)調(diào)整空間大小。vector 的實(shí)現(xiàn)技術(shù),關(guān)鍵在于它對(duì)大小的控制及重新配置時(shí)的數(shù)據(jù)移動(dòng)效率。
在文中,將會(huì)挑選 vector 的一些常用接口來(lái)模擬實(shí)現(xiàn),但并不和標(biāo)準(zhǔn)庫(kù)中實(shí)現(xiàn)方式相同。標(biāo)準(zhǔn)庫(kù)中使用了大量?jī)?nèi)存操作函數(shù)來(lái)提高效率,代碼晦澀難懂,不利用初學(xué)者學(xué)習(xí)。本文實(shí)現(xiàn)方式相對(duì)簡(jiǎn)單,但也需要讀者有一定的 STL 使用經(jīng)驗(yàn)。下面就讓我們開(kāi)始吧。
接口總覽
namespace qgw {
template <class T>
class vector {
typedef T* iterator;
typedef const T* const_iterator;
typedef T& reference;
public:
// 默認(rèn)成員函數(shù)
vector(); // 默認(rèn)構(gòu)造函數(shù)
vector(size_t n, const T& val = T()); // 構(gòu)造 n 個(gè) T 類(lèi)型對(duì)象
vector(InputIterator first, InputIterator last);// 用一段區(qū)間構(gòu)造
vector(const vector<T>& v); // 拷貝構(gòu)造函數(shù)
vector<T>& operator=(const vector<T>& v); // 復(fù)制賦值函數(shù)
~vector();
// 迭代器函數(shù)
iterator begin();
const_iterator begin() const;
iterator end();
const_iterator end() const;
// 元素訪問(wèn)
reference operator[](size_t pos);
const reference operator[](size_t pos) const;
reference back();
const reference back() const;
// 容量
bool empty() const;
size_t size() const;
size_t capacity() const;
void resize(size_t cnt, T val = T());
void reserve(size_t cap);
// 修改
iterator insert(iterator pos, const T& val);
void push_back(const T& val);
iterator erase(iterator pos);
iterator erase(iterator first, iterator last);
void pop_back();
void swap(vector<T>& v);
private:
iterator _start; // 表示目前使用空間的頭
iterator _finish; // 表示目前使用空間的尾
iterator _end_of_storage; // 表示已分配空間的尾
};
}
成員變量介紹

vector 中有三個(gè)成員變量,_start 指向使用空間的頭,_finish 指向使用空間的尾,_end_of_storage 指向已分配空間的尾。
由上圖也可以清晰的看出,_finish - _start 就是 size 的大小,_end_of_storage - _start 就是 capacity 的大小。
默認(rèn)成員函數(shù)
構(gòu)造函數(shù)
默認(rèn)構(gòu)造函數(shù)
vector 支持一個(gè)無(wú)參的構(gòu)造函數(shù),在這個(gè)構(gòu)造函數(shù)中我們直接將上文中三個(gè)成員變量初始化為空即可。
/// @brief 默認(rèn)構(gòu)造函數(shù),將指針初始化為空
vector() {
_start = nullptr;
_finish = nullptr;
_end_of_storage = nullptr;
}
構(gòu)造 n 個(gè) T 類(lèi)型對(duì)象
vector 支持構(gòu)造 n 個(gè) 值為 val 的對(duì)象??梢韵扔?reserve開(kāi)辟容量,在調(diào)用 push_back 插入即可。
注意:reserve 改變的是 capacity 的大小,不改變 size 的大小,先開(kāi)辟容量為防止需多次擴(kuò)容降低效率。
/// @brief 構(gòu)造 n 個(gè)值為 val 的對(duì)象
/// @param n 容器的大小
/// @param val 用來(lái)初始化容器元素的值
vector(size_t n, const T& val = T()) {
_start = nullptr;
_finish = nullptr;
_end_of_storage = nullptr;
reserve(n);
for (int i = 0; i < n; ++i) {
push_back(val);
}
}
一段區(qū)間構(gòu)造
vector 支持使用一段迭代器區(qū)間構(gòu)造,區(qū)間范圍是 [first, last),這里的迭代器不一定要是 vector 的迭代器,只有是具有輸入功能的迭代器都可以。
/// @brief 用給定的迭代器區(qū)間初始化,STL 中的區(qū)間均為左閉右開(kāi)形式,即 [first, last)
/// @tparam InputIterator 所需最低階的迭代器類(lèi)型,具有輸入功能的迭代器都可以
/// @param first 迭代器起始位置
/// @param last 迭代器終止位置
template< class InputIterator>
vector(InputIterator first, InputIterator last) {
_start = nullptr;
_finish = nullptr;
_end_of_storage = nullptr;
// 和上一個(gè)類(lèi)似,先開(kāi)辟空間,尾減頭即為要開(kāi)辟的個(gè)數(shù)
reserve(last - first);
while (first != last) {
push_back(*first);
++first;
}
}
析構(gòu)函數(shù)
析構(gòu)函數(shù)的實(shí)現(xiàn)很簡(jiǎn)單,首先檢查容器是否為空,不為空就釋放空間,再把指針置空即可。
注意:因?yàn)槲覀冮_(kāi)辟了連續(xù)的空間,要使用 delete[] 來(lái)釋放空間,對(duì)應(yīng)的也要使用 new[] 來(lái)開(kāi)辟空間。即使我們只開(kāi)辟一個(gè)空間也不能使用 new,否則對(duì)自定義類(lèi)型在釋放時(shí)程序會(huì)崩潰。
具體原因請(qǐng)看:new 和 delete 為什么要匹配使用
/// @brief 釋放開(kāi)辟的空間
~vector() {
if (_start != nullptr) {
// 釋放開(kāi)辟的空間,從此處可以看出,開(kāi)辟空間一定要用 new[]
// 否則對(duì)于自定義類(lèi)型程序?qū)?huì)崩潰
delete[] _start;
_start = nullptr;
_finish = nullptr;
_end_of_storage = nullptr;
}
}
拷貝構(gòu)造函數(shù)
下面給出的實(shí)現(xiàn)方法比較簡(jiǎn)單,直接用容器 v 初始化創(chuàng)建一個(gè)臨時(shí)容器 tmp,再交換 tmp 和 this 指針指向就好了。
此時(shí) this 指向的容器是由 v 初始化出來(lái)的,tmp 指向了一個(gè)全空的容器,tmp 出了作用域就銷(xiāo)毀了。
需要注意的是此處一定要先將指針初始化為空,否則交換給 tmp 的指針要是不為空而是隨機(jī)值的話(huà),tmp 銷(xiāo)毀時(shí)調(diào)用析構(gòu)函數(shù)就會(huì)導(dǎo)致程序崩潰。有些 ide(vs 2022) 可能會(huì)自動(dòng)賦空,dev 下就不會(huì),對(duì)指針類(lèi)型編譯器本來(lái)就不會(huì)處理。
/// @brief 用給定容器初始化
/// @param v 用來(lái)初始化的容器
vector(const vector<T>& v) {
_start = nullptr;
_finish = nullptr;
_end_of_storage = nullptr;
vector<T> tmp(v.begin(), v.end());
swap(tmp);
}
復(fù)制賦值函數(shù)
該函數(shù)與拷貝構(gòu)造函數(shù)類(lèi)似,不同的是這里指針不應(yīng)賦空,否則指針原本指針指向的東西就無(wú)法正確釋放了。
/// @brief 替換容器內(nèi)容
/// @param v 用作數(shù)據(jù)源的另一容器
/// @return *this
vector<T>& operator=(const vector<T>& v) {
vector<T> tmp(v);
swap(tmp);
// 返回值為對(duì)象的引用,為的是可以連續(xù)賦值
return *this;
}
vector 的迭代器
vector 維護(hù)的是一個(gè)連續(xù)線性空間,不論元素是什么類(lèi)型,普通指針都可以作為 vector 的迭代器滿(mǎn)足所有的條件,因此元素類(lèi)型的指針就是 vector 的迭代器。
typedef T* iterator; typedef const T* const_iterator;
begin 和 end
begin 和 end 獲取的是正向迭代器,begin 指向第一個(gè)元素,end 指向最后一個(gè)元素的下一個(gè)位置,begin++ 是向后即 end 方向移動(dòng)。
對(duì)應(yīng)的還有反向迭代器 rbegin 和 rend,rbegin 指向最后一個(gè)元素,rend 指向第一個(gè)元素的前一個(gè)位置,rbegin++ 是向前即 rend 方向移動(dòng)。兩者對(duì)應(yīng)如下圖,因?yàn)榉聪虻鲝?fù)雜的多,這里就不實(shí)現(xiàn)了。

/// @brief 返回指向 vector 首元素的迭代器
/// @return 指向首元素的迭代器
iterator begin() {
return _start;
}
// const 版本供 const 容器使用
const_iterator begin() const {
return _start;
}
/// @brief 返回指向 vector 最后元素后一元素的迭代器
/// @return 指向最后元素下一個(gè)位置的迭代器
iterator end() {
return _finish;
}
const_iterator end() const {
return _finish;
}
元素訪問(wèn)
operator[]
vector 也支持向數(shù)組一樣的 [] 訪問(wèn)方式,可以隨機(jī)讀取每個(gè)位置,返回該位置元素的引用。
需要注意的是,該函數(shù)并不做邊界檢查,需程序員自行檢查。
/// @brief 返回位于指定位置 pos 的元素的引用,不進(jìn)行邊界檢查
/// @param pos 要返回的元素的位置
/// @return 到所需元素的引用
reference operator[](size_t pos) {
return _start[pos];
}
// 與上面的唯一不同就是用于 const 容器
const reference operator[](size_t pos) const {
return _start[pos];
}
back
back 可以獲取最后一個(gè)元素的引用。
/// @brief 返回到容器中最后一個(gè)元素的引用
/// @return 最后元素的引用
reference back() {
return *(end() - 1);
}
const reference back() const {
return *(end() - 1);
}
容量相關(guān)函數(shù)

size
根據(jù)圖可知 _finish - _start 即為 size。
/// @brief 返回容器中的元素?cái)?shù)
/// @return 容器中的元素?cái)?shù)量
size_t size() const {
return _finish - _start;
}
capacity
由圖可知 _end_of_storage - _start 即為 capacity。
/// @brief 返回容器當(dāng)前已為之分配空間的元素?cái)?shù)
/// @return 當(dāng)前分配存儲(chǔ)的容量
size_t capacity() const {
return _end_of_storage - _start;
}
empty
檢查是否為空的方法很簡(jiǎn)單,直接比較 _start 是否 _finish 相等即可,相等就為空,返回 true。
/// @brief 檢查容器是否無(wú)元素
/// @return 若容器為空則為 true,否則為 false
bool empty() const {
return _start == _finish;
}
resize
重設(shè)容器大小以容納 cnt 個(gè)元素。
如果當(dāng)前大小大于 cnt,那么減小容器到它的開(kāi)頭 cnt 個(gè)元素。
如果當(dāng)前大小小于 cnt,那么就調(diào)用 insert 在最后插入 cnt - size() 個(gè) val。
一定要注意,不能只改變 _finish,呢樣會(huì)因沒(méi)調(diào)用析構(gòu)函數(shù)從而引發(fā)內(nèi)存泄漏。你可能會(huì)想,我們會(huì)在最后容器銷(xiāo)毀的時(shí)候調(diào)用它的析構(gòu)函數(shù),它的析構(gòu)函數(shù)中有 delete[],這個(gè)語(yǔ)句會(huì)調(diào)用數(shù)據(jù)的析構(gòu)函數(shù)不會(huì)引起內(nèi)存泄漏。這樣想有一定的道理,但有沒(méi)有可能我們 vector 中一開(kāi)始有 10 個(gè)元素,我們用 resize 將其大小改變?yōu)?5,再調(diào)用 5 次 insert 將其大小變?yōu)?10,最后對(duì)象銷(xiāo)毀調(diào)用析構(gòu)。

如上圖,我們一開(kāi)始有 10 個(gè) int* 類(lèi)型的元素,分別指向一塊空間,后面 resize 為 5 后又添加了 5 個(gè)新的數(shù)據(jù)(用藍(lán)色標(biāo)識(shí))。當(dāng)我們析構(gòu)的時(shí)候我們會(huì)析構(gòu)下圖中的元素,釋放它們指向的空間,那上圖中的 6、7、8、9、10 呢,沒(méi)辦法,因?yàn)槲覀円呀?jīng)找不到它們的指針了,也就沒(méi)辦法釋放它們的空間了。
/// @brief 重設(shè)容器大小以容納 cot 個(gè)元素
/// @param cnt 容器的大小
/// @param val 用以初始化新元素的值
void resize(size_t cnt, T val = T()) {
if (cnt < size()) {
// 新大小小于原來(lái)的,需要將多余的部分刪除掉
// 不能只改變 _finish 指向,要使用 erase 來(lái)刪除,以便調(diào)用析構(gòu)函數(shù)
erase(begin() + cnt, end());
} else {
// 新空間更大,直接調(diào)用 insert 插入即可
for (int i = 0; i < cnt - size(); ++i) {
insert(end(), val);
}
}
}
reserve
reserve 用來(lái)預(yù)留存儲(chǔ)空間,如果要 push_back 大量的數(shù)據(jù),可能會(huì)引起多次空間分配,從而多次轉(zhuǎn)移元素浪費(fèi)大量時(shí)間??梢灶A(yù)先開(kāi)辟足夠的空間,減少空間分配的次數(shù),來(lái)提高效率。
注意:
1.若 cap 的值大于當(dāng)前的 capacity,則分配新存儲(chǔ),否則不做任何事,也就是說(shuō) reserve 不會(huì)縮小容量
為什么一定要分配新存儲(chǔ),而不是在原空間之后接一部分新空間(因?yàn)闊o(wú)法保證原空間之后還有足夠的可用空間)
2.同時(shí),capacity 的改變也不會(huì)影響 size 的大小
reserve 擴(kuò)容的思路也比較簡(jiǎn)單,開(kāi)辟一段新空間,將原數(shù)據(jù)拷貝到新空間,釋放舊空間即可。
需要注意的是,一定要在開(kāi)始記錄之前元素的個(gè)數(shù),因?yàn)?_finish 還指向原空間最后一個(gè)有效數(shù)據(jù)的下一個(gè)位置,需要將其更新指向新空間。
/// @brief 增加 vector 的容量到大于或等于 cap 的值
/// @param cap vector 的新容量
void reserve(size_t cap) {
size_t len = size();
if (cap > capacity()) {
T* tmp = new T[cap];
if (_start != nullptr) {
// 如果容器內(nèi)之前有數(shù)據(jù),將數(shù)據(jù)拷貝到新位置
for (int i = 0; i < size(); ++i) {
tmp[i] = _start[i];
}
delete[] _start; // 釋放掉舊的空間
}
_start = tmp;
}
// 指向新的地址空間
_finish = _start + len;
_end_of_storage = _start + cap;
}修改函數(shù)
insert
insert 函數(shù)的功能很多,可以在指定位置插入一個(gè)或多個(gè)值,也可以插入一段區(qū)間。這里我們只實(shí)現(xiàn)插入一個(gè)值的函數(shù),在 pos 前面插入 val。
插入之前首先要檢查容量,不夠就進(jìn)行擴(kuò)容。然后將插入位置之后的數(shù)據(jù)向后挪動(dòng)一個(gè)位置,插入 val 即可。
需要注意的是,擴(kuò)容的話(huà)需要提前記錄 pos 之前的元素個(gè)數(shù)。因?yàn)?pos 指向的是之前空間的某個(gè)位置,要將其更新為新空間的地址。
/// @brief 將 val 插入到 pos 迭代器之前
/// @param pos 將內(nèi)容插入到它前面的迭代器
/// @param val 要插入的元素值
/// @return 指向被插入 val 的迭代器
iterator insert(iterator pos, const T& val) {
// 檢查參數(shù)是否在合法返回,assert 只在 degug 版本下有效
assert(pos >= _start && pos <= _finish);
if (_finish == _end_of_storage) {
// 首先檢查容量,空間不夠要進(jìn)行擴(kuò)容
// 先記錄插入位置之前元素個(gè)數(shù)
size_t len = pos - _start;
// 第一次開(kāi)辟空間給 10 個(gè),后續(xù)擴(kuò)容為 2 倍
size_t newCap = capacity() == 0 ? 10 : capacity() * 2;
reserve(newCap);
// 更新 pos 在新空間中的位置
pos = _start + len;
}
// 將插入位置之后的所有數(shù)據(jù)向后挪動(dòng)一個(gè)位置
iterator end = _finish - 1;
while (end >= pos) {
*(end + 1) = *end;
--end;
}
*pos = val;
++_finish;
return pos;
}
push_back
push_back 是向容器的最后添加一個(gè)元素,直接調(diào)用 insert 即可。
/// @brief 添加給定元素 val 到容器尾
/// @param val 要添加的元素值
void push_back(const T& val) {
// 直接調(diào)用 insert 即可
insert(end(), val);
}
erase
erase 從容器中刪除指定的元素:
1.移除位于 pos 的元素
2.移除范圍 [first, last) 中的元素
移除 pos 位置的元素方法很簡(jiǎn)單,首先調(diào)用該位置的析構(gòu)函數(shù),然后將后面的數(shù)據(jù)向前移動(dòng)一個(gè)位置,最后的 --_finish 就可以了。
/// @brief 從容器擦除指定位置的元素
/// @param pos 指向要移除的元素的迭代器
/// @return 移除元素之后的迭代器
iterator erase(iterator pos) {
assert(pos >= _start && pos < _finish);
// 在 pos 位置調(diào)用析構(gòu)函數(shù),釋放資源
pos->~T();
// 將 pos 位置之后的的元素都向前移動(dòng)一個(gè)位置
iterator it = pos + 1;
while (it != _finish) {
*(it - 1) = *it;
++it;
}
--_finish;
// 此時(shí)的 pos 指向沒(méi)刪除前 pos 位置下一個(gè)元素
return pos;
}范圍刪除同樣也很簡(jiǎn)單,首先要計(jì)算出要?jiǎng)h除的個(gè)數(shù),循環(huán)調(diào)用 erase 刪除就可以了。
注:下面這種循環(huán)調(diào)用的方式效率十分低,庫(kù)函數(shù)并沒(méi)有使用這種方法,庫(kù)函數(shù)首先對(duì)要?jiǎng)h除的范圍調(diào)用析構(gòu)函數(shù),然后將區(qū)間后面的數(shù)據(jù)移到前面。這樣就只會(huì)移動(dòng)一次數(shù)據(jù),不向下面需要移動(dòng) cnt 次。
/// @brief 移除一段范圍的元素
/// @param first 要移除的起始范圍
/// @param last 要移除的結(jié)束返回
/// @return 移除的最后一個(gè)元素之后的迭代器
iterator erase(iterator first, iterator last) {
int cnt = last - first;
while (cnt--) {
first = erase(first);
}
return first;
}
pop_back
pop_back 用來(lái)刪除容器的最后一個(gè)元素,直接調(diào)用 erase 刪除就行。
/// @brief 移除容器最后一個(gè)元素
void pop_back() {
erase(_finish - 1);
}
swap
swap 函數(shù)可以用來(lái)交換兩個(gè)容器的內(nèi)容,不過(guò)不用實(shí)際交換數(shù)據(jù),只需要改變兩個(gè)容器指針的指向即可。
/// @brief 交換 this 指向的容器和 v 的內(nèi)容
/// @param v 要交換內(nèi)容的容器
void swap(vector<T>& v) {
// 直接交換所有指針即可
std::swap(_start, v._start);
std::swap(_finish, v._finish);
std::swap(_end_of_storage, v._end_of_storage);
}到此這篇關(guān)于詳解C++ STL模擬實(shí)現(xiàn)vector的文章就介紹到這了,更多相關(guān)C++ STL vector內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
C++中fstream,ifstream及ofstream用法淺析
這篇文章主要介紹了C++中fstream,ifstream及ofstream用法,適合C++初學(xué)者學(xué)習(xí)文件流的操作,需要的朋友可以參考下2014-08-08
通過(guò)先序遍歷和中序遍歷后的序列還原二叉樹(shù)(實(shí)現(xiàn)方法)
下面小編就為大家?guī)?lái)一篇通過(guò)先序遍歷和中序遍歷后的序列還原二叉樹(shù)(實(shí)現(xiàn)方法)。小編覺(jué)得挺不錯(cuò)的,現(xiàn)在就分享給大家,也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2017-06-06
c語(yǔ)言設(shè)計(jì)模式之單例模式中的餓漢與懶漢詳解
這篇文章主要介紹了c語(yǔ)言設(shè)計(jì)模式之單例模式中的餓漢與懶漢詳解,單例模式是指一個(gè)類(lèi)只能創(chuàng)建一個(gè)對(duì)象,保證系統(tǒng)中該類(lèi)只有一個(gè)實(shí)例,并提供一個(gè)可供訪問(wèn)的全局訪問(wèn)點(diǎn),該實(shí)例被所有程序模塊共享,需要的朋友可以參考下2023-08-08
c/c++?Error:?redefinition?of?'xxx'的問(wèn)題及解決方法
兩個(gè)類(lèi)/文件同時(shí)引用定義ReplyInfo的頭文件,會(huì)造成頭文件中定義重復(fù)定義,本文給大家分享c/c++?Error:?redefinition?of?‘xxx’?的問(wèn)題及解決方法,感興趣的朋友一起看看吧2023-08-08

