力扣面试 150二叉搜索树迭代器中序遍历栈模拟递归步骤拆分

最新推荐文章于 2024-07-22 16:27:42 发布

兑生

最新推荐文章于 2024-07-22 16:27:42 发布

阅读量395

点赞数 9

分类专栏： # 面试150 文章标签： leetcode 面试算法

兑生：前世大厂架构师、上市公司CEO，被小人陷害，重生为25应届生，三连加关注，看我如何逆风翻盘。

本文链接：https://blog.csdn.net/lt6666678/article/details/138171304

版权

面试150 专栏收录该内容

30 篇文章 0 订阅

订阅专栏

Problem: 173. 二叉搜索树迭代器

思路

👩‍🏫 三叶
在这里插入图片描述

复杂度

时间复杂度: $O (1)$

空间复杂度: $O (h)$

Code

class BSTIterator {
Stack<TreeNode> d = new Stack<>();

	public BSTIterator(TreeNode root)
	{
		dfsLeft(root);
	}

	public int next()
	{
		TreeNode root = d.pop();
		int ans = root.val;
		root = root.right;
		dfsLeft(root);
		return ans;
	}

	public boolean hasNext()
	{
		return !d.isEmpty();
	}

	void dfsLeft(TreeNode root)
	{
		while (root != null)
		{
			d.push(root);
			root = root.left;
		}
	}
}