请设计并实现一个能够展开二维向量的迭代器。该迭代器需要支持 next
和 hasNext
两种操作。、
示例:
Vector2D iterator = new Vector2D([[1,2],[3],[4]]); iterator.next(); // 返回 1 iterator.next(); // 返回 2 iterator.next(); // 返回 3 iterator.hasNext(); // 返回 true iterator.hasNext(); // 返回 true iterator.next(); // 返回 4 iterator.hasNext(); // 返回 false
注意:
- 请记得 重置 在 Vector2D 中声明的类变量(静态变量),因为类变量会 在多个测试用例中保持不变,影响判题准确。请 查阅 这里。
- 你可以假定
next()
的调用总是合法的,即当next()
被调用时,二维向量总是存在至少一个后续元素。
思路:
在初始化时就把二维数组展开。
class Vector2D(object):
def __init__(self, v):
"""
:type v: List[List[int]]
"""
self.list = []
for nums in v:
for item in nums:
self.list.append(item)
self.index = 0
def next(self):
"""
:rtype: int
"""
self.index += 1
return self.list[self.index - 1]
def hasNext(self):
"""
:rtype: bool
"""
return self.index != len(self.list)
# Your Vector2D object will be instantiated and called as such:
# obj = Vector2D(v)
# param_1 = obj.next()
# param_2 = obj.hasNext()