力扣617题：合并二叉树

力扣617题：合并二叉树

题目描述

给你两棵二叉树： root1 和 root2 。

想象一下，当你将其中一棵覆盖到另一棵之上时，两棵树上的一些节点将会重叠（而另一些不会）。你需要将这两棵树合并成一棵新二叉树。合并的规则是：如果两个节点重叠，那么将这两个节点的值相加作为合并后节点的新值；否则，不为 null 的节点将直接作为新二叉树的节点。

返回合并后的二叉树。

注意: 合并过程必须从两个树的根节点开始。

输入输出样例

输入：root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7]
输出：[3,4,5,5,4,null,7]

输入：root1 = [1], root2 = [1,2]
输出：[2,2]

Tips

• 两棵树中的节点数目在范围 [0, 2000] 内
• -104 <= Node.val <= 104

解法一：使用递归DFS

//使用递归进行实现，深度优先搜索
    TreeNode *mergeTrees(TreeNode*root1,TreeNode*root2)
    {
        
        TreeNode *res=dfs(root1,root2);
        return res;
    }

    TreeNode * dfs(TreeNode *root1,TreeNode *root2)
    {
        if(!root1)
        {
            return root2;
        }
        if(!root2)
        {
            return root1;
        }

        TreeNode *tempNode=new TreeNode(root1->val+root2->val);

        tempNode->left=dfs(root1->left,root2->left);
        tempNode->right=dfs(root1->right,root2->right);
        return tempNode;

    }

解法二：使用迭代BFS

//使用迭代进行实现，广度优先搜索
//当两棵树的左节点不为null,就将他们放入队列中
//只要两棵树的右节点都不为null了，也将他们放入队列中
    TreeNode *mergeTrees2(TreeNode*root1,TreeNode*root2)
    {
        //当两颗树有一个为空，则返回另一个树
        if(!root1||!root2)
        {
            return root1==nullptr?root2:root1;
        }

        queue<TreeNode*>que;
        //将两棵树的根节点入队
        que.push(root1);
        que.push(root2);

        //进行层序遍历
        while(!que.empty())
        {
            TreeNode* tempNode1=que.front();
            que.pop();
            TreeNode* tempNode2=que.front();

            
            que.pop();

            //在root1上添加结点
            tempNode1->val+=tempNode2->val;

            //当tempNode1和tempNode2的左子树都不为空，放入队列中
            //若tempNode1的左子树为空，便将tempNode2的左子树挂到tempNode1的左子树上面
            if(tempNode1->left&&tempNode2->left)
            {
                que.push(tempNode1->left);
                que.push(tempNode2->left);
            }
            else if(!tempNode1->left)
            {
                tempNode1->left=tempNode2->left;
            }
            //同理
            if(tempNode1->right&&tempNode2->right)
            {
                que.push(tempNode1->right);
                que.push(tempNode2->right);
            }
            else if(!tempNode1->right)
            {
                tempNode1->right=tempNode2->right;
            }
        }
        return root1;
    }