17370845950

c++如何写一个迭代器 c++自定义iterator实现【进阶】
c++kquote>自定义C++迭代器需显式定义value_type、difference_type、pointer、reference、iterator_category五类typedef,并按所选category(如random_access_iterator_tag)完整实现对应操作符,否则无法兼容STL算法与范围for。

c++自定义迭代器的核心要点

写一个符合标准容器要求的自定义迭代器,关键不是“能遍历”,而是让 operator*operator->operator++operator!=(或 operator==)等行为与 STL 算法和范围 for 兼容,并正确声明类型别名(如 value_typeiterator_category 等)。否则即使能用,也会在 std::sortstd::distancefor (auto& x : container) 中出错。

必须定义的五类 typedef(类型别名)

标准要求迭代器至少提供以下 5 个嵌套类型(C++17 起部分可由 std::iterator_traits 推导,但显式定义更稳妥、更清晰):

  • value_type:所指元素的类型(如 T
  • difference_type:两个迭代器距离的有符号整数类型(通常为 std::ptrdiff_t
  • pointer:指向元素的指针类型(如 T*
  • reference:解引用返回的引用类型(如 T&const T&
  • iterator_category:迭代器类别标签(如 std::random_access_iterator_tag),决定支持哪些操作

按需求选择迭代器类别并实现对应接口

类别决定你能用哪些算法,也决定你要实现多少操作。常见场景:

  • 输入迭代器(input_iterator_tag):只读、单向、一次遍历(如 I/O 流)。只需支持 ++it*itit != other,不可保存副本多次使用
  • 前向迭代器(forward_iterator_tag):可多次遍历、支持 it++(后置)、可拷贝保存。需额外实现 operator==
  • 双向迭代器(bidirectional_iterator_tag):支持 --itit--。比如链表节点迭代器
  • 随机访问迭代器(random_access_iterator_tag):支持 it + nit[n]it1 - it2it1 。底层是连续内存(如 vector)时应实现它

注意:operator-operator+ 不是必须重载的成员函数,但若声明为 random_access_iterator_tag,就必须能让 std::distancestd::advance 正常工作——通常靠提供 operator- 成员或非成员函数实现。

完整示例:只读随机访问迭代器(包装 raw pointer)

假设你有一个固定大小的 T* 数组,想封装成安全易用的迭代器:

template 
class const_span_iterator {
public:
    using value_type = T;
    using difference_type = std::ptrdiff_t;
    using pointer = const T*;
    using reference = const T&;
    using iterator_category = std::random_access_iterator_tag;
const_span_iterator(const T* ptr = nullptr) : ptr_(ptr) {}

reference operator*() const { return *ptr_; }
pointer operator->() const { return ptr_; }

const_span_iterator& operator++() { ++ptr_; return *this; }
const_span_iterator operator++(int) { auto tmp = *this; ++ptr_; return tmp; }
const_span_iterator& operator--() { --ptr_; return *this; }
const_span_iterator operator--(int) { auto tmp = *this; --ptr_; return tmp; }

const_span_iterator& operator+=(difference_type n) { ptr_ += n; return *this; }
const_span_iterator& operator-=(difference_type n) { ptr_ -= n; return *this; }

const_span_iterator operator+(difference_type n) const { return const_span_iterator(ptr_ + n); }
const_span_iterator operator-(difference_type n) const { return const_span_iterator(ptr_ - n); }

reference operator[](difference_type n) const { return *(ptr_ + n); }

friend difference_type operator-(const const_span_iterator& a, const const_span_iterator& b) {
    return a.ptr_ - b.ptr_;
}

friend bool operator==(const const_span_iterator& a, const const_span_iterator& b) {
    return a.ptr_ == b.ptr_;
}
friend bool operator!=(const const_span_iterator& a, const const_span_iterator& b) {
    return !(a == b);
}
friend bool operator<(const const_span_iterator& a, const const_span_iterator& b) {
    return a.ptr_ < b.ptr_;
}
// 其他比较运算符(>, <=, >=)可依需补充

private: const T* ptr_; };

这个迭代器可直接用于 std::sort(first, last, comp)(只读)、std::lower_bound、范围 for(配合 begin()/end()),也满足 std::is_random_access_iterator_v<...>

进阶提示:避免常见坑

  • 不要把 iteratorconst_iterator 实现成同一类型——它们的 referencepointer 必须不同(一个是 T&,一个是 const T&),否则无法支持 container.begin()container.cbegin() 的语义分离
  • 迭代器必须是 可平凡复制(trivially copyable)可默认构造 的;析构函数不应有副作用
  • 如果容器可能被移动(move),确保迭代器不持有 dangling 指针——可考虑用索引代替裸指针,或绑定容器生命周期(如 span 类型)
  • C++20 起推荐继承 std::iterator(已弃用)或直接用概念约束(std::random_access_iterator),但手动定义仍是理解本质的最佳方式