AcWing
  • 首页
  • 课程
  • 题库
  • 更多
    • 竞赛
    • 题解
    • 分享
    • 问答
    • 应用
    • 校园
  • 关闭
    历史记录
    清除记录
    猜你想搜
    AcWing热点
  • App
  • 登录/注册

173. 二叉搜索树迭代器

作者: 作者的头像   owonder ,  2020-08-13 11:05:48 ,  所有人可见 ,  阅读 680


0


/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class BSTIterator {
public:
   stack<TreeNode*> stk;
    BSTIterator(TreeNode* root) {
        if(root){
stk.push(root);
        if(root->left){
            TreeNode*left=root->left;
            while(left){
                stk.push(left);
                left=left->left;
            }
        }
        }

    }

    /** @return the next smallest number */
    int next() {

            TreeNode*top=stk.top();
            stk.pop();
            if(top->right){
                TreeNode*right=top->right;
                stk.push(right);
                if(right->left){
                    TreeNode*left=right->left;
            while(left){
                stk.push(left);
                left=left->left;
            }
                }
            }
         return top->val;
    }

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

/**
 * Your BSTIterator object will be instantiated and called as such:
 * BSTIterator* obj = new BSTIterator(root);
 * int param_1 = obj->next();
 * bool param_2 = obj->hasNext();
 */

0 评论

App 内打开
你确定删除吗?
1024
x

© 2018-2025 AcWing 版权所有  |  京ICP备2021015969号-2
用户协议  |  隐私政策  |  常见问题  |  联系我们
AcWing
请输入登录信息
更多登录方式: 微信图标 qq图标 qq图标
请输入绑定的邮箱地址
请输入注册信息