关闭

【Leetcode】Binary Search Tree Iterator

250人阅读 评论(0) 收藏 举报
分类:

题目链接:https://leetcode.com/problems/binary-search-tree-iterator/

题目:

Implement an iterator over a binary search tree (BST). Your iterator will be initialized with the root node of a BST.

Calling next() will return the next smallest number in the BST.

Note: next() and hasNext() should run in average O(1) time and uses O(h) memory, where h is the height of the tree.

思路:

1、BST中序遍历,遍历结果存到队列中。next不断出队。next和hasNext时间复杂度为O(1),空间复杂度为O(n)。

2、用栈保存,空间复杂度为O(h)。

算法1:

	public class BSTIterator {
		Queue<Integer> q = null;

		public BSTIterator(TreeNode root) {
			q = new LinkedList<Integer>();
			q = inOrder(root, q);
		}

		public Queue<Integer> inOrder(TreeNode p, Queue<Integer> queue) {
			if (p != null) {
				queue = inOrder(p.left, queue);
				queue.offer(p.val);
				queue = inOrder(p.right, queue);
			}
			return queue;
		}

		/**
		 * @return whether we have a next smallest number
		 */
		public boolean hasNext() {
			return !q.isEmpty();
		}

		/**
		 * @return the next smallest number
		 */
		public int next() {
			return q.poll();
		}
	}


算法2:

public class BSTIterator {
	Stack<TreeNode> q = null;

	public BSTIterator(TreeNode root) {
		q = new Stack<TreeNode>();
		while (root != null) {
			q.push(root);
			root = root.left;
		}
	}

	/**
	 * @return whether we have a next smallest number
	 */
	public boolean hasNext() {
		return !q.isEmpty();
	}

	/**
	 * @return the next smallest number
	 */
	public int next() {
		TreeNode t = q.pop();
		TreeNode p = t.right;
		if (p != null) { // 如果t的右子树不为空,则入栈
			q.push(p);
			while (p.left != null) {// 压入右子树自上而下所有最左结点
				q.push(p.left);
				p = p.left;
			}
		}
		return t.val;
	}
}


0
0
查看评论
发表评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场

二叉搜索树(Binary Search Tree )的定义及分析

定义:     二叉搜索树或者是一棵空树,或者是具有下列性质的二叉树: 每个结点都有一个作为搜索依据的关键码(key),所有结点的关键码互不相同。 左子树(如果非空)上所...
  • jsh13417
  • jsh13417
  • 2013-01-22 21:59
  • 3318

二分查找树的C++实现 Binary Search Tree(BST)

用C++实现了二叉查找树的 创建 插入 删除 遍历(宽度优先和三种深度优先[前序,中序,后序]) 查找某值 查找最大值和最小值 清空 复制二叉树(深复制) #include #include u...
  • showonstage1
  • showonstage1
  • 2014-03-13 19:14
  • 1601

Leetcode-Symmetric Tree——判断二叉树是否对称

题目 Symmetric Tree  Total Accepted: 12814 Total Submissions: 40358My Submissions Given a bina...
  • disappearedgod
  • disappearedgod
  • 2014-04-20 01:00
  • 2044

Leetcode 173 Binary Search Tree Iterator 二叉查找树迭代器

实现一个适用于二叉查找树的迭代器,该迭代器通过二叉查找树的根结点来实例化。调用next()返回二叉查找树中的最小的元素。next()和hasNext()运行的平均时间复杂度为O(1),空间复杂度为O(...
  • smile_watermelon
  • smile_watermelon
  • 2015-08-04 18:27
  • 1257

[LeetCode]problem 173. Binary Search Tree Iterator

TAG中序遍历二叉查找树得到一个递增序列二叉查找树题目链接方法这道题挺有意思的,在二叉查找树的基础上建立一个迭代器,要求hasnext和next操作要是平均O(1)的,且最大空间消耗是O(h),h是树...
  • x5942110
  • x5942110
  • 2016-06-15 21:51
  • 198

LeetCode - 173. Binary Search Tree Iterator

要求实现Binary Search Tree的hasNext()和next()操作,最小的元素自然是从root开始一直找左子树左子树直到为空,这是我们找到了我们需要的第一个元素,但是接下来的元素怎么办...
  • shenzhu0127
  • shenzhu0127
  • 2016-07-20 22:20
  • 121

Leetcode: Binary Search Tree Iterator

题目: Implement an iterator over a binary search tree (BST). Your iterator will be initialized with t...
  • T_27080901
  • T_27080901
  • 2015-03-29 00:08
  • 899

[C++]LeetCode: 93 Binary Search Tree Iterator (经典题,非递归的中序遍历)

题目: Implement an iterator over a binary search tree (BST). Your iterator will be initialized with...
  • cinderella_niu
  • cinderella_niu
  • 2015-01-14 16:32
  • 1307

leetcode_173_Binary Search Tree Iterator

思路: 这道题想了好久,知道用中序遍历来解决,用一个list将遍历的元素存储起来一下就解决了,但是空间复杂度不行。具体怎么解决,如何控制程序的终止困扰了我好久。知道我想起来至多用O(h) memor...
  • dfb198998
  • dfb198998
  • 2015-04-03 15:14
  • 575

[leetcode 173] Binary Search Tree Iterator-------为二叉搜索树实现迭代器功能

Question: Implement an iterator over a binary search tree (BST). Your iterator will be initiali...
  • Xiaohei00000
  • Xiaohei00000
  • 2016-03-30 22:23
  • 397
    个人资料
    • 访问:555555次
    • 积分:8547
    • 等级:
    • 排名:第2649名
    • 原创:305篇
    • 转载:6篇
    • 译文:0篇
    • 评论:37条
    博客专栏
    文章分类
    最新评论