diff --git a/tries/trie_postorder.py b/tries/trie_postorder.py index 992e66c..07aa755 100644 --- a/tries/trie_postorder.py +++ b/tries/trie_postorder.py @@ -3,7 +3,7 @@ # author: bt3gl -def postorder(self, root: 'Node') -> List[int]: +def postorder(self, root: 'Node'): if root is None: return [] @@ -11,9 +11,12 @@ def postorder(self, root: 'Node') -> List[int]: stack, result = [root, ], [] while stack: + node = stack.pop() + if node is not None: result.append(node.val) + for c in node.children: stack.append(c)