Leetcode面试经典150题-173.二叉搜索树迭代器
看本文之前,强烈建议大家看一下我的另一篇文章,那个文章也是面试经典150题之一:
Leetcode面试经典150题-114.二叉树展开为链表-CSDN博客
实现一个二叉搜索树迭代器类BSTIterator
,表示一个按中序遍历二叉搜索树(BST)的迭代器:
BSTIterator(TreeNode root)
初始化BSTIterator
类的一个对象。BST 的根节点root
会作为构造函数的一部分给出。指针应初始化为一个不存在于 BST 中的数字,且该数字小于 BST 中的任何元素。boolean hasNext()
如果向指针右侧遍历存在数字,则返回true
;否则返回false
。int next()
将指针向右移动,然后返回指针处的数字。
注意,指针初始化为一个不存在于 BST 中的数字,所以对 next()
的首次调用将返回 BST 中的最小元素。
你可以假设 next()
调用总是有效的,也就是说,当调用 next()
时,BST 的中序遍历中至少存在一个下一个数字。
示例:
输入 ["BSTIterator", "next", "next", "hasNext", "next", "hasNext", "next", "hasNext", "next", "hasNext"] [[[7, 3, 15, null, null, 9, 20]], [], [], [], [], [], [], [], [], []] 输出 [null, 3, 7, true, 9, true, 15, true, 20, false] 解释 BSTIterator bSTIterator = new BSTIterator([7, 3, 15, null, null, 9, 20]); bSTIterator.next(); // 返回 3 bSTIterator.next(); // 返回 7 bSTIterator.hasNext(); // 返回 True bSTIterator.next(); // 返回 9 bSTIterator.hasNext(); // 返回 True bSTIterator.next(); // 返回 15 bSTIterator.hasNext(); // 返回 True bSTIterator.next(); // 返回 20 bSTIterator.hasNext(); // 返回 False
提示:
- 树中节点的数目在范围
[1, 105]
内 0 <= Node.val <= 106
- 最多调用
105
次hasNext
和next
操作
进阶:
- 你可以设计一个满足下述条件的解决方案吗?
next()
和hasNext()
操作均摊时间复杂度为O(1)
,并使用O(h)
内存。其中h
是树的高度。
我理解我的解法应该就是进阶的解,其他的就不多说了,上代码,看不懂的请留言或者私信,收到第一时间解答
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class BSTIterator {
TreeNode cur;
TreeNode root;
public BSTIterator(TreeNode root) {
this.root = root;
cur = findMostLeft(root);
process(root);
}
public int next() {
TreeNode curNode = cur;
cur = cur.right;
return curNode.val;
}
public boolean hasNext() {
return cur != null;
}
/**这里把树直接转成链表的形式,返回最后一个节点*/
public TreeNode process(TreeNode root) {
if(root == null) {
return null;
}
/**如果就一个节点,当然最后一个节点就是自己 */
if(root.left == null && root.right == null) {
return root;
}
/**先把最后一个节点设置为自己*/
TreeNode last = root;
/**如果左孩子不为空,拿到左孩子最后一个节点,把它的right指针指向自己 */
if(root.left != null) {
last = process(root.left);
last.right = root;
last = root;
}
/**如果右孩子不为空,把之前的最后一个节点的right指向右孩子的最左节点 */
if(root.right != null) {
/**注意这里root的right指针要变了,一定要先把right拿出来 */
TreeNode right = root.right;
root.right = findMostLeft(root.right);
last = process(right);
}
return last;
}
public TreeNode findMostLeft(TreeNode root) {
TreeNode cur = root;
TreeNode mostLeft = null;
while(cur != null) {
mostLeft = cur;
cur = cur.left;
}
return mostLeft;
}
}
/**
* Your BSTIterator object will be instantiated and called as such:
* BSTIterator obj = new BSTIterator(root);
* int param_1 = obj.next();
* boolean param_2 = obj.hasNext();
*/
原文地址:https://blog.csdn.net/Chang_Yafei/article/details/142326677
免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!