迭代器是我们遍历容器的利器没在后面还会引入各种迭代器及其他作用
我们先来实现一个简单的迭代器
//迭代器一般实现为容器的嵌套类型
//将这个类插入到之前写好的vector容器中
class iterator
{
public:
friend class vector<T, Alloc>;//声明称友元类
iterator(vector<T,Alloc>*pvec,T *ptr = nullptr)
:_pVec(pvec), _ptr(ptr)
{
_pVec->_head._next = itb;
}
bool operator!=(const iterator &it)const
{
if (_pVec == nullptr || _pVec != it._pVec)
{
throw"iterator incompatable";
}
return _ptr != it._ptr;
}
void operator++()
{
if (_pVec == nullptr)
{
throw"iterator invalid";
}
_ptr++;
}
T& operator*()
{
if (_pVec == nullptr)
{
throw"iterator invalid";
}
return *_ptr;
}
const T& operator*()const { return *_ptr; }
private:
T *_ptr;
vector<T, Alloc> *_pVec;//声明指向哪个容器
};
iterator begin() { return iterator(this, _first); }
iterator end() { return iterator(this, _last); }
int main()
{
Test t1, t2, t3;
cout << "-------------------" << endl;
vector<Test>vec;
vec.push_back(t1);
vec.push_back(t2);
vec.push_back(t3);
cout << "-------------------" << endl;
vec.pop_back();
vector<int>a;
for (int i = 0;i < 20;++i)
{
a.push_back(rand() % 100 + 1);
}
vector<int>::iterator it = a.begin();
for (;it != a.end();++it)
{
cout << *it << " ";
}
return 0;
}
一、迭代器失效
当我们定义一种特殊的环境的时候,比如删除容器中所有元素,或者给容器添加几个连续的元素,当我们用for循环利用迭代器++进行删除/添加的时候,我们会发现程序异常中断了,因为我们第一次调用earse之后迭代器就失效了,对一个已经失效的迭代器进行++操作的时候,就会产生错误。但是我们在插入或者删除点进行it自减操作的时候仍然可以继续利用迭代器继续访问容器元素。
下面我们给出迭代器失效的规则:
当容器进行插入和删除的时候,从当前插入或者删除位置到容器末尾所有元素的迭代器全部失效。如果是插入引起的元素扩容,则所有元素迭代器全部失效。
二、迭代器失效的解决
以vector容器来举例吧,在easer或者insert之后函数都会返回一个指向插入点或者删除点的迭代器,用it对其进行接收
vector<int> vec;
for (int i = 0; i < 20; i++)
{
vec.push_back(rand() % 100 + 1);
}
auto it = vec.begin();
while (it != vec.end())
{
if (*it % 2 == 0)
{
it = vec.erase(it);//对迭代器进行更新
}
else
{
++it;
}
}
当我们运行上述代码的时候,可以正常的插入和删除
三、迭代器失效底层原理
在进行之前,我们要在原有代码的基础上加上一个成员变量vector<T, Alloc>* _pVec,表明当前迭代器迭代的是哪个对象,防止不同容器之间的迭代器进行比较(不同容器的迭代器之间是不可以进行比较的)
因此各个函数都要进行一下判断的改变
bool operator!=(const iterator& it)const
{
if (_pVec == nullptr || _pVec != it._pVec)
{
throw"ierator incompatable!";
return false;
}
return _ptr != it._ptr;
}
void operator++()
{
//检查迭代器的有效性
if (_pVec == nullptr)
{
throw"iterator invalid";
}
_ptr++;
}
T& operator*()
{
//检查迭代器的有效性
if (_pVec == nullptr)
{
throw"iterator invalid";
}
return *_ptr;
}
Iterator_Base结构
struct Iterator_Base
{
//用链表获取元素的迭代器
Iterator_Base(iterator *c = nullptr, Iterator_Base *n = nullptr)
:_cur(c), _next(n) {}
iterator *_cur;//指向迭代器的指针
Iterator_Base *_next;//指向下一个Iterator_Base节点的地址
};
构造函数如下:
iterator(vector<T,Alloc>*pvec,T *ptr = nullptr)
:_pVec(pvec), _ptr(ptr)
{
Iterator_Base *itb = new Iterator_Base(this, _pVec->_head._next);
_pVec->_head._next = itb;
}
verfiy函数(检查迭代器是否有效)
void verify(T *first,T *last)
{
Iterator_Base* pre = &this->_head;
Iterator_Base* it = this->_head._next;
while (it != nullptr)
{
if (it->_cur->_ptr >= first&& it->_cur->_ptr <= last)
{
//迭代器失效,把itreator持有的容器指针置nullptr
it->_cur->_pVec = nullptr;
//删除当前迭代器节点,继续判断后面的迭代器节点是否失效
pre->_next = it->_next;
delete it;
it = pre->_next;
}
else
{
pre = it;
it = it->_next;
}
完整代码如下
template<typename T>
struct Allocator
{
T* allocate(size_t size)//内存开辟
{
set_new_handler(0);
return (T*)malloc(sizeof(T)*size);
}
void deallocate(void *p)//内存释放
{
free(p);
}
void construct(T *p, const T &val)//负责对象构造
{
new(p) T(val);//定位new
}
void destroy(T *p)//负责对象析构
{
p->~T(); //~T()代表T类型的析构函数
}
};
template<typename T,typename Alloc = Allocator<T>>
class vector
{
public:
vector(int size = 10)
{
//需要把给内存开辟和对象构造分开处理
//_first = new T[size];//new 即使没有赋值 也会开辟空间
_first = _allocator.allocate(size);
_last = _first;
_end = _first + size;
}
~vector()
{
//析构容器的有效元素,然后释放first指针指向的堆内存
//delete[]_first;
for (T * p= _first;p != _last; ++p)
{
_allocator.destroy(p);//把first指针指向的数组的有效元素进行析构操作
}
_allocator.deallocate(_first);//释放堆上的数组内存
_first = _last = _end = nullptr;
}
vector(const vector<T> &swq)
{
int size = swq._end - swq._first;
//_first = new T[size];
_first = _allocator.allocate(size);
int len = swq._last - swq._first;
for (int i = 0;i < len;i++)
{
//_first[i] = swq._first[i];
_allocator.construct(_first + i, swq._first[i]);
}
_last = _first + len;
_end = _first + size;
}
vector<T>& operator = (const vector<T> &swq)
{
if (this = &swq)
{
return *this;
}
//delete[]_first;
for (T * p= _first;p != _last; ++p)
{
_allocator.destroy(p);//把first指针指向的数组的有效元素进行析构操作
}
_allocator.deallocate(_first);//释放堆上的数组内存
int size = swq._end - swq._fiest;
//_first = new T[size];
_first = _allocator.allocate(size);
int len = swq._last - swq._first;
for (int i = 0;i < len;i++)
{
//_first[i] = swq._first[i];
_allocator.construct(_first + i, swq._first[i]);
}
_last = _first + len;
_end = _first + size;
return *this;
}
void push_back(const T &val) //向容器末尾添加元素
{
if (full())
expand();
//*_last++ = val; _last指针指向的内存构造一个值为val的对象
_allocator.construct(_last, val);
_last++;
}
void pop_back()//从容器末尾删除元素
{
if (empty())
return;
//--_last; 不仅要把——last指针-- 还需要析构删除的元素
verify(_last - 1, _last);
--_last;
_allocator.destroy(_last);
}
T back()const//返回容器末尾元素值
{
return *(_last - 1);
}
bool full()const
{
return _last == _end;
}
bool empty()const
{
return _first == _last;
}
int size()
{
return _end - _first;
}
T& operator[](int index)
{
if (index < 0 || index >= size())
{
throw "out of size";
}
return _first[index];
}
//迭代器一般实现为容器的嵌套类型
class iterator
{
public:
friend class vector<T, Alloc>;
iterator(vector<T,Alloc>*pvec,T *ptr = nullptr)
:_pVec(pvec), _ptr(ptr)
{
Iterator_Base *itb = new Iterator_Base(this, _pVec->_head._next);
_pVec->_head._next = itb;
}
bool operator!=(const iterator &it)const
{
if (_pVec == nullptr || _pVec != it._pVec)
{
throw"iterator incompatable";
}
return _ptr != it._ptr;
}
void operator++()
{
if (_pVec == nullptr)
{
throw"iterator invalid";
}
_ptr++;
}
T& operator*()
{
if (_pVec == nullptr)
{
throw"iterator invalid";
}
return *_ptr;
}
const T& operator*()const { return *_ptr; }
private:
T *_ptr;
vector<T, Alloc> *_pVec;
};
iterator begin() { return iterator(this,_first); }
iterator end() { return iterator(this,_last); }
//检查迭代器失效
void verify(T *first, T *last)
{
Iterator_Base *pre = this->_head;
Iterator_Base *it = this->_head._next;
while (it != nullptr)
{
//迭代器失效,把iterator持有的容器指针置nullptr
if (it->_cur->_ptr > first && it->_cur->_ptr <= last)
{
//删除当前迭代器节点,继续判断后续的迭代器节点是否失效
pre->_next = it->_next;
delete it;
it = pre->_next;
}
else
{
pre = it;
it = it->_next;
}
}
}
iterator insert(iterator it, const T &val)
{
verify(it._ptr, _last);
T *p = _last;
while (p > it.ptr)
{
_allocator.construct(p, *(p - 1));
_allocator.destroy(p - 1);
p--:
}
_allocator.construct(p, val);
_last++;
return iterator(this, p);
}
iterator erase(iterator it)
{
verify(it._ptr - 1, _last);
T *p = it._ptr;
while (p < _last - 1)
{
_allocator.destroy(p);
_allocator.construct(p, *(p + 1));
p++;
}
_allocator.destroy(p);
_last--;
return iterator(this, it._ptr);
}
private:
T *_first;//指向数组起始的位置
T *_last; //指向数组中有效元素的后继位置
T *_end; //指向数组空间的后继位置
Alloc _allocator;//定义容器的空间配置器对象
//容器迭代器失效增加代码
struct Iterator_Base
{
//用链表获取元素的迭代器
Iterator_Base(iterator *c = nullptr, Iterator_Base *n = nullptr)
:_cur(c), _next(n) {}
iterator *_cur;//指向迭代器的指针
Iterator_Base *_next;//指向下一个Iterator_Base节点的地址
};
Iterator_Base _head;
void expand()//容器的二倍扩容接口
{
int size = _end - _first;
//T *ptmp = new T[2 * size];
T *ptmp = _allocator.allocate(2 * size);
for (int i = 0;i < size;i++)
{
_allocator.construct(ptmp + i, _first[i]);
//ptmp[i] = _first[i];
}
//delete[]_first;
for (T *p = _first; p != _last; ++p)
{
_allocator.destroy(p);
}
_allocator.deallocate(_first);
_first = ptmp;
_last = _first + size;
_end = _first + 2 * size;
}
};