leetcode刷题笔记284题 顶端迭代器

leetcode刷题笔记284题 顶端迭代器

源地址：284. 顶端迭代器

问题描述：

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

示例:

假设迭代器被初始化为列表 [1,2,3]。

调用 next() 返回 1，得到列表中的第一个元素。
现在调用 peek() 返回 2，下一个元素。在此之后调用 next() 仍然返回 2。
最后一次调用 next() 返回 3，末尾元素。在此之后调用 hasNext() 应该返回 false。
进阶：你将如何拓展你的设计？使之变得通用化，从而适应所有的类型，而不只是整数型？

// Scala Iterator reference:
// https://www.scala-lang.org/api/2.12.2/scala/collection/Iterator.html

class PeekingIterator(private val iterator: Iterator[Int]) {
    var cur = 0
    var has_next = iterator.hasNext
    if (has_next == true) cur = iterator.next.toInt

    def peek(): Int = {
        return cur
    }
    
    def next(): Int = {
        val temp = cur
        has_next = iterator.hasNext
        if (has_next == true) cur = iterator.next.toInt
        return temp
    }
    
    def hasNext(): Boolean = {
        return has_next
    }
}

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