프로그래밍/LeetCode

[LeetCode][Python] 341. Flatten Nested List Iterator

snoopybox 2019. 3. 22. 01:39

Problem :

https://leetcode.com/problems/flatten-nested-list-iterator/


My Solution :

class NestedIterator(object):

def __init__(self, nestedList):
def generator(nestedList):
for item in nestedList:
if item.isInteger():
yield item.getInteger()
else:
for yielded in generator(item.getList()):
yield yielded

self.generator = generator(nestedList)

def next(self):
return self.value

def hasNext(self):
self.value = next(self.generator, None)
return self.value is not None