【C++】模拟实现STL容器：vector

原创已于 2025-04-09 14:39:39 修改 · 457 阅读

3 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

收录于

C++

于 2024-11-04 19:39:12 首次发布

一、vector迭代器失效的问题

（一）什么是迭代器失效？

#include<iostream>
#include<vector>
int main()
{
	std::vector<int> v = { 1,2,3,4 };
	std::vector<int>::iterator it = v.begin();
	for (int i = 0; i < 5; ++i)
	{
		v.push_back(i);
	}
	std::cout << *it << std::endl;
	return 0;
}

以上代码在Visual Studio2022中会使编译器报错。原因是迭代器记录了 vector 的首元素位置，在持续的插入元素中，vector 会发生扩容，这时迭代器的值已不是有效值（类似于野指针），之后再使用迭代器便会引发编译器报错。

同样的，下段代码也会出现迭代器失效的情况。

#include<iostream>
#include<vector>
int main()
{
	std::vector<int> v = { 1,2,3,4,5 };
	std::vector<int>::iterator it = std::find(v.begin(), v.end(), 3);
	if (it != v.end())
	{
		v.erase(it);
	}
	++it;
	return 0;
}

当把 vector 中的元素3删除以后，迭代器原先记录的值为3的位置，但是之后元素3已经被删除，此时再操纵迭代器将会引发编译器报错。

其实在不同平台下，各个编译器对迭代器失效的问题的处理不同，例如上述代码在Linux的g++编译器（4.8.5版本）下运行相同debug版本的程序，编译器并未报错，迭代器仍能正常使用。

但为了我们所编辑的代码在各个平台下都能正常执行，因此我们认定 vector 在使用 insert() 和 erase() （空间发生变化）之后迭代器失效。

（二）解决方法

在使用 insert() 或者 erase() 之后，使用迭代器接受其返回值更新迭代器即可。

#include<iostream>
#include<vector>
int main()
{
	std::vector<int> v = { 1,2,3,4,5 };
	std::vector<int>::iterator it = std::find(v.begin(), v.end(), 3);
	if (it != v.end())
	{
		it = v.erase(it);
	}
	++it;
	return 0;
}

在插入元素时返回插入元素的迭代器，而删除元素时返回删除元素的后一个元素的迭代器。

二、模拟实现

#pragma once
#include<iostream>
#include <assert.h>
using namespace std;
namespace mh
{
    template<class T>
    class vector
    {
    public:
        // Vector的迭代器是一个原生指针
        typedef T* iterator;
        typedef const T* const_iterator;
        iterator begin()
        {
            return _start;
        }
        iterator end()
        {
            return _finish;
        }
        const_iterator cbegin() const
        {
            return _start;
        }
        const_iterator cend() const
        {
            return _finish;
        }
        
        // construct and destroy
        vector()
            :_start(nullptr)
            , _finish(nullptr)
            , _endOfStorage(nullptr)
        {}
        vector(int n, const T& value = T())
        {
            for (int i = 0; i < n; ++i)
            {
                push_back(value);
            }
        }

        template<class InputIterator>
        vector(InputIterator first, InputIterator last)
        {
            assert(first < last);
            for (InputIterator it = first; it != last; ++it)
            {
                push_back(*it);
            }
        }
        vector(const vector<T>& v)
            :_start(nullptr)
            ,_finish(nullptr)
            ,_endOfStorage(nullptr)
        {
            reserve(v.capacity());
            for (size_t i = 0; i < v.size(); ++i)
            {
                _start[i] = v[i];
            }
            _finish = _start + v.size();
            _endOfStorage = _start + v.capacity();
        }
        vector<T>& operator= (vector<T> v)
        {
            swap(v);
            return *this;
        }
        ~vector()
        {
            delete[] _start;
            _start = _finish = _endOfStorage = nullptr;
        }

         // capacity
        size_t size() const
        {
            return _finish - _start;
        }
        size_t capacity() const
        {
            return _endOfStorage - _start;
        }
        void reserve(size_t n)
        {
            if (n > capacity())
            {
                size_t oldSize = size();
                iterator tmp = new T[n];
                for (size_t i = 0; i < size(); ++i)
                {
                    tmp[i] = _start[i];
                }
                delete _start;
                _start = tmp;
                _finish = _start + oldSize;
                _endOfStorage = _start + n;
            }
        }
        void resize(size_t n, const T& value = T())
        {
            if (n <= size())
            {
                _finish = _start + n;
            }
            else if(size() < n && n <= capacity())
            {
                for (size_t i = size(); i < n; ++i)
                {
                    _start[i] = value;
                }
                _finish = _start + n;
            }
            else
            {
                size_t oldSize = size();
                reserve(n);
                for (size_t i = oldSize; i < n; ++i)
                {
                    _start[i] = value;
                }
                _finish = _start + n;
            }
        }

        ///////////////access///////////////////////////////
        T& operator[](size_t pos)
        {
            assert(pos < size());
            return _start[pos];
        }
        const T& operator[](size_t pos)const
        {
            assert(pos < size());
            return _start[pos];
        }

        ///////////////modify/////////////////////////////
        void push_back(const T& x)
        {
            if (_finish == _endOfStorage)
            {
                size_t newCapacity = capacity() == 0 ? 4 : 2 * capacity();
                reserve(newCapacity);
            }
            *_finish = x;
            ++_finish;
        }
        void pop_back()
        {
            if(_finish > _start)
                --_finish;
        }
        void swap(vector<T>& v)
        {
            std::swap(_start, v._start);
            std::swap(_finish, v._finish);
            std::swap(_endOfStorage, v._endOfStorage);
        }
        iterator insert(iterator pos, const T& x)
        {
            assert(_start <= pos);
            assert(pos <= _finish);
            if (_finish == _endOfStorage)
            {
                size_t newCapacity = capacity() == 0 ? 4 : 2 * capacity();
                reserve(newCapacity);
            }
            iterator end = _finish;
            while (end > pos)
            {
                *end = *(end - 1);
                --end;
            }
            *pos = x;
            ++_finish;
            return pos;
        }
        iterator erase(iterator pos)
        {
            assert(_start <= pos);
            assert(pos < _finish);
            iterator end = pos + 1;
            while (end != _finish)
            {
                *(end - 1) = *end;
                ++end;
            }
            --_finish;
            return pos;
        }
    private:
        iterator _start; // 指向数据块的开始
        iterator _finish; // 指向有效数据的尾
        iterator _endOfStorage; // 指向存储容量的尾
    };

}

标签

#c++ #开发语言