LeetCode 算法入门 — 合并二叉树

Offer 驾到，掘友接招！我正在参与2022春招打卡活动，点击查看活动详情。

合并二叉树

原题地址

给你两棵二叉树： 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]

思路分析

1. 使用递归块可以很迅速的实现功能；
2. 若 root1 为空时，返回 root2；
3. 若 root2 为空时，返回 root1；
4. 定义一个新的二叉树，以 root1.val + root2.val 作为新二叉树的根节点；
5. 递归获得新二叉树的左节点和右节点；
6. 最终返回新二叉树 root 即可。

AC 代码

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root1
 * @param {TreeNode} root2
 * @return {TreeNode}
 */
var mergeTrees = function(root1, root2) {
    if(!root1) return root2
    if(!root2) return root1
    const root = new TreeNode()
    root.val = root1.val + root2.val
    root.left = mergeTrees(root1.left, root2.left)
    root.right = mergeTrees(root1.right, root2.right)
    return root
};

结果：

• 执行结果： 通过
• 执行用时：112 ms, 在所有 JavaScript 提交中击败了18.58%的用户
• 内存消耗：48.8 MB, 在所有 JavaScript 提交中击败了44.37%的用户

通过测试用例：182 / 182

END