【算法模板】利用维护栈实现二叉搜索树的迭代器

最新推荐文章于 2025-11-28 22:56:25 发布

原创最新推荐文章于 2025-11-28 22:56:25 发布 · 436 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #leetcode #b树

算法模板笔记专栏收录该内容

17 篇文章

订阅专栏

这篇博客介绍了如何使用栈实现二叉搜索树的中序遍历迭代器，并利用该迭代器合并两个二叉搜索树的所有元素。通过中序遍历得到有序序列，实现了高效地合并两个树的元素。

原题 1305. 两棵二叉搜索树中的所有元素

我们可以利用栈这种数据结构保存每一个二叉搜索树中序遍历的当前状态

可以直接对这两个栈进行判断与操作

这里我参考评论区思路封装了一个迭代器，这样主函数的逻辑就比较清晰

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */

class DBTtreeIter{
    private:
    stack<TreeNode*> stk;
    TreeNode* root;
    public:
    DBTtreeIter(TreeNode* roots) : root(roots) {}

    int peek(){     //取二叉树当前的最小值
        while(root!=nullptr){
            stk.emplace(root);
            root = root -> left;
        }
        return stk.top()->val;
    }
    
    bool hasNest()     // 判断迭代器中还有没有下一个值
    {
        return root != nullptr || !stk.empty();
    }

    int next(){      // 迭代器移动到下一个值，并给一个当前值返回
        int res = 0;
        while(root!=nullptr||!stk.empty())
        {
            if(root!=nullptr)
            {
                stk.emplace(root);
                root = root -> left;
            }
            else{
                root = stk.top();
                stk.pop();
                res = root->val;
                root = root -> right;
                break;
            }
        }
        return res;
    }
};






class Solution {
public:
    vector<int> getAllElements(TreeNode* root1, TreeNode* root2) {
        vector<int> res;
        DBTtreeIter iter1(root1),iter2(root2);

        while(iter1.hasNest()&&iter2.hasNest())
        {
            if(iter1.peek()<iter2.peek()){
                res.emplace_back(iter1.next());
            }
            else{
                res.emplace_back(iter2.next());
            }
        }

        while(iter1.hasNest()) res.emplace_back(iter1.next());
        while(iter2.hasNest()) res.emplace_back(iter2.next());

        return res;
    }
};