每日一题(4)

题目：顶端迭代器

题目描述

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

实现 PeekingIterator 类：

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

解题思路

PeekingIterator 类继承了Iterator类，那首先需要去了解Iterator类，本题不是让我们从零去实现这些功能的
Iterator接口：
hasnext()：判断对象是否还有下一个元素，如果已经是最后一个则返回false
next()：返回当前元素的值，并把指针指向下一个元素的位置

代码

class PeekingIterator : public Iterator {
public:
	PeekingIterator(const vector<int>& nums) : Iterator(nums) {
	    flag = Iterator::hasNext();
        if(flag)
        {
            num = Iterator::next();
        }
	}
	
	int peek() {
        return num;
	}
	int next() {
        int res = num;
	    flag = Iterator::hasNext();
        if(flag)
        {
            num = Iterator::next();
        }
        return res;
	}
	bool hasNext() const {
	    return flag;
	}
private:
    int num;
    bool flag;
};