From a2dce9493e92e2a80a6a303c7c41758ccf1805b7 Mon Sep 17 00:00:00 2001 From: marina <138340846+bt3gl-cryptographer@users.noreply.github.com> Date: Thu, 3 Aug 2023 13:10:38 -0700 Subject: [PATCH] Create bst_inorder_iterator.py --- trees/bst_inorder_iterator.py | 36 +++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 trees/bst_inorder_iterator.py diff --git a/trees/bst_inorder_iterator.py b/trees/bst_inorder_iterator.py new file mode 100644 index 0000000..3bbf05b --- /dev/null +++ b/trees/bst_inorder_iterator.py @@ -0,0 +1,36 @@ +#!/usr/bin/env python3 +# -*- coding: utf-8 -*- +# author: bt3gl + +class Node: + def __init__(self, left=None, right=None): + self.val = val + self.left = left + self.right = right + +class BST_Iterator: + + def __init__(self, root: Optional[Node]): + + self.stack = [] + self.left_inorder(root) + + def left_inorder(self, root): + + while root: + self.stack.append(root) + root = root.left + + def next(self) -> int: + + top_node = self.stack.pop() + + if top_node.right: + self.left_inorder(top_node.right) + + return top_node.val + + def has_next(self) -> bool: + + return len(self.stack) > 0 +