描述
已知两颗二叉树,将它们合并成一颗二叉树。合并规则是:都存在的结点,就将结点值加起来,否则空的位置就由另一个树的结点来代替。例如:
两颗二叉树是:
合并后的树为
数据范围:树上节点数量满足0≤n≤500,树上节点的值一定在32位整型范围内。
进阶:空间复杂度 O(1),时间复杂度 O(n)
示例1
输入:{1,3,2,5},{2,1,3,#,4,#,7}
返回值:{3,4,5,5,4,#,7}
说明:如题面图
示例2
输入:{1},{}
返回值:{1}
代码
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
class Solution {
public:
/**
*
* @param t1 TreeNode类
* @param t2 TreeNode类
* @return TreeNode类
*/
void addValue(TreeNode* t1, TreeNode* t2){
if(t1==NULL&&t2==NULL)
return;
t1->val+=t2->val;
if(t1->left&&t2->left){
addValue(t1->left, t2->left);
}
if(!t1->left&&t2->left){
t1->left=new TreeNode(0);
addValue(t1->left, t2->left);
}
if(t1->left&&!t2->left){
t2->left=new TreeNode(0);
addValue(t1->left, t2->left);
}
if(t1->right&&t2->right)
addValue(t1->right, t2->right);
if(!t1->right&&t2->right){
t1->right=new TreeNode(0);
addValue(t1->right, t2->right);
}
if(t1->right&&!t2->right){
t2->right=new TreeNode(0);
addValue(t1->right, t2->right);
}
return;
}
TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
// write code here
if(t1==NULL)
return t2;
if(t2==NULL)
return t1;
addValue(t1, t2);
return t1;
}
};