|
| 1 | +# 实现一个二叉搜索树迭代器类BSTIterator ,表示一个按中序遍历二叉搜索树(BST)的迭代器: |
| 2 | +# |
| 3 | +# |
| 4 | +# |
| 5 | +# BSTIterator(TreeNode root) 初始化 BSTIterator 类的一个对象。BST 的根节点 root 会作为构造函数的一部分给出 |
| 6 | +# 。指针应初始化为一个不存在于 BST 中的数字,且该数字小于 BST 中的任何元素。 |
| 7 | +# boolean hasNext() 如果向指针右侧遍历存在数字,则返回 true ;否则返回 false 。 |
| 8 | +# int next()将指针向右移动,然后返回指针处的数字。 |
| 9 | +# |
| 10 | +# |
| 11 | +# 注意,指针初始化为一个不存在于 BST 中的数字,所以对 next() 的首次调用将返回 BST 中的最小元素。 |
| 12 | +# |
| 13 | +# |
| 14 | +# |
| 15 | +# 你可以假设 next() 调用总是有效的,也就是说,当调用 next() 时,BST 的中序遍历中至少存在一个下一个数字。 |
| 16 | +# |
| 17 | +# |
| 18 | +# |
| 19 | +# 示例: |
| 20 | +# |
| 21 | +# |
| 22 | +# 输入 |
| 23 | +# ["BSTIterator", "next", "next", "hasNext", "next", "hasNext", "next", "hasNext |
| 24 | +# ", "next", "hasNext"] |
| 25 | +# [[[7, 3, 15, null, null, 9, 20]], [], [], [], [], [], [], [], [], []] |
| 26 | +# 输出 |
| 27 | +# [null, 3, 7, true, 9, true, 15, true, 20, false] |
| 28 | +# |
| 29 | +# 解释 |
| 30 | +# BSTIterator bSTIterator = new BSTIterator([7, 3, 15, null, null, 9, 20]); |
| 31 | +# bSTIterator.next(); // 返回 3 |
| 32 | +# bSTIterator.next(); // 返回 7 |
| 33 | +# bSTIterator.hasNext(); // 返回 True |
| 34 | +# bSTIterator.next(); // 返回 9 |
| 35 | +# bSTIterator.hasNext(); // 返回 True |
| 36 | +# bSTIterator.next(); // 返回 15 |
| 37 | +# bSTIterator.hasNext(); // 返回 True |
| 38 | +# bSTIterator.next(); // 返回 20 |
| 39 | +# bSTIterator.hasNext(); // 返回 False |
| 40 | +# |
| 41 | +# |
| 42 | +# |
| 43 | +# |
| 44 | +# 提示: |
| 45 | +# |
| 46 | +# |
| 47 | +# 树中节点的数目在范围 [1, 105] 内 |
| 48 | +# 0 <= Node.val <= 106 |
| 49 | +# 最多调用 105 次 hasNext 和 next 操作 |
| 50 | +# |
| 51 | +# |
| 52 | +# |
| 53 | +# |
| 54 | +# 进阶: |
| 55 | +# |
| 56 | +# |
| 57 | +# 你可以设计一个满足下述条件的解决方案吗?next() 和 hasNext() 操作均摊时间复杂度为 O(1) ,并使用 O(h) 内存。其中 h 是树的高 |
| 58 | +# 度。 |
| 59 | +# |
| 60 | +# Related Topics 栈 树 设计 二叉搜索树 二叉树 迭代器 |
| 61 | +# 👍 620 👎 0 |
| 62 | + |
| 63 | + |
| 64 | +# leetcode submit region begin(Prohibit modification and deletion) |
| 65 | +# Definition for a binary tree node. |
| 66 | +# class TreeNode: |
| 67 | +# def __init__(self, val=0, left=None, right=None): |
| 68 | +# self.val = val |
| 69 | +# self.left = left |
| 70 | +# self.right = right |
| 71 | +class BSTIterator: |
| 72 | + |
| 73 | + def __init__(self, root: TreeNode): |
| 74 | + self.root = root |
| 75 | + self.stack = [] |
| 76 | + while self.root: |
| 77 | + self.stack.append(self.root) |
| 78 | + self.root = self.root.left |
| 79 | + |
| 80 | + def next(self) -> int: |
| 81 | + node = self.stack.pop() |
| 82 | + root = node.right |
| 83 | + while root: |
| 84 | + self.stack.append(root) |
| 85 | + root = root.left |
| 86 | + return node.val |
| 87 | + |
| 88 | + def hasNext(self) -> bool: |
| 89 | + if self.stack: |
| 90 | + return True |
| 91 | + else: |
| 92 | + return False |
| 93 | + |
| 94 | + |
| 95 | + |
| 96 | +# Your BSTIterator object will be instantiated and called as such: |
| 97 | +# obj = BSTIterator(root) |
| 98 | +# param_1 = obj.next() |
| 99 | +# param_2 = obj.hasNext() |
| 100 | +# leetcode submit region end(Prohibit modification and deletion) |
0 commit comments