lc.284 顶端迭代器
请你设计一个迭代器,除了支持 hasNext 和 next 操作外,还支持 peek 操作。
实现 PeekingIterator 类:
PeekingIterator(int[] nums) 使用指定整数数组 nums 初始化迭代器。
int next() 返回数组中的下一个元素,并将指针移动到下个元素处。
bool hasNext() 如果数组中存在下一个元素,返回 true ;否则,返回 false 。
int peek() 返回数组中的下一个元素,但 不 移动指针。
- C++
/** 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。