[Leetcode] Binary search tree -- 173. Binary Search Tree Iterator

Implement an iterator over a binary search tree (BST). Your iterator will be initialized with the root node of a BST.

Calling next() will return the next smallest number in the BST.

Note: next() and hasNext() should run in average O(1) time and uses O(h) memory, where h is the height of the tree.

 

Solution:

Getting the next iterator is similarly to call  the generator in the python. for the next smallest number, it's like the inorder tree traversal printed out (sorted order). we use stack the store the node visited while using iterators

so the next could pop the node in order from stack.

 

    def __init__(self, root):
        """
        :type root: TreeNode
        """
        self.stk = []
        self.getAllLeftNode(root)

    def hasNext(self):
        """
        :rtype: bool
        """
        
        return (len(self.stk) != 0)

    def next(self):
        """
        :rtype: int
        """
        #pop from stack
        node = self.stk.pop()
        self.getAllLeftNode(node.right)
        return node.val
    
    def getAllLeftNode(self, node):
        '''
        get the most left nodes and put in the stack
        '''
        while (node is not None):
            self.stk.append(node)
            node = node.left