题目(leecode T617):
给你两棵二叉树: root1
和 root2
。
想象一下,当你将其中一棵覆盖到另一棵之上时,两棵树上的一些节点将会重叠(而另一些不会)。你需要将这两棵树合并成一棵新二叉树。合并的规则是:如果两个节点重叠,那么将这两个节点的值相加作为合并后节点的新值;否则,不为 null 的节点将直接作为新二叉树的节点。
返回合并后的二叉树。
注意: 合并过程必须从两个树的根节点开始。
方法:合并两个二叉树,思路上也是按照同一个逻辑重复操作,因此使用递归法。分析三要素
1:传入参数与返回值:因为要合并两个树,因此传入的是两个树的节点指针。返回的是最终合并好的树的节点指针。
2:确定终止条件:当任一节点指针指向空时就该结束,同时当遍历到空节点时,该节点指向的节点值由另一个树的值确定,即if(node1 == NULL) return node2。因为即使node2为空也是合理的。
3:确定单层的处理逻辑:单层处理是将两个树中对应节点的值相加起来即可。注意这里我们不需要创建一个新的树,只需要使用树1的原始结构上修改即可。
class Solution {
public:
TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
if(root1 == NULL) return root2;
if(root2 == NULL) return root1;
root1->val += root2->val; //利用root1的结构修改
root1->left = mergeTrees(root1->left, root2->left);
root1->right = mergeTrees(root1->right, root2->right);
return root1;
}
};