Leetcode 173. Binary Search Tree Iterator

最新推荐文章于 2025-02-03 15:37:10 发布

木车

最新推荐文章于 2025-02-03 15:37:10 发布

阅读量101

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode C++ 文章标签： Leetcode

本文链接：https://blog.youkuaiyun.com/muche001/article/details/79276421

Leetcode C++ 专栏收录该内容

241 篇文章

订阅专栏

本文介绍了一种在二叉搜索树(BST)上实现迭代器的方法。该迭代器通过中序遍历的方式，能够在平均O(1)的时间复杂度内返回BST中的下一个最小值。文章提供了具体的C++实现代码。

原题：

Implement an iterator over a binary search tree (BST). Your iterator will be initialized with the root node of a BST.

Calling next() will return the next smallest number in the BST.

Note: next() and hasNext() should run in average O(1) time and uses O(h) memory, whereh is the height of the tree.

解决方法：

其实一个中序遍历的过程。

从根节点开始加入左节点，这样最左边的叶节点最先弹出，然后是它的父节点，当弹出的父节点时，父节点的右子节点又被加入，如此循环。

代码：

class BSTIterator {
    stack<TreeNode*> st;
    void push(TreeNode* node){
        while(node){
            st.push(node);
            node = node->left;
        }
    }
public:
    BSTIterator(TreeNode *root) {
        push(root);
    }

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

    /** @return the next smallest number */
    int next() {
        TreeNode* node = st.top();
        st.pop();
        push(node->right);
        return node->val;
    }
};