顶端迭代器(JS实现)

1 题目
给定一个迭代器类的接口,接口包含两个方法: next() 和 hasNext()。设计并实现一个支持 peek() 操作的顶端迭代器 – 其本质就是把原本应由 next() 方法返回的元素 peek() 出来。
示例:
假设迭代器被初始化为列表 [1,2,3]。
调用 next() 返回 1,得到列表中的*个元素。
现在调用 peek() 返回 2,下一个元素。在此之后调用 next() 仍然返回 2。
*后一次调用 next() 返回 3,末尾元素。在此之后调用 hasNext() 应该返回 false。
进阶:你将如何拓展你的设计?使之变得通用化,从而适应所有的类型,而不只是整数型?

链接:https://leetcode-cn.com/problems/peeking-iterator

2 思路
这道题需要支持查看下一个元素,因此我们就多执行一次next操作,然后把结果暂时存下来,需要时返回即可

3代码
/**
* // This is the Iterator’s API interface.
* // You should not implement it, or speculate about its implementation.
* function Iterator() {
* @ return {number}
* this.next = function() { // return the next number of the iterator
* …
* };
*
* @return {boolean}
* this.hasNext = function() { // return true if it still has numbers
* …
* };
* };
*/

/**
* @param {Iterator} iterator
*/
var PeekingIterator = function(iterator) {
this.iterator = iterator;
this.topNum = null;
};

/**
* @return {number}
*/
PeekingIterator.prototype.peek = function() {
if (!this.topNum) this.topNum = this.iterator.next();
return this.topNum;
};

/**
* @return {number}
*/
PeekingIterator.prototype.next = function() {
if (this.topNum) {
let temp = this.topNum;
this.topNum = null;
return temp;
} else {
return this.iterator.next();
}
};

/**
* @return {boolean}
*/
PeekingIterator.prototype.hasNext = function() {
if (this.topNum) return true;
return this.iterator.hasNext();
};

/**
* Your PeekingIterator object will be instantiated and called as such:
* var obj = new PeekingIterator(arr)
* var param_1 = obj.peek()
* var param_2 = obj.next()
* var param_3 = obj.hasNext()
*/