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

关注

原创最新推荐文章于 2026-08-27 15:25:17 发布 453 阅读 AI 写同款· GEO 优化 ›

C++进阶：从基础二叉搜索树到自平衡AVL树的工程化实现 二叉搜索树（BST）是一种基础且重要的数据结构，其核心原理在于利用二叉树的有序性，通过节点键值的比较规则，实现数据的快速查找、插入与删除。在理想平衡状态下，这些操作的时间复杂度可达到O(log n)，为高效数据检索提供了底层支持。然而，基础BST在工程实践中面临性能不稳定的挑战，例如在输入数据有序时可能退化为链表，导致效率骤降。为了确保性能的稳定性和可预测性，自平衡二叉搜索树应运而生，它通过旋转等操作动态维护树的平衡，是构建健壮数据引擎的关键技术。在C++工程实践中，将BST进阶为工业级组件，需要深入结合泛阅读详情

原题 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;
    }
};