LeetCode -- Binary Search Tree Iterator

最新推荐文章于 2022-06-10 17:57:38 发布

_iorilan

最新推荐文章于 2022-06-10 17:57:38 发布

阅读量1.3k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：数据结构与算法 LeetCode

本文链接：https://blog.csdn.net/lan_liang/article/details/49519959

数据结构与算法同时被 2 个专栏收录

306 篇文章

订阅专栏

LeetCode

229 篇文章

订阅专栏

本文介绍了一种实现二叉搜索树迭代器的方法。该方法通过对二叉搜索树进行中序遍历并将元素存储在队列中来初始化迭代器。通过检查队列是否为空来确定是否有下一个元素，并通过移除队列首元素来获取下一个最小值。

题目描述：

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.

就是实现一个搜索二叉树的迭代器。

思路：
初始化时对树中序遍历元素入队列；hasNext：判断队列是否空；Next:拿出队列当前元素。
缺陷：这个做法的缺陷在于如果只拿1个元素，仍然需要对整树遍历一次；可是大多数情况下，使用迭代器的目的显然是从头遍历到尾，因此这个方案是make sense的。

实现代码：

/**
 * Definition for binary tree
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left;
 *     public TreeNode right;
 *     public TreeNode(int x) { val = x; }
 * }
 */


public class BSTIterator {


    private Queue<int> _values ;
    public BSTIterator(TreeNode root) 
	{
       _values = new Queue<int>();
	   Init(root);
    }
	
	private void Init(TreeNode current)
	{
		if(current == null)
		{
			return;
		}
		Init(current.left);
		_values.Enqueue(current.val);
		Init(current.right);
	}


    /** @return whether we have a next smallest number */
    public bool HasNext() {
        return _values.Count > 0;
    }


    /** @return the next smallest number */
    public int Next() {
		var v = _values.Dequeue();
		return v;
    }
}


/**
 * Your BSTIterator will be called like this:
 * BSTIterator i = new BSTIterator(root);
 * while (i.HasNext()) v[f()] = i.Next();
 */