给定一个迭代器类的接口,接口包含两个方法: next()
和 hasNext()
。设计并实现一个支持 peek()
操作的顶端迭代器 -- 其本质就是把原本应由 next()
方法返回的元素 peek()
出来。
示例:
假设迭代器被初始化为列表[1,2,3]
。 调用next()
返回 1,得到列表中的第一个元素。 现在调用peek()
返回 2,下一个元素。在此之后调用next()
仍然返回 2。 最后一次调用next()
返回 3,末尾元素。在此之后调用hasNext()
应该返回 false。
进阶:你将如何拓展你的设计?使之变得通用化,从而适应所有的类型,而不只是整数型?
思路:
用一个list把迭代出来的数据都存起来就好了。
class PeekingIterator(object):
def __init__(self, iterator):
"""
Initialize your data structure here.
:type iterator: Iterator
"""
self.list = list()
while(iterator.hasNext()):
self.list.append(iterator.next())
def peek(self):
"""
Returns the next element in the iteration without advancing the iterator.
:rtype: int
"""
return self.list[0]
def next(self):
"""
:rtype: int
"""
return self.list.pop(0)
def hasNext(self):
"""
:rtype: bool
"""
return len(self.list) != 0