Leetcode284. Peeking Iterator顶端迭代器

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

示例:

假设迭代器被初始化为列表 [1,2,3]。 调用 next() 返回 1，得到列表中的第一个元素。 现在调用 peek() 返回 2，下一个元素。在此之后调用 next() 仍然返回 2。 最后一次调用 next() 返回 3，末尾元素。在此之后调用 hasNext() 应该返回 false。

 

class Iterator {
    struct Data;
	Data* data;
public:
	Iterator(const vector<int>& nums);
	Iterator(const Iterator& iter);
	virtual ~Iterator();
	// Returns the next element in the iteration.
	int next();
	// Returns true if the iteration has more elements.
	bool hasNext() const;
};


class PeekingIterator : public Iterator {
    private:
    bool flag;
    int value;
public:
	PeekingIterator(const vector<int>& nums) : Iterator(nums) {
	    // Initialize any member here.
	    // **DO NOT** save a copy of nums and manipulate it directly.
	    // You should only use the Iterator interface methods.
        flag = false;
	    
	}

    // Returns the next element in the iteration without advancing the iterator.
	int peek() {
        if(flag)
            return value;
        flag = true;
        value = Iterator :: next();
        return value;
	}

	// hasNext() and next() should behave the same as in the Iterator interface.
	// Override them if needed.
	int next() {
	    if(flag)
        {
            flag = false;
            return value;
        }
        return Iterator::next();
	}

	bool hasNext() const {
	    if(flag) return true;
        if(Iterator::hasNext())
            return true;
        return false;
	}
};