Ai
1 Star 2 Fork 5

LilithSangreal/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
flatten-2d-vector.py 612 Bytes
一键复制 编辑 原始数据 按行查看 历史
kamyu 提交于 2018-12-28 09:08 +08:00 . Update flatten-2d-vector.py
# Time: O(1)
# Space: O(1)
from collections import deque
class Vector2D(object):
def __init__(self, vec2d):
"""
Initialize your data structure here.
:type vec2d: List[List[int]]
"""
self.stack = deque((len(v), iter(v)) for v in vec2d if v)
def next(self):
"""
:rtype: int
"""
length, iterator = self.stack.popleft()
if length > 1:
self.stack.appendleft((length-1, iterator))
return next(iterator)
def hasNext(self):
"""
:rtype: bool
"""
return bool(self.stack)
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/LilithSangreal/LeetCode-Solutions.git
git@gitee.com:LilithSangreal/LeetCode-Solutions.git
LilithSangreal
LeetCode-Solutions
LeetCode-Solutions
master

搜索帮助