leetcode 617.合并二叉树
给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。
注意: 合并必须从两个树的根节点开始。
递归三部曲
1.参数和返回值:输入两个二叉树的根节点,返回值就是合并之后二叉树的根节点。
2.单层递归逻辑:两个数的节点值之和赋值给其中一个节点,然后当前节点的左子树为两个树左子树之和,右子树为两个树的右子树之和
3.终止条件:两个树存在空树在这里插入代码片
class Solution {
public:
TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
if(!root1 && !root2)
return nullptr;
if(!root1)
return root2;
if(!root2)
return root1;
root1->val += root2->val;
root1->left = mergeTrees(root1->left, root2->left);
root1->right = mergeTrees(root1->right, root2->right);
return root1;
}
};