617.合并二叉树

题目

给你两棵二叉树： root1 和 root2 。想象一下，当你将其中一棵覆盖到另一棵之上时，两棵树上的一些节点将会重叠（而另一些不会）。你需要将这两棵树合并成一棵新二叉树。合并的规则是：如果两个节点重叠，那么将这两个节点的值相加作为合并后节点的新值；否则，不为 null 的节点将直接作为新二叉树的节点。返回合并后的二叉树。注意: 合并过程必须从两个树的根节点开始。

示例1

解题思路

//递归
class Solution {
public:
    
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        if(!root1&&!root2)return 0;//考虑最底层，都为空返回null
        TreeNode* root=new TreeNode(0);
        root->left=mergeTrees(root1?root1->left:0,root2?root2->left:0);//中间，处理root左右子树
        root->right=mergeTrees(root1?root1->right:0,root2?root2->right:0);
      
        if(root1)root->val+=root1->val;
        if(root2)root->val+=root2->val;//上层，更新root的value

        return root;  //返回给更上一层
    }
};

//层序遍历
class Solution {
public:
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        if(root1==nullptr||root2==nullptr){
            return root1==nullptr?root2:root1;
        }
        TreeNode* r1;
        TreeNode* r2;
        queue<TreeNode*> Q;
        Q.push(root1);
        Q.push(root2);

        while(Q.size()>0){//因为这题的BFS不需要区分层数，所以不用采取 sz=Q.size()的方法得到一层中的结点个数从而控制循环次数，这种情况下需要使用两个while
            r1=Q.front(); Q.pop();//r1为树1的结点
            r2=Q.front(); Q.pop();//r2为树2中的结点
            r1->val+=r2->val;
            if(r1->left!=nullptr&&r2->left!=nullptr){//只把需要计算的结点压入队列中
                Q.push(r1->left);
                Q.push(r2->left);
            }
            else if(r1->left==nullptr&&r2->left!=nullptr){
                r1->left=r2->left;//说明 r1->left及其下面都为空，直接将r2->left 下面的整颗子树引用过来即可
            }

            if(r1->right!=nullptr&&r2->right!=nullptr){
                Q.push(r1->right);
                Q.push(r2->right);
            }
            else if(r1->right==nullptr&&r2->right!=nullptr){
                r1->right=r2->right;
            }

        }
        return root1;
        
    }
};