[Lintcode]Binary Search Tree Iterator

原创于 2017-01-19 11:09:36 发布 · 320 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#lintcode

算法专栏收录该内容

191 篇文章

订阅专栏

本文介绍了一种用于遍历二叉搜索树的迭代器实现方法。该方法通过使用栈来实现非递归的中序遍历，可以有效地遍历二叉搜索树并返回节点值。文章提供了一个具体的示例，展示了如何利用栈来完成这一过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Example

For the following binary search tree, in-order traversal by using iterator is [1, 6, 10, 11, 12]

分析：考察非递归中序遍历二叉树。使用一个stack即可。复杂度满足O(h).

public class BSTIterator {
    //@param root: The root of binary tree.
    Stack<TreeNode> stack = new Stack<TreeNode>();
    public BSTIterator(TreeNode root) {
        while(root != null) {
            stack.push(root);
            root = root.left;
        }
    }

    //@return: True if there has next node, or false
    public boolean hasNext() {
        return !stack.isEmpty();
    }
    
    //@return: return next node
    public TreeNode next() {
        TreeNode res = stack.pop();
        TreeNode tmp = null;
        if((tmp = res.right) != null) {
            while(tmp != null) {
                stack.push(tmp);
                tmp = tmp.left;
            }
        }
        return res;
    }
}