173:Binary Search Tree Iterator【树】【栈】

题目链接：click~

/*题意：实现BST迭代器类，next()函数返回下一个最小的结点，hasNext()判断是否有下一个最小结点
        要求O(1)的时间和O(h)的的复杂度，h为树的高度
*/

/**
 *思路：next()返回的其实就是中序遍历二叉树时，当前结点的相邻结点
 *          考虑使用一个栈，栈顶代表下一个要返回的结点，栈里保存着从根结点
 *      到栈顶结点的路径，但路径不包含已经出栈的结点。
 *      1)初始化栈：根据中序遍历的顺序，第一个元素是最左结点，将从根结点到
 *        最左结点路径上的结点压入栈。
 *      2)最小值出栈：
          a)栈顶结点有右子树，需将其右子树的最左结点路径上的结点压入栈中
          b)栈顶结点无右子树，那么下一会返回的结点即为其父节点
 */




class BSTIterator {
public:
    stack<TreeNode *> s;
    BSTIterator(TreeNode *root) {
        //初始化，压入最左结点，栈头是最小值
        pushLeft(root);
    }

    /** @return whether we have a next smallest number */
    bool hasNext() {
        //以栈是否为空作为标准
        return !s.empty();
    }

    void pushLeft(TreeNode *T) {
        //压入最左结点
        while(T != NULL) {
            s.push(T);
            T = T->left;
        }
    }

    /** @return the next smallest number */
    int next() {
        TreeNode *T = s.top();
        s.pop();
        //找出栈头的中序遍历的下一个结点
        pushLeft(T->right);
        return T->val;
    }
};