LeetCode-173. 二叉搜索树迭代器

实现一个二叉搜索树迭代器。你将使用二叉搜索树的根节点初始化迭代器。

调用 next() 将返回二叉搜索树中的下一个最小的数。

示例：

BSTIterator iterator = new BSTIterator(root);
iterator.next();    // 返回 3
iterator.next();    // 返回 7
iterator.hasNext(); // 返回 true
iterator.next();    // 返回 9
iterator.hasNext(); // 返回 true
iterator.next();    // 返回 15
iterator.hasNext(); // 返回 true
iterator.next();    // 返回 20
iterator.hasNext(); // 返回 false
 

提示：

next() 和 hasNext() 操作的时间复杂度是 O(1)，并使用 O(h) 内存，其中 h 是树的高度。
你可以假设 next() 调用总是有效的，也就是说，当调用 next() 时，BST 中至少存在一个下一个最小的数。

二叉搜索树的基础题目，耗费太多时间了。直接一个中序遍历就可以了。

#include <iostream>
#include <stack>
#include <queue>
using namespace std;

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class BSTIterator {
public:
    BSTIterator(TreeNode* root) {
        if(root==NULL)
            return;

        MidTreeVal(root);
    }

    /** @return the next smallest number */
    int next() {
        int front = TreeQueue.front();
        TreeQueue.pop();
        return front;
    }

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

    /* 中序遍历二叉搜索树 */
    void MidTreeVal(TreeNode *root){
        if(root->left){
            MidTreeVal(root->left);
        }
        TreeQueue.push(root->val);
        if(root->right){
            MidTreeVal(root->right);
        }
    }
private:
    queue<int> TreeQueue;
};


int main(){
    TreeNode *root = new TreeNode(7);
    TreeNode *x1 = new TreeNode(3);
    TreeNode *x2 = new TreeNode(15);
    TreeNode *x3 = new TreeNode(9);
    TreeNode *x4 = new TreeNode(20);

    root->left = x1;
    root->right = x2;
    x2->left = x3;
    x2->right = x4;


    BSTIterator *obj = new BSTIterator(root);
    cout<<obj->next()<<endl;

    system("pause");

    return 0;
}