173. 二叉搜索树迭代器

173. 二叉搜索树迭代器

1.题目

2.我的解决方案

• 题目本身不太好理解
• 二叉树题目已经给出，所以只需要root初始化
• 其他就是简单的中序遍历上附加的操作

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class BSTIterator:
    def __init__(self, root: TreeNode): # 形成二叉搜索树中序遍历序列
        def middleTraverse(root):
            if not root:
                return root
            
            middleTraverse(root.left)
            self.nums.append(root.val)
            middleTraverse(root.right)
        
        self.root = root
        self.nums = list()

        middleTraverse(self.root)

    def next(self) -> int:  # 中序遍历二叉搜索树
        return self.nums.pop(0)


    def hasNext(self) -> bool:  # 判断右侧有没有数字
        if len(self.nums) >= 1:
            return True

        return False


# Your BSTIterator object will be instantiated and called as such:
# obj = BSTIterator(root)
# param_1 = obj.next()
# param_2 = obj.hasNext()

• 时间复杂度:$O(n)$
• 空间复杂度:$O(n)$