C++自定义迭代器

实现自己的迭代器

最近在写数据结构,使用类模板实现,碰到了一些问题,其中有一个就是遍历的问题,查阅资料最后实现了自己的迭代器,让我实现的数据结构能像STL一样进行for循环遍历。

类的构成

#include <stdexcept>
#include <iterator>
template <typename T>
class Queue
{
public:
    // 迭代器
    class iterator;
    class const_iterator;
    typedef std::reverse_iterator<iterator> reverse_iterator;
    typedef std::reverse_iterator<const_iterator> const_reverse_iterator;

    Queue() : _capacity(5), _size(0), _items(new T[_capacity]), _front(0), _rear(-1) {}
    ~Queue();

    // 入队
    void enqueue(const T &item);
    // 出队
    T dequeue();
    // 查看队首元素
    T front() const;
    // 查看队尾元素
    T rear() const;
    // 获取队列元素数量
    unsigned int size() const;

    iterator begin() noexcept;
    iterator end() noexcept;
    const_iterator begin() const noexcept;
    const_iterator end() const noexcept;
    reverse_iterator rbegin() noexcept;
    reverse_iterator rend() noexcept;
    const_reverse_iterator rbegin() const noexcept;
    const_reverse_iterator rend() const noexcept;

private:
    // 存储元素的数组
    T *_items;
    // 队列当前容量
    unsigned int _capacity;
    // 队列当前大小
    unsigned int _size;
    // 队首
    unsigned int _front;
    // 队尾
    unsigned int _rear;

    // 调整队列的大小
    void resize();
    // 判断队列是否为空
    bool empty() const;
    // 判断当前队列是否为满
    bool full() const;
};

可以看到,我声明了两个类iteratorconst_iterator,这两个类就是需要我们自己定义的迭代器。

官方定义

template <class Category, class T, class Distance = ptrdiff_t,
          class Pointer = T*, class Reference = T&>
  struct iterator {
    typedef T         value_type;
    typedef Distance  difference_type;
    typedef Pointer   pointer;
    typedef Reference reference;
    typedef Category  iterator_category;
  };

对于官方定义的迭代器结构,有5个参数,在后续实现自己的迭代器使需要设置。

自定义迭代器的实现

iterator

template <typename T>
class Queue<T>::iterator
{
public:
    // 标准写法
    // 迭代器的类型
    using iterator_category = std::bidirectional_iterator_tag;
    // 数据类型
    using value_type = T;
    // 迭代器之间的逻辑距离
    using difference_type = std::ptrdiff_t;
    // 指针
    using pointer = T*;
    // 引用
    using reference = T&;

    // 构造函数
    iterator(pointer data = nullptr,
             unsigned int pos = 0,
             unsigned int capacity = 0,
             unsigned int start = 0,
             unsigned int size = 0)
        : _data(data), _pos(pos), _capacity(capacity), _start(start), _size(size) {}

    // 实现运算符重载
    // 解引用
    reference operator*() const 
    {
        return this->_data[(this->_start + this->_pos) % this->_capacity];
    }
    // 成员访问
    pointer operator->() const
    {
        return &(operator*());
    }
    // 前++运算符
    iterator& operator++()
    {
        ++this->_pos;
        return *this;
    }
    // 后++运算符
    iterator operator++(int)
    {
        iterator tmp = *this;
        ++this->_pos;
        return tmp;
    }

    // 前--运算符
    iterator& operator--()
    {
        --this->_pos;
        return *this;
    }
    // 后--运算符
    iterator operator--(int)
    {
        iterator tmp = *this;
        --this->_pos;
        return tmp;
    }

    // 比较运算符
    bool operator==(const iterator& other) const
    {
        return this->_pos == other._pos;
    }
    bool operator!=(const iterator& other) const
    {
        return !(*this == other);
    }
    

private:
    pointer _data;          // 指向队列数据数组的指针
    unsigned int _pos;      // 当前逻辑位置
    unsigned int _capacity; // 队列容量
    unsigned int _start;    // 队列实际起始位置
    unsigned int _size;     // 队列当前元素数量
};

const_iterator

template <typename T>
class Queue<T>::const_iterator
{
    // 标准写法
    // 迭代器的类型
    using iterator_category = std::bidirectional_iterator_tag;
    // 数据类型
    using value_type = const T;
    // 迭代器之间的逻辑距离
    using difference_type = std::ptrdiff_t;
    // 指针
    using pointer = const T*;
    // 引用
    using reference = const T&;

    // 构造函数
    const_iterator(const pointer data = nullptr,
             unsigned int pos = 0,
             unsigned int capacity = 0,
             unsigned int start = 0,
             unsigned int size = 0)
        : _data(data), _pos(pos), _capacity(capacity), _start(start), _size(size) {}

    // 实现运算符重载
    // 解引用
    reference operator*() const 
    {
        return this->_data[(this->_start + this->_pos) % this->_capacity];
    }
    // 成员访问
    pointer operator->() const
    {
        return &(operator*());
    }
    // 前++运算符
    const_iterator& operator++()
    {
        ++this->_pos;
        return *this;
    }
    // 后++运算符
    const_iterator operator++(int)
    {
        const_iterator tmp = *this;
        ++this->_pos;
        return tmp;
    }

    // 前--运算符
    const_iterator& operator--()
    {
        --this->_pos;
        return *this;
    }
    // 后--运算符
    const_iterator operator--(int)
    {
        const_iterator tmp = *this;
        --this->_pos;
        return tmp;
    }


    // 比较运算符
    bool operator==(const const_iterator& other) const
    {
        return this->_pos == other._pos;
    }
    bool operator!=(const const_iterator& other) const
    {
        return !(*this == other);
    }
    

private:
    pointer _data;          // 指向队列数据数组的指针
    unsigned int _pos;      // 当前逻辑位置
    unsigned int _capacity; // 队列容量
    unsigned int _start;    // 队列实际起始位置
    unsigned int _size;     // 队列当前元素数量
};

begin()和end()的实现

template <typename T>
typename Queue<T>::iterator Queue<T>::begin() noexcept
{
    return iterator(this->_items, 0, this->_capacity, this->_front, this->_size);
}

template <typename T>
typename Queue<T>::iterator Queue<T>::end() noexcept
{
    return iterator(this->_items, this->_size, this->_capacity, this->_front, this->_size);
}

template <typename T>
typename Queue<T>::const_iterator Queue<T>::begin() const noexcept
{
    return const_iterator(this->_items, 0, this->_capacity, this->_front, this->_size);
}

template <typename T>
typename Queue<T>::const_iterator Queue<T>::end() const noexcept
{
    return const_iterator(this->_items, this->_size, this->_capacity, this->_front, this->_size);
}

rbegin()和rend()的实现

template <typename T>
typename Queue<T>::reverse_iterator Queue<T>::rbegin() noexcept
{
    return reverse_iterator(this->end());
}

template <typename T>
typename Queue<T>::reverse_iterator Queue<T>::rend() noexcept
{
    return reverse_iterator(this->begin());
}

template <typename T>
typename Queue<T>::const_reverse_iterator Queue<T>::rbegin() const noexcept
{
    return const_reverse_iterator(this->end());
}

template <typename T>
typename Queue<T>::const_reverse_iterator Queue<T>::rend() const noexcept
{
    return const_reverse_iterator(this->begin());
}

其他代码自己就不放出来了,主要是为了说明自定义迭代器,可以参照我的代码实现自己的迭代器。

测试

void testQueue()
{
    Queue<int> que;
    que.enqueue(4);
    que.enqueue(2);
    que.enqueue(5);
    std::cout << que.rear() << std::endl;

    for(const auto& item : que)
    {
        std::cout << item << " ";
    }
    std::cout << std::endl;

    for(auto it = que.rbegin(); it != que.rend(); ++it)
    {
        std::cout << *it << " ";
    }
    std::cout << std::endl;
}


int main() {
    testQueue();
    return 0;
}

输出的结果为:

5
4 2 5
5 2 4
### 创建或使用自定义迭代器实现方法 #### C++ 中的自定义迭代器实现C++ 中,可以通过继承 `std::iterator` 或者更现代的方式通过继承 `boost::iterator_facade` 来创建自定义迭代器。这种方式允许开发者灵活地控制迭代行为并支持 STL 风格的操作。 以下是基于 Boost 库的一个简单例子,展示如何构建一个随机访问类型的迭代器: ```cpp #include <boost/iterator/iterator_facade.hpp> #include <vector> class MyIterator : public boost::iterator_facade< MyIterator, int, boost::random_access_traversal_tag> { private: std::vector<int>::const_iterator current; public: explicit MyIterator(std::vector<int>::const_iterator it) : current(it) {} private: friend class boost::iterator_core_access; reference dereference() const { return *current; } bool equal(MyIterator const& other) const { return current == other.current; } void increment() { ++current; } void decrement() { --current; } difference_type distance_to(MyIterator const& other) const { return std::distance(current, other.current); } void advance(difference_type n) { std::advance(current, n); } }; ``` 上述代码展示了如何利用 `boost::iterator_facade` 构建一个简单的随机访问迭代器[^2]。该迭代器封装了底层容器(这里是 `std::vector<int>` 的常量迭代器),并通过重载必要的操作符来满足 STL 要求。 --- #### Python 中的自定义迭代器实现 在 Python 中,自定义迭代器通常涉及实现两个特殊方法:`__iter__()` 和 `__next__()`。如果希望支持切片或其他复杂功能,则还需要额外重载其他方法,比如 `__getitem__()` 和 `__len__()`。 下面是一个示例,演示如何为一个简单的数据结构创建迭代器: ```python class MyCustomContainer: def __init__(self, data): self.data = data self.index = 0 def __iter__(self): # 返回迭代器对象本身 return self def __next__(self): # 定义每次调用 next() 方法的行为 if self.index >= len(self.data): raise StopIteration # 停止迭代条件 value = self.data[self.index] self.index += 1 return value container = MyCustomContainer([1, 2, 3]) for item in container: print(item) ``` 此代码片段实现了基本的迭代逻辑,并能够用于循环语句中。对于更加复杂的场景,还可以扩展此类以支持索引访问和其他特性[^4]。 --- #### 关键技术点总结 无论是 C++ 还是 Python,在设计自定义迭代器时都需要考虑以下几个方面: - **接口一致性**:确保遵循目标语言的标准协议(如 C++ 的 STL 协议或 Python 的迭代器协议)。 - **性能优化**:针对特定需求调整内部存储机制和算法效率。 - **灵活性增强**:提供多种遍历模式的支持(前向、双向或者随机访问)[^1]。 ---
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

ht巷子

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值