617. 合并二叉树
问题描述:传送门
思路:
感觉像是遍历两个二叉树,其实和遍历一个树逻辑一样。
只不过传入两个数的节点,同时操作。
s
1、递归
以前序遍历为例
① 确定递归函数的参数和返回值
参数:两个二叉树的根节点
返回值:合并之后二叉树的根节点
TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
② 确定终止条件
两个树 t1 和 t2 ,合并的结果。
如果t1== null
,两个树合并就是t2。
反之,亦然。
都是null的情况不用考虑。
if (t1 == NULL) return t2; // 如果t1为空,合并之后就应该是t2
if (t2 == NULL) return t1; // 如果t2为空,合并之后就应该是t1
③ 确定单层递归的逻辑
修改t1的树,成为合并后的树就行。
所以,在单层递归时,把两颗树的元素加到一起。
t1->val += t2->val;
接下来,就是设置
t1的左子树:t1原来左子树+t2左子树
t2的有子树:t1原来右子树+t2右子树
最后,回归t1根节点。
t1->left = mergeTrees(t1->left, t2->left);
t1->right = mergeTrees(t1->right, t2->right);
return t1;
此时前序遍历就写出来了,如下:
class Solution {
public:
TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
if (t1 == NULL) return t2; // 如果t1为空,合并之后就应该是t2
if (t2 == NULL) return t1; // 如果t2为空,合并之后就应该是t1
// 修改了t1的数值和结构
t1->val += t2->val; // 中
t1->left = mergeTrees(t1->left, t2->left); // 左
t1->right = mergeTrees(t1->right, t2->right); // 右
return t1;
}
};
当然,中序遍历也能写出来。
class Solution {
public:
TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
if (t1 == NULL) return t2; // 如果t1为空,合并之后就应该是t2
if (t2 == NULL) return t1; // 如果t2为空,合并之后就应该是t1
// 修改了t1的数值和结构
t1->left = mergeTrees(t1->left, t2->left); // 左
t1->val += t2->val; // 中
t1->right = mergeTrees(t1->right, t2->right); // 右
return t1;
}
};
当然,也可以不去更改t1的结构
class Solution {
public:
TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
if (t1 == NULL) return t2;
if (t2 == NULL) return t1;
// 重新定义新的节点,不修改原有两个树的结构
TreeNode* root = new TreeNode(0);
root->val = t1->val + t2->val;
root->left = mergeTrees(t1->left, t2->left);
root->right = mergeTrees(t1->right, t2->right);
return root;
}
};