算法笔记
本题对于二叉树进行学习,主要特点在于同时对于两棵二叉树进行操作,习惯操作后本题就很容易解决。
题目描述
解题步骤
-
在选择参数方面,传入两颗二叉树的根节点就可以了,返回值为合并后的二叉树根节点。
-
在终止条件处,如果第一棵树节点为空,就返回第二棵树,合并后就是第二棵树;同理对于另一棵树来说也是一样的,至此就可以确定终止条件。
-
当两个树节点都存在时,就可以将两棵树的节点值相加就合并了,这里既可以新建一个root3去装合并的结果,也可以省空间,直接覆盖在某一棵树上面就可以,本题我使用的是第二种方法。对于本道题三种遍历方式都可以,可以按照自己的喜欢去选择,本题使用的遍历方法为前序遍历。
代码部分
class Solution {
public:
TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
if(root1 == nullptr) {
return root2;
}
if(root2 == nullptr) {
return root1;
}
if(root1 != nullptr && root2 != nullptr) {
root1->val = root1->val + root2->val;
}
root1->left = mergeTrees(root1->left, root2->left);
root1->right = mergeTrees(root1->right, root2->right);
return root1;
}
};
结语
本题弄清楚逻辑还是比较简单,希望大伙可以变得更强!