[ 二叉树 ] 合并二叉树

617. 合并二叉树 - 力扣（LeetCode） (leetcode-cn.com)

合并二叉树

递归

• 不存在子树对当前树的影响，遍历方式任意

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        if (!root1) return root2;
        if (!root2) return root1;
    	// 注意返回值可能是原root1， 可以是root2
    	// 所以每次需要接收返回值，不能省略
        root1->val += root2->val;
        root1->left = mergeTrees(root1->left, root2->left);
        root1->right = mergeTrees(root1->right, root2->right);
        return root1;
    }
};

非递归

• 层次遍历

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        // 将root2 加到 root1 上 完成 合并
        if (!root1) return root2;
        if (!root2) return root1;
        queue<TreeNode*> que;
        que.push(root1);
        que.push(root2);
        while (!que.empty()) {
            TreeNode* node1 = que.front();
            que.pop(); 
            TreeNode* node2 = que.front();
            que.pop();
            node1->val += node2->val;

            // 如果root1和root2 的子节点都存在，那么正常入列
            if (node1->left && node2->left) {
                que.push(node1->left);
                que.push(node2->left);
            }
            if (node1->right && node2->right) {
                que.push(node1->right);
                que.push(node2->right);
            }
            // 如果root1孩子不存在 而root2的孩子存在，那么把root2的孩子变为root1的孩子
            if (!node1->left && node2->left) node1->left = node2->left;
            if (!node1->right && node2->right) node1->right = node2->right;

            // 至于root1孩子存在，而root2孩子不存在的情况，则不需要操作
        }
        return root1;
    }
};