LeetCode_173. 二叉搜索树迭代器

AD_youyu

于 2022-06-13 15:46:39 发布

阅读量78

点赞数

文章标签： leetcode 数据结构算法

本文链接：https://blog.youkuaiyun.com/dwh19992018/article/details/125261038

版权

该博客介绍了如何使用迭代方式实现二叉搜索树的迭代器。定义了一个`BSTIterator`结构体，包含当前节点、栈和栈大小等成员。通过`bSTIteratorCreate`初始化迭代器，`bSTIteratorNext`遍历下一个节点，`bSTIteratorHasNext`检查是否还有更多节点，最后`bSTIteratorFree`释放内存。整个实现充分利用了二叉搜索树的性质，确保了迭代的正确性。

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

173. 二叉搜索树迭代器

本人解答如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */


//迭代
typedef struct {   
    //结构体
    struct TreeNode* cur;
    struct StackTreeNode* stk[128];
    int stkSize;

} BSTIterator;


BSTIterator* bSTIteratorCreate(struct TreeNode* root) {
    //创建新的存储节点
    BSTIterator *ret = malloc(sizeof(BSTIterator));
    ret -> cur = root;
    ret -> stkSize = 0;
    return ret;
}

int bSTIteratorNext(BSTIterator* obj) {
    //左节点
    while(obj -> cur != NULL) {
        obj->stk[(obj->stkSize)++] = obj->cur;
        obj->cur = obj->cur->left;
    }
    obj->cur = obj->stk[--(obj->stkSize)];
    int ret = obj->cur->val;
    //右节点
    obj->cur = obj->cur->right;
    return ret;


    
}

bool bSTIteratorHasNext(BSTIterator* obj) {
    //obj->cur不是空和stkSize有值就是true，反之为false
    return obj -> cur != NULL || obj -> stkSize;
}

void bSTIteratorFree(BSTIterator* obj) {
    //释放
    free(obj);
}

/**
 * Your BSTIterator struct will be instantiated and called as such:
 * BSTIterator* obj = bSTIteratorCreate(root);
 * int param_1 = bSTIteratorNext(obj);
 
 * bool param_2 = bSTIteratorHasNext(obj);
 
 * bSTIteratorFree(obj);
*/