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

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