LeetCode-284. 顶端迭代器

给定一个迭代器类的接口，接口包含两个方法： next() 和 hasNext()。设计并实现一个支持 peek() 操作的顶端迭代器 -- 其本质就是把原本应由 next() 方法返回的元素 peek() 出来。

示例:

假设迭代器被初始化为列表 [1,2,3]。

调用 next() 返回 1，得到列表中的第一个元素。
现在调用 peek() 返回 2，下一个元素。在此之后调用 next() 仍然返回 2。
最后一次调用 next() 返回 3，末尾元素。在此之后调用 hasNext() 应该返回 false。
进阶：你将如何拓展你的设计？使之变得通用化，从而适应所有的类型，而不只是整数型？

构造的时候就执行一次next，记录peek节点。

每次进行next操作的时候需要判断是否到达尾部。

#include <iostream>
#include <vector>
using namespace std;


class Iterator {
    struct Data;
    Data* data;
public:
    Iterator(const vector<int>& nums);
    Iterator(const Iterator& iter);
    virtual ~Iterator();
    int next();
    bool hasNext() const;
};


class PeekingIterator : public Iterator {
public:
    PeekingIterator(const vector<int>& nums) : Iterator(nums) {
        IsPeekEnd = false;
        if(Iterator::hasNext()){
            PeekHead = Iterator::next();
        }else{
            IsPeekEnd = true;
        }
    }

    int peek() {
        return PeekHead;
    }

    int next() {
        int CurNext = PeekHead;
        if(!IsPeekEnd){
            if(Iterator::hasNext()){
                PeekHead = Iterator::next();
            }else{
                IsPeekEnd =true;
            }
        }
        return CurNext;
    }

    bool hasNext() const {
        return (!IsPeekEnd);
    }

private:
    int  PeekHead;   /* 记录当前peek节点 */
    bool IsPeekEnd;  /* 记录是否到末尾 */
};