173.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. 
调用next()返回二叉查找树中的最小的元素。

Note: next() and hasNext() should run in average O(1) time and uses O(h) memory, where h is the height of the tree. 
注意：next()和hasNext()运行的平均时间复杂度为O(1)，空间复杂度为O(h)，其中h是树的高度。

解题思路
如果不看注意的话，一个非常简单的解决办法是遍历这棵二叉树，把二叉树的元素以从大到小的方式放到一个栈里，这样next()从栈顶获取元素，hasNext()调用栈不为空的判定方法。这种方法能满足题目中的时间复杂度要求，但是O(n)的空间复杂度无法满足题目中O(h)空间复杂度要求。

我们维护一个栈，暂且称之为最小元素栈，栈中存放的是结点指针，向栈中添加元素发生在两种情况下：1.初始化迭代器时；2.调用next()从栈中弹出一个元素时。我们结合下面的实例来描述这个栈中元素的添加和删除过程。对于hasNext()方法，调用栈的empty()方法即可，不做具体分析，仅分析next()方法以及迭代器的数据结构。

综上所述，我们在栈中存储的是一部分极小值，且栈顶一定是当前树中未访问过的结点中的最小结点。在调用next()弹出最小结点时，需要把被弹出结点的右子树添加到栈中，也就是把比当前最小结点大但是比次小结点小的那些结点中的一部分（不是全部，只是左孩子分支上的一部分）添加到栈中。

然后我们来看，我们的这个数据结构是否满足题目要求。我们可以看到每一个结点一定要进入栈中，且该操作是在构造过程和next()中执行的，其中构造过程只执行一次，而next()执行多次。但是进栈的操作存在差异性，有时候调用next()不会往栈中添加任何元素，有时候会添加多个，但是至多不超过h个，其中h是树的高度，因为添加时只顺着树的左分支往下添加。我们一共要调用n次next()，在这n次调用中，每个结点都进栈/出栈，共n次入栈操作，n次出栈操作，平均起来，每次next()只有一次入栈操作+一次出栈操作，因此满足平均时间复杂度O(1)的要求。再看空间复杂度，刚才我们已经说过，在往栈中添加元素时只顺着树的左分支往下添加，因此至多添加h个结点，其中h就是树的高度，因此满足空间复杂度O(h)的要求。

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

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

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

    /** @return the next smallest number */
    public int next() {
        
        TreeNode res = stack.pop();
        TreeNode cur = res.right;
        while(cur != null){
            stack.push(cur);
            cur = cur.left;
        }
        return res.val;
    }
}

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