小黑leetcode之旅:341. 扁平化嵌套列表迭代器

小黑答案

# """
# This is the interface that allows for creating nested lists.
# You should not implement it, or speculate about its implementation
# """
#class NestedInteger:
# def isInteger(self) -> bool:
# """
# @return True if this NestedInteger holds a single integer, rather than a nested list.
# """
#
# def getInteger(self) -> int:
# """
# @return the single integer that this NestedInteger holds, if it holds a single integer
# Return None if this NestedInteger holds a nested list
# """
#
# def getList(self) -> [NestedInteger]:
# """
# @return the nested list that this NestedInteger holds, if it holds a nested list
# Return None if this NestedInteger holds a single integer
# """

class NestedIterator:
    def __init__(self, nestedList: [NestedInteger]):
        self.arr = []
        print(nestedList)
        self.get_list(nestedList)
        self.i = 0
        self.len = len(self.arr)
    def get_list(self,nestedList):
        for item in nestedList:
            if item.getList():
                self.get_list(item.getList())
            elif item.isInteger():
                self.arr.append(item.getInteger())

    def next(self) -> int:
        item = self.arr[self.i]
        self.i += 1
        return item
    
    def hasNext(self) -> bool:
         if self.i < self.len:
             return True
         else:
             return False

# Your NestedIterator object will be instantiated and called as such:
# i, v = NestedIterator(nestedList), []
# while i.hasNext(): v.append(i.next())