题目:
实现一个二叉搜索树迭代器。你将使用二叉搜索树的根节点初始化迭代器。
调用 next()
将返回二叉搜索树中的下一个最小的数。
示例:
BSTIterator iterator = new BSTIterator(root); iterator.next(); // 返回 3 iterator.next(); // 返回 7 iterator.hasNext(); // 返回 true iterator.next(); // 返回 9 iterator.hasNext(); // 返回 true iterator.next(); // 返回 15 iterator.hasNext(); // 返回 true iterator.next(); // 返回 20 iterator.hasNext(); // 返回 false
提示:
next()
和hasNext()
操作的时间复杂度是 O(1),并使用 O(h) 内存,其中 h 是树的高度。- 你可以假设
next()
调用总是有效的,也就是说,当调用next()
时,BST 中至少存在一个下一个最小的数。
思路:
最左值一定是最小值。我们只需要将左边不断压栈,那么栈中存放的就是从大到小的值。调用next
的时候,我们只需要弹出栈顶元素即可,同时我们需要将right
加入栈中(非递归先序遍历二叉树的思路)。
代码:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class BSTIterator:
def __init__(self, root: TreeNode):
self.stack = list()
self.PushLeft(root)
def next(self) -> int:
"""
@return the next smallest number
"""
node = self.stack.pop()
self.PushLeft(node.right)
return node.val
def PushLeft(self,node):
while node:
self.stack.append(node)
node = node.left
def hasNext(self) -> bool:
"""
@return whether we have a next smallest number
"""
return True if self.stack else False
# Your BSTIterator object will be instantiated and called as such:
# obj = BSTIterator(root)
# param_1 = obj.next()
# param_2 = obj.hasNext()