浅谈如何实现自定义的|浅谈如何实现自定义的 iterator

实现你自己的迭代器 使用 std::iterator
在 C++17 之前,实现自定义的迭代器被推荐采用从 std::iterator 派生的方式。
std::iterator 的基本定义 Std::iterator 具有这样的定义:

template< class Category, class T, class Distance = std::ptrdiff_t, class Pointer = T*, class Reference = T& > struct iterator;

其中,T 是你的容器类类型,无需多提。而 Category 是必须首先指定的所谓的 迭代器标签,参考 这里 。Category 主要可以是:
  1. input_iterator_tag:输入迭代器
  2. output_iterator_tag:输出迭代器
  3. forward_iterator_tag:前向迭代器
  4. bidirectional_iterator_tag:双向迭代器
  5. random_access_iterator_tag:随机访问迭代器
  6. contiguous_iterator_tag:连续迭代器
这些标签看起来似乎相当莫名其妙,仿佛我知道它们的用意,但实际上却又难以明白,难以挑选。
迭代器标签 下面粗略地对它们及其关联实体进行特性上的介绍,以帮助你理解。
这些 tags 实际上绑定关联着一些同名实体类如 input_iterator 等等,通过模板特化技术分别实现专有的 distance() 和 advance() ,以达到特定的迭代优化效果。
input_iterator_tag input_iterator_tag 可以包装函数的输出——以用作它人的输入流。所以它是仅可递增的(只能 +1),你不能对它 +n,只能通过循环 n 次递增来模拟相应的效果。input_iterator 无法递减(-1),因为输入流没有这样的特性。它的迭代器值(*it)是只读的,你不能对其置值。
但 output_iterator_tag,forward_iterator_tag 的迭代器值是可读写的。可读写的迭代器值是指:
std::list l{1,2,3}; auto it = l.begin(); ++it; (*it) = 5; // <- set value back into the container pointed by iterator

input_iterator 将容器呈现为一个输入流,你可以通过 input_iterator 接收输入数据流。
output_iterator_tag output_iterator_tag 很少被用户直接使用,它通常和 back_insert_iterator/ front_insert_iterator/ insert_iterator 以及 ostream_iterator 等配合使用。
output_iterator 没有 ++/-- 能力。你可以向 output_iterator 指向的容器中写入/置入新值,仅此而已。
如果你有输出流样式的呈现需求,可以选择它。
forward_iterator_tag forward_iterator_tag 表示前向迭代器,所以只能增量,不能回退,它继承 input_iterator_tag 的一切基本能力,但又有所增强,例如允许设置值。
从能力上说,input_iterator 支持读取/设置值,也支持递增行走,不支持递减行走(需要模拟,低效),+n 需要用循环模拟故而低效,但如果你的容器只有这样的外露的需求,那么 forward_iterator_tag 就是最佳选择。
从理论上来说,支持 forward_iterator_tag 的迭代器必须至少实现 begin/end。
bidirectional_iterator_tag bidirectional_iterator_tag 的关联实体 bidirectional_iterator 是双向可行走的,既可以 it++ 也可以 it--,例如 std::list。如同 forward_iterator_tag 一样,bidirectional_iterator_tag 不能直接 +n (和 -n),所以 +n 需要一个特化的 advance 函数来循环 n 次,每次 +1(即通过循环 n 次递增或递减来模拟)。
从理论上来说,支持 bidirectional_iterator_tag 的迭代器必须同时实现 begin/end 以及 rbegin/rend。
random_access_iterator_tag random_access_iterator_tag 表示的随机访问迭代器,random_access_iterator 支持读取/设置值,支持递增递减,支持 +n/-n。
由于 random_access_iterator 支持高效的 +n/-n,这也意味着它允许高效的直接定位,这种迭代器的所属容器,通常也顺便支持 operator [] 下标存取,如同 std::vector 那样。
contiguous_iterator_tag contiguous_iterator_tag 在 C++17 中开始引入,但是编译器们的支持力度有问题,所以目前我们不能对其进行详细介绍,对于实作来说不必考虑它的存在。
自定义迭代器的实现 一个定制迭代器需要选择一个迭代器标签,也就是选择迭代器的支持能力集合。下面是一个示例:
namespace customized_iterators { template class Range { public: // member typedefs provided through inheriting from std::iterator class iterator : public std::iterator { long num = FROM; public: iterator(long _num = 0) : num(_num) {} iterator &operator++() { num = TO >= FROM ? num + 1 : num - 1; return *this; } iterator operator++(int) { iterator ret_val = *this; ++(*this); return ret_val; } bool operator==(iterator other) const { return num == other.num; } bool operator!=(iterator other) const { return !(*this == other); } long operator*() { return num; } }; iterator begin() { return FROM; } iterator end() { return TO >= FROM ? TO + 1 : TO - 1; } }; void test_range() { Range<5, 13> r; for (auto v : r) std::cout << v << ','; std::cout << '\n'; }}

这个示例的原型来自于 cppreference 上 std::iterator 及其原作者,略有修改。
自增自减运算符重载 专门独立一个小节,因为发现垃圾教程太多了。
自增自减的运算符重载分为前缀后缀两种形式,前缀方式返回引用,后缀方式返回新副本:
struct X { // 前缀自增 X& operator++() { // 实际上的自增在此进行 return *this; // 以引用返回新值 }// 后缀自增 X operator++(int) { X old = *this; // 复制旧值 operator++(); // 前缀自增 return old; // 返回旧值 }// 前缀自减 X& operator--() { // 实际上的自减在此进行 return *this; // 以引用返回新值 }// 后缀自减 X operator--(int) { X old = *this; // 复制旧值 operator--(); // 前缀自减 return old; // 返回旧值 } };

或者去查看 cppreference 的 文档 以及 文档,别去看那些教程了,找不出两个正确的。
正确的编码是实现一个前缀重载,然后基于它实现后缀重载:
struct incr { int val{}; incr &operator++() { val++; return *this; } incr operator++(int d) { incr ret_val = *this; ++(*this); return ret_val; } };

如果有必要,你可能需要实现 operator= 或者 X(X const& o) 拷贝构造函数。但对于简单平凡 struct 来说可以省略(如果你不能确定自动内存拷贝是否被提供,考虑查看汇编代码,或者干脆显式实现 operator= 或者 X(X const& o) 拷贝构造函数)
C++17 起
但从 C++17 起 std::iterator 被弃用了。
如果你真的很关心流言飞语,可以去 这里 看看有关的讨论。
在多数情况下,你仍然可以使用 std::iterator 来简化代码编写,但这一特性以及早期的迭代器标签、类别等等概念已经过时。
完全手写迭代器 所以在从 C++17 开始的新时代,自定义迭代器原则上暂时只有手写。
namespace customized_iterators { namespace manually { template class Range { public: class iterator { long num = FROM; public: iterator(long _num = 0) : num(_num) {} iterator &operator++() { num = TO >= FROM ? num + 1 : num - 1; return *this; } iterator operator++(int) { iterator ret_val = *this; ++(*this); return ret_val; } bool operator==(iterator other) const { return num == other.num; } bool operator!=(iterator other) const { return !(*this == other); } long operator*() { return num; } // iterator traits using difference_type = long; using value_type = long; using pointer = const long *; using reference = const long &; using iterator_category = std::forward_iterator_tag; }; iterator begin() { return FROM; } iterator end() { return TO >= FROM ? TO + 1 : TO - 1; } }; } // namespace manuallyvoid test_range() { manually::Range<5, 13> r; for (auto v : r) std::cout << v << ','; std::cout << '\n'; }}

示例中的 iterator traits 部分不是必需的,你完全可以不必支持它们。
需要照顾到的事情 【浅谈如何实现自定义的|浅谈如何实现自定义的 iterator】完全手写迭代器的注意事项包括:
  1. begin() 和 end()
  2. 迭代器嵌入类(不必被限定为嵌入),至少实现:
    1. 递增运算符重载,以便行走
    2. 递减运算符重载,如果是双向行走(bidirectional_iterator_tag)或随机行走(random_access_iterator_tag)
    3. operator* 运算法重载,以便迭代器求值
    4. operator!= 运算符重载,以便计算迭代范围;必要时也可以显式重载 operator==(默认时编译器自动从 != 运算符上生成一个配套替代品)
如果你编码对迭代范围进行了支持,那么就可以使用 for 范围循环:
your_collection coll; for(auto &v: coll) { std::cout << v << '\n'; }

关于 for 范围循环的展开式,可以查看 这里。
C++20 之后
在 C++20 之后,迭代器发生了巨大的变化。但由于它的工程实作还早的很,所以本文中暂且不予讨论。
其它相关
除了 iterator 还有 const_iterator 为了代码规范和安全性,getter 通常一次提供两个,可写的和不可写的:
struct incr { int &val(){ return _val; } int const &val() const { return _val; } private: int _val{}; }

同样的道理,迭代器的 begin() 和 end() 也至少要提供 const 和 非 const 的两种版本。一般来说你可以通过独立实现来帮助提供多套版本:
struct XXX {// ... struct leveled_iter_data { //static leveled_iter_data begin(NodePtr root_) {...} //.static leveled_iter_data end(NodePtr root_) {...} // }using iterator = leveled_iter_data; using const_iterator = const iterator; iterator begin() { return iterator::begin(this); } const_iterator begin() const { return const_iterator::begin(this); } iterator end() { return iterator::end(this); } const_iterator end() const { return const_iterator::end(this); }}

这是不费脑子的一种方式,读写安全性被约束在 XXX 之内:owner 当然能够明白哪些应该可被暴露,哪些需要暂时约束暴露出来的能力。
除了 iterator 和 const_iterator 之外,rbegin/rend, cbegin/cend 等也可以考虑被实现。
注意事项:迭代器的使用 迭代器的使用一定要注意随用随取的准则。
void test_iter_invalidate() { std::vector vi{3, 7}; auto it = vi.begin(); it = vi.insert(it, 11); vi.insert(it, 5000, 23); vi.insert(it, 1, 31); // crach here! std::cout << (*it) << '\n'; return; }

在多数 OS 环境中,vi.insert(it, 5000, 23); 语句有极大概率导致 vector 不得不重新分配内部的数组空间,因此该语句执行之后,it 所持有的内部指针就已经无意义了(it 仍指向旧的缓冲区的某个位置),所以下一行语句继续使用 it 将会导致错误的指向与写入。由于过时的缓冲区有很大的可能已经被调度处于缺页状态,所以这个错误往往会导致 SIGSEGV 致命异常。如果产生了 SIGSEGV 信号,你可能是很幸运的,反而若是过时的缓冲区尚且有效,那么这一语句能够被执行且不报任何错误,那才是要命。
迭代器的搜索并删除 stdlib 的容器采用一种叫做 erase and remove 的惯用法来事实上删除一个元素。以 std::list 为例,remove_if() 能够从 list 中找到符合条件的元素,并将他们聚集(收集)起来移动到 list 的末尾,然后返回这组元素中的第一个元素的位置 iter,然而这些元素并未被从 list 中删除,如果你需要去掉他们的话,你需要以 list.erase(iter, list.end()) 来明确地移除它们。
所以删除元素是这样的:
bool IsOdd(int i) { return i & 1; }std::vector v = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9}; v.erase(std::remove_if(v.begin(), v.end(), IsOdd), v.end()); std::list l = { 1,100,2,3,10,1,11,-1,12 }; l.erase(l.remove_if(IsOdd), l.end());

由于 std::vector 不能像 std::list 那样聚集元素到链表末尾,所以它没有 remove_if() 成员函数,故而在它上面做 search & erase 需要 std::remove_if 的参与。而 std::list 可以直接使用成员函数 remove_if 来完成,代码也显得稍微简洁一些。
自 C++20 起,erase and remove_if 可以被简化为 std::erase_if() 或 erase_if() 成员函数,例如 std::erase, std::erase_if (std::vector) 。
后记 这次的 About customizing your own STL-like iterator 贡献了一些个人理解和最佳实践的准则,但是还有点点意犹未尽。
下回考虑是不是介绍一个 tree_t 及其迭代器实现,或许能够更有参考价值。
浅谈如何实现自定义的|浅谈如何实现自定义的 iterator
文章图片

Refs
  • 基于范围的 for 循环 (C++11 起) - cppreference.com
  • std::iterator - cppreference.com
  • std::input_iterator_tag, std::output_iterator_tag, std::forward_iterator_tag, std::bidirectional_iterator_tag, std::random_access_iterator_tag, std::contiguous_iterator_tag - cppreference.com
  • Increment/decrement operators - cppreference.com
  • operator overloading - cppreference.com
  • Traversing Trees with Iterators
  • c++ - How to implement an STL-style iterator and avoid common pitfalls? - Stack Overflow
  • c++ - Writing your own STL Container - Stack Overflow
  • STL & Generic Programming: Writing Your Own Iterators - Dr Dobb's
    c++ - How to correctly implement custom iterators and const_iterators? - Stack Overflow

    推荐阅读