上节我们实现了back_inserter和front_inserter,接下来是更为普通的插入迭代器,它允许用户指定插入位置。
实现代码如下:
#ifndef ITERATOR_HPP
#define ITERATOR_HPP template <typename Container>
class InsertIterator
{
public:
typedef typename Container::value_type value_type;
typedef typename Container::iterator iterator; InsertIterator(Container &cont, iterator iter) :cont_(cont), iter_(iter) { } InsertIterator<Container> &operator=(const value_type &val)
{
cont_.insert(iter_, val);
++iter_;
return *this;
} InsertIterator<Container> &operator*()
{
return *this;
} InsertIterator<Container> &operator++()
{
return *this;
}
InsertIterator<Container> &operator++(int)
{
return *this;
} private:
Container &cont_;
iterator iter_;
}; template <typename Container>
InsertIterator<Container> inserter(Container &c)
{
return InsertIterator<Container>(c);
} #endif //ITERATOR_HPP
可以看出,赋值操作使得内部存储的迭代器前移,而++操作同样什么都没有做。
或许我们想:可不可以在赋值操作符中不改变迭代器,而是到了++中改变?
答案是否定的。
设想,按照刚才设想的去实现,那么如果用户做了以下操作:
iter++;
iter++;
iter++;
*iter = ;
那么我们没法保证最后一行时,iter指向的位置是可以插入元素的。
那么上面的写法为什么可行?原因是执行insert操作时,有个特殊位置为end(),它指向最后一个元素的下一个位置,也就是第一个非法的位置。这也是唯一一个合法的非法位置。
按照上面源码的实现,仅仅在赋值时迭代器前移,iter++无实质操作,而赋值时前移,iter实际指向了新的end()位置,就保证了无论用户怎么执行++操作,都丝毫不会影响iter的有效性。
测试代码如下:
#include "Iterator.hpp"
#include <iostream>
#include <string>
#include <vector>
using namespace std; template <typename T>
void printElems(const T &t, const string &s = "")
{
cout << s << " ";
for(typename T::const_iterator it = t.begin();
it != t.end();
++it)
{
cout << *it << " ";
}
cout << endl;
} int main(int argc, char const *argv[])
{
vector<int> coll;
coll.push_back();
coll.push_back();
coll.push_back();
printElems(coll); inserter(coll, coll.begin()) = ;
inserter(coll, coll.begin()) = ; printElems(coll); inserter(coll, coll.end()) = ;
inserter(coll, coll.end()) = ; printElems(coll); return ;
}