173. Binary Search Tree Iterator(二叉查找树迭代器)的C++解法

题目描述：https://www.lintcode.com/problem/binary-search-tree-iterator/description
利用栈：

class BSTIterator {
public:
    stack<TreeNode *> myStack;
    TreeNode *current;
    
    BSTIterator(TreeNode *root) {
        while (!myStack.empty()) {
            myStack.pop();
        }
        current = root;
    }

    bool hasNext() {
        return (current != NULL || !myStack.empty());
    }

    /** @return the next smallest number */
    TreeNode* next() {
        while (current != NULL) {
            myStack.push(current);
            current = current->left;
        }//从当前节点一路向左到头就是最小的；
        current = myStack.top(); myStack.pop();
        TreeNode *nxt = current;
        //为下一个最小值做准备，当前已经到最左边了，下一个点将在这个点的右子树中出现
        current = current->right;
        return nxt;
    }
};

题目要求空间复杂度为O(h),h是二叉查找树的深度，因此考虑每次只放一部分树的结点入栈。首先我们知道根的左子树的左子树的左子树...一直到最左的左子树L一定是最小值，那么先把这部分压入栈，栈顶元素L就是最小值。而下一个最小值，有可能是这个结点的右子树(根L->right，或者根的左子树L->right->.....left)，那么把这部分压入栈中；如果没有的话就向上找L->father(树的结构里没有这个值但是栈中已经保存了)。
图形解释可以看这篇博客。

class BSTIterator {
public:
    stack<TreeNode *> s;
    BSTIterator(TreeNode *root) {
        while(root!=NULL)
       {
           s.push(root);
           root=root->left;
       }
    }
    bool hasNext() {
        if (s.empty()) return false;
        else return true;
    }
    int next() {
        TreeNode * res=s.top();
        s.pop();
        TreeNode * tmp=res->right;
        while (tmp!=NULL)
        {
            s.push(tmp);
            tmp=tmp->left;
        }
        return res->val;
    }
};

• 还有要求空间复杂度为O(1)的，我想可以参考Morris算法。Morris算法主要利用了叶子节点的左右子树均为空的特性，构建一个从叶子到根的回路来代替栈。

另一种理解方法：

class BSTIterator {
public:
    stack<TreeNode *> myStack;
    TreeNode *current;
    
    BSTIterator(TreeNode *root) {
        while (!myStack.empty()) {
            myStack.pop();
        }
        current = root;
    }

    bool hasNext() {
        return (current != NULL || !myStack.empty());
    }

    /** @return the next smallest number */
    TreeNode* next() {
        while (current != NULL) {
            myStack.push(current);
            current = current->left;
        }//从当前节点一路向左到头就是最小的；
        current = myStack.top(); myStack.pop();
        TreeNode *nxt = current;
        //为下一个最小值做准备，当前已经到最左边了，下一个点将在这个点的右子树中出现
        current = current->right;
        return nxt;
    }
};