284.顶端迭代器
题目描述
思路:模拟
存储上一个peek的元素,如果有就在peek时直接返回,不移动指针;next时清空peek的元素。
# Below is the interface for Iterator, which is already defined for you.
#
# class Iterator:
# def __init__(self, nums):
# """
# Initializes an iterator object to the beginning of a list.
# :type nums: List[int]
# """
#
# def hasNext(self):
# """
# Returns true if the iteration has more elements.
# :rtype: bool
# """
#
# def next(self):
# """
# Returns the next element in the iteration.
# :rtype: int
# """
class PeekingIterator:
def __init__(self, iterator):
"""
Initialize your data structure here.
:type iterator: Iterator
"""
self.iter = iterator
self.pk = None
def peek(self):
"""
Returns the next element in the iteration without advancing the iterator.
:rtype: int
"""
if self.pk is None:
self.pk = self.iter.next()
return self.pk
def next(self):
"""
:rtype: int
"""
if self.pk is not None:
val = self.pk
self.pk = None
return val
return self.iter.next()
def hasNext(self):
"""
:rtype: bool
"""
return self.pk is not None or self.iter.hasNext()
# Your PeekingIterator object will be instantiated and called as such:
# iter = PeekingIterator(Iterator(nums))
# while iter.hasNext():
# val = iter.peek() # Get the next element but not advance the iterator.
# iter.next() # Should return the same value as [val].