day20
同时操作两个二叉树
与 对称二叉树类似,也是要操作两个二叉树,传入的参数是两个子根结点。
注意点:
1.终止条件比较难想
碰到叶子节点要怎么办呢?
1. root1 为 空,就返回 root2 的值。因为 两个二叉树是同步遍历的。
2.root2 为空,就返回 root1 的值。
2. 这里是直接将 root1 进行改造。(我当时想,既然是改造,那 root2为空的时候,没必要返回root1。但其实是错误的。因为 root2 为空,但不进行处理的话,后续会操作空指针。
class Solution {
public:
TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
if(root1 == nullptr)
return root2;
else if(root2 == nullptr)
return root1;
root1->val += root2->val; // 中
root1->left = mergeTrees(root1->left, root2->left); // 左
root1->right = mergeTrees(root1->right, root2->right); // 右
return root1;
}
};