【leetcode】617. 合并二叉树

题目

leetcode原题链接

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

示例 1：

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

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

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

思路

因为要从根节点开始遍历，也就是前序遍历，使用递归来做，我们比较两棵树的差异，改造第二棵树。改造的过程中我们假设只有不为空的节点才能进入递归，因为我们要比较的是它们的左右子树，这里我称传入的第一棵树为1号树，第二棵树为2号树，即我们改造2号树。

首先第一步是2号树节点值加上1号树的节点值得到新值。然后比较左右子树。

以左子树为例，有几种情况：

第一种：1号树的子树不为空，2号树的子树为空，则要把1号树子树所在的那截树枝给2号树。

第二种情况反过来，则不用任何操作

第三种情况，左子树都为空，也不用做任何操作。

第四种就是左右都有值，这时才需要递归，那么就回到第一步，进入下一层递归。

递归三部曲：

• 递归函数参数和返回值：参数就是两棵树的根节点，没有返回值，我们只是用递归来改造2号树。
• 递归的返回条件：不用返回，因为我们是使用条件判断是否进入下一层递归，即上面分析的第4种情况才进入递归。等待当层函数运行完就自动返回了。
• 单层递归逻辑：2号节点的值累加上1号节点的值。

代码

var mergeTrees = function(root1, root2) {
    
    function traversal(node1 , node2){
        node2.val += node1.val
        
        if(node1.left && !node2.left){
            node2.left = node1.left
            node1.left = null
        }
        if(node1.right && !node2.right){
            node2.right = node1.right
            node1.right = null
        }
        if(node1.left && node2.left){
            traversal(node1.left , node2.left)
        }
        if(node1.right && node2.right){
            traversal(node1.right , node2.right)
        }
    }

    if(!root1 || !root2) return root1 ? root1 : root2
    traversal(root1 , root2)
    return root2
};

关注我的专栏，每天更新三道leetcode题解，一起变强！