Skip to content

Instantly share code, notes, and snippets.

@vrat28
Created April 13, 2021 10:28
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save vrat28/9d785bdcaf3f18e45d87b0eb9ced4563 to your computer and use it in GitHub Desktop.
Save vrat28/9d785bdcaf3f18e45d87b0eb9ced4563 to your computer and use it in GitHub Desktop.
Flatten Nest List iterator (Python)
class NestedIterator:
def __init__(self, nestedList: [NestedInteger]):
def flatten(a):
req = []
for ele in a:
if ele.isInteger():
req.append(ele.getInteger())
else:
req.extend(flatten(ele.getList()))
return req
self.flatten = flatten(nestedList)
self.curr = 0
def next(self) -> int:
t = self.flatten[self.curr]
self.curr+=1
return t
def hasNext(self) -> bool:
return self.curr<len(self.flatten)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment