[99클럽] 34일차 TIL

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

 

class NestedIterator(object):

    def __init__(self, nestedList):

        def dfs(nestedList):
            for item in nestedList:
                if item.isInteger():
                    self.stack.append(item.getInteger())
                else:
                    dfs(item.getList())
        self.stack = []
        dfs(nestedList)
        

    def next(self):
        
        if self.hasNext():
            return self.stack.pop(0)


    def hasNext(self):
        
        return self.stack != []