题目简介:
给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。
示例 1:
1 2 3 4 5 6 7 8 9 10 11 12 13 14
| 输入: Tree 1 Tree 2 1 2 / \ / \ 3 2 1 3 / \ \ 5 4 7 输出: 合并后的树: 3 / \ 4 5 / \ \ 5 4 7
|
注意: 合并必须从两个树的根节点开始。
思路:
常规深搜即可。
若两个结点都为空,则直接返回空;若两个结点中有一个为空,则返回另一个;若两个都不为空,则建立一个新结点,其val
值为两个结点的和。
代码如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
|
class Solution { public: TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
if(!t1 && !t2) return NULL;
if(!t1) return t2; if(!t2) return t1;
TreeNode* root = new TreeNode(t1 -> val + t2 -> val);
root -> left = mergeTrees(t1 -> left, t2 -> left); root -> right = mergeTrees(t1 -> right, t2 -> right);
return root; } };
|