LeetCode 每日一题 919. 完全二叉树插入器

题目链接

919. 完全二叉树插入器

注意事项

在构建该数据结构的时候，可以使用两个队列，来找到完全二叉树不含两个子结点的结点

代码

class CBTInserter {
private:
    queue<TreeNode*> q;
    TreeNode* root;
public:
    CBTInserter(TreeNode* root) {

        this->root = root;
        queue<TreeNode*> temp;
        temp.push(root);

        while (!temp.empty()){
            TreeNode *node = temp.front();
            temp.pop();

            if (node->left != nullptr){
                temp.push(node->left);
            }

            if (node->right != nullptr){
                temp.push(node->right);
            }

            if (node->left == nullptr || node->right == nullptr){
                q.push(node);
            }
        }
    }

    int insert(int val) {

        TreeNode *node = new TreeNode(val);
        TreeNode *front = q.front();
        int ans = front->val;

        if (front->left == nullptr){
            front->left = node;
        }else{
            front->right = node;
            q.pop();
        }

        q.push(node);

        return ans;
    }

    TreeNode* get_root() {
        return root;
    }
};