Leetcode#173 Binary Search Tree Iterator

原题地址

普遍的做法是：用栈保存从根节点到当前尚未遍历过的最小节点的路径（栈顶始终是最小节点）

constructor：遍历至最小节点，同时将路径上出现的节点压栈保存

hasNext：返回栈里面是否还有元素

next：栈顶元素即为所求，弹栈的同时更新栈，使栈里始终保存的是从根节点到剩余未遍历节点的最小节点的路径

因为每个元素入栈+出栈各一次，所以平均时间复杂度是O(1)。

代码：

class BSTIterator {
public:
    stack<TreeNode *> st;
    
    BSTIterator(TreeNode *root) {
        pushLeft(root);
    }

    /** @return whether we have a next smallest number */
    bool hasNext() {
        return !st.empty();
    }

    /** @return the next smallest number */
    int next() {
        if (!hasNext())
            return -1;
            
        TreeNode *top = st.top();
        st.pop();

        if (top->right)
            pushLeft(top->right);
        
        return top->val;
    }
    
    void pushLeft(TreeNode *root) {
        while (root) {
            st.push(root);
            root = root->left;
        }
    }
};