1005 lc.284 iterator设计

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

实现 PeekingIterator 类：

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

• C++

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

    PeekingIterator(const vector& 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。