Skip to content

341 Flatten Nested List Iterator

341. Flatten Nested List Iterator

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

难度: Medium

class NestedIterator(object):

    def __init__(self, nestedList):
        """
        Initialize your data structure here.
        :type nestedList: List[NestedInteger]
        """
        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):
        """
        :rtype: int
        """
        if self.hasNext():
            return self.stack.pop(0)


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


回到顶部