记录：Leetcode173

中等题，理所当然，直接解题。

题目描述

题目本身不难，需要额外考虑的是空间复杂度和时间复杂度的限制。想要实现 next和 hasNext 功能，照理来说只需要造一个数组即可。但问题是，数组的空间复杂的是 O(n)。那么应该怎么办呢？再看一眼条件，会发现，这棵树是二叉搜索树。

思路 1

对于二叉搜索树，左子树 < 根 < 右子树 是确定的。那么我们可以仅仅存储根节点，那么左右节点大小关系也就出来了。
那么我为什么在上图之中把 3 也存进去了呢？因为我原来打算采用递归，那么如何在到达叶子节点之前就停下来，是一个比较麻烦的事情。那么干脆就把叶子 3 也存进去也无妨。
上面的数据结构，我只是画了个长框，没有指明。由于我们遍历树的顺序肯定是先根后叶，但输出要求先左叶，那么无疑，先进后出，采用栈即可。

操作过程如下，每次调用 next，就从栈中弹出一个元素，并将该元素的右子树入栈。当然，右子树无需全部入栈，只进入右子树的左部分即可。

代码如下：

public class BSTIterator {
    private final Stack<TreeNode> min = new Stack<>();


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

    /** @return the next smallest number */
    public int next() {
        TreeNode pop = min.pop();
        insertLeftHand(pop.right);
        return pop.val;
    }

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

    // call when insert left part tree to stack "min"
    public void insertLeftHand(TreeNode root) {
        TreeNode temp = root;
        while (temp != null) {
            min.push(temp);
            temp = temp.left;
        }
    }
}

一次过，很简单，扫了一眼答案，做法相同。那么本文就结束了。

如有错误，欢迎指正交流。