Leetcode 173.二叉搜索树迭代器

二叉搜索树迭代器

实现一个二叉搜索树迭代器。你将使用二叉搜索树的根节点初始化迭代器。

调用 next() 将返回二叉搜索树中的下一个最小的数。

注意:next() 和hasNext() 操作的时间复杂度是O(1)，并使用 O(h) 内存，其中 h 是树的高度。

维护一个栈，先将根结点的左子树全部压栈，每次弹出栈顶元素，若某次弹出的栈顶元素有右子树，比如3，此时需要将以该节点的右子树为根的子树的左子节点全部压栈

 

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

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

    public int next(){
        TreeNode minCurrent=stack.pop();
        if(minCurrent.right!=null){
            TreeNode rightNode=minCurrent.right;
            while(rightNode!=null){
                stack.push(rightNode);
                rightNode=rightNode.left;
            }
        }
        return minCurrent.val;
    }
}