Advertisement
LikeRampage

Python3 Leetcode 341. Flatten Nested List Iterator Chatgpt

May 2nd, 2024
555
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.55 KB | Software | 0 0
  1. class NestedIterator:
  2.     def __init__(self, nestedList: [NestedInteger]):
  3.         self.stack = []
  4.         for i in reversed(nestedList):
  5.             self.stack.append(i)
  6.        
  7.     def next(self) -> int:
  8.         return self.stack.pop().getInteger()
  9.        
  10.     def hasNext(self) -> bool:
  11.         while self.stack:
  12.             top = self.stack[-1]
  13.             if top.isInteger():
  14.                 return True
  15.             self.stack.pop()
  16.             for i in reversed(top.getList()):
  17.                 self.stack.append(i)
  18.         return False
Tags: python3
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement