284. 顶端迭代器（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()
 */