编写自己的迭代器

自定义迭代器必须提供iterator_traits的五种特性，分别是迭代器类型、元素类型、距离类型、指针类型与reference类型。

这里我们继承标准库提供的iterator<>即可。

代码如下:

#ifndef ITERATOR_H
#define ITERATOR_H 
#include <iterator>

template <typename Container>
class Iterator : public std::iterator<std::output_iterator_tag, void, void, void, void>
{
public:
    explicit Iterator(Container &c)
        :_container(c)
    { }

    Iterator<Container> &operator= (const typename Container::value_type &value)
    {
        _container.insert(value);
        return *this;
    }

    Iterator<Container> operator* ()
    { return *this; }

    Iterator<Container> &operator++ ()
    { return *this; }

    Iterator<Container> &operator++ (int)
    { return *this; }

protected:
    Container &_container;
};

template <typename Container>
Iterator<Container> MyInsert(Container &c)
{
    return Iterator<Container>(c);
}

#endif  /*ITERATOR_H*/

注意=操作符执行了insert操作，所以当我们写下：

*iter = 3; 时，  自动将3插入容器中。

我们还提供了一个MyInsert函数，用于快速生成迭代器对象，于是我们可以这样使用：

MyInsert(vec) = 55;

我们只需改动代码中的

_container.insert(value);

如果改为push_back就变成了back_insert， 相反的， 改为push_front就变成了front_insert。

测试代码如下:

#include "iterator.hpp"
#include <set>
#include <string>
#include <iostream>
using namespace std;

template <typename CONT>
void print(const CONT &s)
{
    for(typename CONT::const_iterator it = s.begin();
        it != s.end();
        ++it)
    {
        cout << *it << " ";
    }
    cout << endl;
}

int main(int argc, char const *argv[])
{
    set<int> coll;

    Iterator<set<int> > iter(coll);

    *iter = 1;
    iter++;
    *iter = 2;
    iter++;
    *iter = 3;

    print(coll);

    MyInsert(coll) = 44;
    MyInsert(coll) = 55;


    print(coll);

    int vals[] = {33, 67, -4, 13, 5, 2};
    int size = sizeof(vals) / sizeof(vals[0]);
    copy(vals, vals + size, MyInsert(coll));

    print(coll);

    return 0;
}

测试结果为:

1 2 3 
1 2 3 44 55 
-4 1 2 3 5 13 33 44 55 67