力扣(LeetCode)173. 二叉搜索树迭代器(C++）

清墨韵染

于 2022-12-21 18:08:29 发布

阅读量418

点赞数 3

分类专栏：墨染leetcode

本文链接：https://blog.csdn.net/Innocence02/article/details/128400148

版权

二叉树中序遍历迭代解法时间复杂度空间复杂度

关键词由CSDN通过智能技术生成

墨染leetcode 专栏收录该内容

138 篇文章 2 订阅

订阅专栏

设计

根据二叉树的中序遍历的迭代解法，稍改代码，就是本题的解法。

初始化 : 传入了根结点，根据迭代思路，将结点的左链依次入栈。

$n e x t$ : 栈顶结点就是所求。根据迭代思路，当前结点要变成栈顶结点的右儿子。由于 $n e x t$ 只想取个值，我们要帮他找到下一个中序结点，将当前结点的左链依次入栈。

$ha s n e x t$ : 结点按中序遍历入栈，看栈内有没有元素，就知道有没有取完结点了。

class BSTIterator {
public:
    stack<TreeNode *> stk;
    BSTIterator(TreeNode* root) {
        while(root){
            stk.push(root);
            root = root->left;
        }
    }
    
    int next(){
        auto t = stk.top();
        stk.pop();
        int ans = t -> val;
        t = t->right;
        while(t){
            stk.push(t);
            t = t->left;
        }
        return ans;
    }

    bool hasNext() {
        return stk.size();
    }
};

时间复杂度 : $O (1)$ ， $h$ 是二叉树的最大深度，对于每个查询， $n e x t$ 最坏时间复杂度 $O (h)$ ，对应走 $h$ 的深度才找到下一个结点。如果查询整个二叉树，每个结点最多入栈一次， $n e x t$ 均摊时间复杂度 $O (1)$ 。
空间复杂度 : $O (h)$ ，最坏空间复杂度 $O (h)$ 。