Peeking Iterator 顶端迭代器

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

示例:

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

进阶:你将如何拓展你的设计?使之变得通用化,从而适应所有的类型,而不只是整数型?

思路:PeekingIterator是自定义的迭代器,是对顶层迭代器Iterator的封装。所以对于next ()方法和hasNext () 方法其实已经封装好了,我们不需要再写,我们需要的只是重新在Iterator的基础上增加peek ()方法。这点很重要,一开始我以为要自己实现next ()方法和hasNext () 。

所以这道题的核心思想和解决方法就是增加一个缓存cache用来存储peek的值,并且用一个bool类型的变量来标识cache是否已经被赋值。

参考代码:

// 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);
	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 {
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.
		hasPeeked = false;
	}

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

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

	bool hasNext() const {
		if (hasPeeked) return true;
		return Iterator::hasNext();
	}
private:
	bool hasPeeked;
	int cache;
};

猜你喜欢

转载自blog.csdn.net/qq_26410101/article/details/82939436