leetcode 173. 二叉搜索树迭代器(用栈完成中序遍历 o(h)空间,就是栈的深度 / miorrs遍历 o(1)空间)

题目
在这里插入图片描述

class BSTIterator {
public:
    stack<TreeNode* >s;
    BSTIterator(TreeNode* root) {
        cal(root);
    }
    void cal(TreeNode* root){//从root一直往左儿子,将这一路都放到栈里
        while(root) s.push(root),root=root->left;
    }
    int next() {
        TreeNode *ans=s.top();s.pop();
        if(ans->right) cal(ans->right);//要弹出ans了,将从ans.right一直往左儿子,这一路都放到栈里
        return ans->val;
    }
    bool hasNext() {
        return s.size()>0;
    }
};

题目要求next() 和 hasNext() 操作的时间复杂度是 O(1),我上一种方法,next()好像是最大o(h)把。
这个方法更不是了。
这里每次next()函数,一定是while(root),不能是if(root),因为if(root),if(root)的话,每次可能root->left,然后添加完🔗,就直接if结束了,然后就没有然后了,,

class BSTIterator {
public:
    TreeNode *root;
    BSTIterator(TreeNode* root) {
        this->root=root;
    }
    
    int next() {
        while(root){
            if(root->left){
                TreeNode *predessor=root->left;
                while(predessor->right!=NULL&&predessor->right!=root) predessor=predessor->right;
                if(predessor->right==NULL){
                    predessor->right=root;
                    root=root->left;
                }else{
                    predessor->right=NULL;
                    TreeNode *tmp=root;
                    root=root->right;
                    return tmp->val;
                }
            }else{
                TreeNode *tmp=root;
                root=root->right;
                return tmp->val;
            }
        }
        return -1;
    }
    
    bool hasNext() {
        return root!=NULL;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值