1005 lc.284 iterator设计-优快云博客

本文链接：https://blog.youkuaiyun.com/qq_52138199/article/details/120619330

该博客介绍了如何在C++中实现一个PeekingIterator类，该类扩展了基本的Iterator接口，增加了peek方法，允许查看但不移动迭代器。PeekingIterator构造时初始化迭代器，并在需要时存储下一个元素。next方法返回并更新下一个元素，而peek方法则直接返回当前的下一个元素。这个实现有助于在遍历序列时提供额外的灵活性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

lc.284 顶端迭代器

请你设计一个迭代器，除了支持 hasNext 和 next 操作外，还支持 peek 操作。

实现 PeekingIterator 类：

PeekingIterator(int[] nums) 使用指定整数数组 nums 初始化迭代器。
int next() 返回数组中的下一个元素，并将指针移动到下个元素处。
bool hasNext() 如果数组中存在下一个元素，返回 true ；否则，返回 false 。
int peek() 返回数组中的下一个元素，但 不 移动指针。

/*
 * Below is the interface for Iterator, which is already defined for you.
 * **DO NOT** modify the interface for Iterator.
 *
 *  class Iterator {
 *		struct Data;
 * 		Data* data;
 *  public:
 *		Iterator(const vector<int>& nums);
 * 		Iterator(const Iterator& iter);
 *
 * 		// Returns the next element in the iteration.
 *		int next();
 *
 *		// Returns true if the iteration has more elements.
 *		bool hasNext() const;
 *	};
 */



class PeekingIterator : public Iterator {
public:
    int nextElement;
    bool flag;

    PeekingIterator(const vector<int>& nums) : Iterator(nums) {
        flag = Iterator::hasNext();
        if (flag) {
            nextElement = Iterator::next();
        }
    }
    
    int peek() {
        return nextElement;
    }
    
    int next() {
        int ret = nextElement;
        flag = Iterator::hasNext();
        if (flag) {
            nextElement = Iterator::next();
        }
        return ret;
    }
    
    bool hasNext() const {
        return flag;
    }


};

C++ 中 PeekingIterator 继承父类 Iterator，Iterator 已经实现方法 next 和 hasNext，在此我们在 PeekingIterator 中主要实现 peek方法即可。我们使用 flag 标记迭代器是否还有剩余元素，使用 nextElement 存储迭代器的下一个元素。

next：首先用 ret存储 nextElement表示返回值，flag 保存 Iterator 调用 hasNext方法的返回结果，然后将 nextElement 向后移动一位，最后返回 ret；

hasNext：返回 flag；

peek：由于 peek\操作不改变指针，因此返回 nextElement。