【LeetCode】617. 合并二叉树（简单）——代码随想录算法训练营Day20

题目链接：617. 合并二叉树

题目描述

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

文章讲解：代码随想录

视频讲解：一起操作两个二叉树？有点懵！| LeetCode：617.合并二叉树_哔哩哔哩_bilibili

题解1：递归法

思路：前序遍历，递归的将 root2 上的元素合并到 root1 上。

/*** 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 && root2) {// 将 root2 上的元素合并到 root1 上root1.val += root2.val;root1.left = mergeTrees(root1.left, root2.left);root1.right = mergeTrees(root1.right, root2.right);return root1;}if (root1) {return root1;}if (root2) {return root2;}return null;
};

分析：时间复杂度为 O(n)，空间复杂度为 O(logn)。

题解2：迭代法

思路：使用前序遍历的迭代法，将 root2 上的元素合并到 root1 上。

/*** 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 && !root2) {return null;}if (!root1) {return root2;}if (!root2) {return root1;}const stack = [[root1, root2]];while (stack.length > 0) {const nodePair = stack.pop();nodePair[0].val += nodePair[1].val; // 中// 右if (nodePair[0].right && nodePair[1].right) {stack.push([nodePair[0].right, nodePair[1].right]);} else if (nodePair[1].right) {nodePair[0].right = nodePair[1].right;}// 左if (nodePair[0].left && nodePair[1].left) {stack.push([nodePair[0].left, nodePair[1].left]);} else if (nodePair[1].left) {nodePair[0].left = nodePair[1].left;}}return root1;
};

分析：时间复杂度为 O(n)，空间复杂度为 O(logn)。