173. Binary Search Tree Iterator

本文介绍了一种二叉搜索树迭代器的实现方法,该迭代器可以在平均O(1)的时间复杂度内返回二叉搜索树中的下一个最小元素,并使用了O(h)的空间复杂度,其中h为树的高度。通过结合栈和树节点的数据结构,利用二叉搜索树的中序遍历特性实现了迭代器的功能。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

173. Binary Search Tree Iterator

Description

mplement 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, where h is the height of the tree.

Implementation

Data structure is combined with Stack and a TreeNode, the tree node has left and right nodes, and the stack is used to store the next number for smallest number.

From the problem we can know, we can know that the this is a binary search problem with in-order search. So in the hasNode I add the left-part value into it and in the output part with pop node, we will set the right node for current node to search the right part.

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class BSTIterator {
public:
    TreeNode* cur;
    stack<TreeNode*> stk;

    BSTIterator(TreeNode *root) {
        this->cur = root;
    }

    void addNode(TreeNode* cur) {
        while(this->cur != NULL) {
            stk.push(this->cur);
            this->cur = this->cur->left;
        }
    }

    /** @return whether we have a next smallest number */
    bool hasNext() {
        if(this->cur != NULL) {
            addNode(this->cur);
            return true;
        }
        else if(stk.size() > 0) 
            return true;
        return false;    
    }

    /** @return the next smallest number */
    int next() {
        TreeNode* tp = stk.top();
        stk.pop();
        cur = tp->right;
        return tp->val;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值